Update app.py

app.py

@@ -28,9 +28,9 @@ def brave_search(query, count=3):
         return []
 
 @traceable 
-def query_teapot(context, user_input, teapot_ai):
+def query_teapot(prompt, context, user_input, teapot_ai):
     response = teapot_ai.query(
-        context=context,
+        context=prompt+"\n"+context,
         query=user_input
     )
 
@@ -47,7 +47,7 @@ def handle_chat(user_input, teapot_ai):
     print(documents)
 
     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."
     response = query_teapot(context, user_input, teapot_ai)
     
     # response = teapot_ai.chat([