Update app.py

app.py

@@ -14,8 +14,7 @@ def brave_search(query, count=3):
     headers = {"Accept": "application/json", "X-Subscription-Token": API_KEY}
     params = {
         "q": query, 
-        "count": count,
-        "extra_snippets": True
+        "count": count
     }
     
     response = requests.get(url, headers=headers, params=params)
@@ -49,7 +48,7 @@ def handle_chat(user_input, teapot_ai):
 
     context="\n".join(documents)
     
-    response = query_teapot(context, query)
+    response = query_teapot(context, user_input)
     
     # response = teapot_ai.chat([
     #     {