Update app.py

app.py

@@ -80,6 +80,12 @@ async def handle_chat(user_input):
 
     context = "\n".join(documents)
     prompt = """You are Teapot, an open-source AI assistant optimized for low-end devices, providing short, accurate responses without hallucinating while excelling at information extraction and text summarization."""
+    
+    # Custom prompt shims
+    if "translate" in user_input:
+        context=""
+        prompt=""
+    
     generation_start_time = time.time()
     response = await query_teapot(prompt, context, user_input)