Update app.py

app.py

@@ -136,16 +136,17 @@ async def initialize_assistant(openai_key, mcp_url):
     await assistant.initialize()
     return f"✅ Initialized with {len(assistant.available_tools)} tools"
 
-def chat_interface(message, history, openai_key, mcp_url):
+async def chat_interface(message, history, openai_key, mcp_url):
     global assistant
     if not assistant:
-        init_result = asyncio.run(initialize_assistant(openai_key, mcp_url))
+        init_result = await initialize_assistant(openai_key, mcp_url)
         if "❌" in init_result:
             history.append([message, init_result])
             return history, ""
 
-    response, _ = asyncio.run(assistant.process_message(message))
-    history.append([message, response])
+    response, tool_info = await assistant.process_message(message)
+    full_response = f"{response}\n\n{tool_info}" if tool_info else response
+    history.append([message, full_response])
     return history, ""
 
 with gr.Blocks() as demo:
@@ -156,7 +157,12 @@ with gr.Blocks() as demo:
     mcp_url = gr.Textbox(label="MCP Server URL")
 
     submit_btn = gr.Button("Send")
-    submit_btn.click(chat_interface, [msg, chatbot, openai_key, mcp_url], [chatbot, msg])
-    msg.submit(chat_interface, [msg, chatbot, openai_key, mcp_url], [chatbot, msg])
+    submit_btn.click(
+        chat_interface, [msg, chatbot, openai_key, mcp_url], [chatbot, msg]
+    )
+    msg.submit(
+        chat_interface, [msg, chatbot, openai_key, mcp_url], [chatbot, msg]
+    )
 
 demo.launch()
+