Update app.py

app.py

@@ -84,16 +84,32 @@ def query_teapot(prompt, context, user_input):
 
 @log_time
 def handle_chat(user_input):
-
+    search_start_time = time.time()
     results = brave_search(user_input)
+    search_end_time = time.time()
     
     documents = [desc.replace('<strong>','').replace('</strong>','') for _, desc, _ in results]
 
     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. If a user asks who you are reply "I am Teapot"."""
+    generation_start_time = time.time()
     response = query_teapot(prompt, context, user_input)
+    generation_end_time = time.time()
+    
+    debug_info = f"""
+Prompt: 
+{prompt}
+
+Context: 
+{context}
 
-    return response
+Search time: {search_end_time - search_start_time:.2f} seconds
+Generation time: {generation_end_time - generation_start_time:.2f} seconds
+Response: {response}
+"""
+    
+    
+    return response, debug_info
 
 st.write("418 I'm a teapot")
 
@@ -118,10 +134,23 @@ async def on_message(message):
     if message.author == client.user:
         return
     print(message.content)
+
+
+    is_debug = "<debug>" in message.content
+    
     # Respond with "pong" if the message contains "ping"
-    response = handle_chat(message.content)
+    response, debug_info = handle_chat(message.content.replace("<debug>","").replace("</debug>",""))
+    
     print(response)
-    await message.channel.send(response)
+    sent_message = await message.channel.send(response)
+
+    # Create a thread from the sent message
+    if is_debug:
+        thread = await sent_message.create_thread(name="Debug Thread", auto_archive_duration=60)
+    
+        # Send a message in the created thread
+        await thread.send(debug_info)
+
 
 # Run the bot with your token