Update app.py

app.py

@@ -139,14 +139,15 @@ async def handle_chat(user_input):
     
         else: # Search task
             search_start_time = time.time()
-            results = await brave_search(user_input)
+            if len(user_input)<400 and "context:" not in user_input and "Context:" not in user_input:
+                results = await brave_search(user_input)
+                if len(results)==0: # No information 
+                    return "I'm sorry but I don't have any information on that.", ""
+                documents = [desc.replace('<strong>', '').replace('</strong>', '') for _, desc, _ in results]
+                context = "\n".join(documents)
+            else:
+                context="" # User provide context
             search_end_time = time.time()
-
-            if len(results)==0:
-                return "I'm sorry but I don't have any information on that.", ""
-            documents = [desc.replace('<strong>', '').replace('</strong>', '') for _, desc, _ in results]
-        
-            context = "\n".join(documents)
         
     generation_start_time = time.time()
     response = await query_teapot(prompt, context, user_input)
@@ -199,9 +200,9 @@ async def on_message(message):
         
     print(message.content)
 
-    is_debug = "debug:" in message.content
+    is_debug = "debug:" in message.content or "Debug:" in message.content
     async with message.channel.typing():
-        cleaned_message=message.content.replace("debug:", "").replace(f'<@{client.user.id}>',"")
+        cleaned_message=message.content.replace("debug:", "").replace("Debug:","").replace(f'<@{client.user.id}>',"")
         response, debug_info = await handle_chat(cleaned_message)
         print(response)
         sent_message = await message.reply(response)