Update app.py

app.py

@@ -378,7 +378,7 @@ async def query_context(
 ##### MCP AGENT #####
 sse_url = "https://pharma-ia-gradio-mcp-server.hf.space/gradio_api/mcp/sse"
 
-# Crear instancia del agente (debes tener tu llm definido)
+# Crear instancia del agente (con llm definido)
 mcp_agent = MCPAgent(sse_url, HF_TOKEN, llm)
 
 # Historial de chat
@@ -485,42 +485,41 @@ with gr.Blocks(title="Herramientas MCP") as tools_tab:
 
 # Creamos el Agente MCP (puedes personalizar esto según necesites)
 with gr.Blocks(title="Agente MCP") as agent_tab:
-    gr.Markdown("# Agente MCP - Asistente con Herramientas")
-    gr.Markdown("Interactúa con el agente que puede consultar múltiples fuentes de información especializada.")
+    # Chat principal
+    chatbot = gr.Chatbot(
+        label="Chat con el Agente MCP",
+        height=600,
+        bubble_full_width=True,
+        render_markdown=True,
+        show_label=False
+    )
     
+    # Barra de entrada
     with gr.Row():
-        with gr.Column(scale=3):
-            chatbot = gr.Chatbot(
-                label="Conversación",
-                height=600,
-                bubble_full_width=False,
-                render_markdown=True
-            )
-            msg = gr.Textbox(
-                label="Mensaje",
-                placeholder="Escribe tu pregunta aquí...",
-                container=False
-            )
-            btn = gr.Button("Enviar")
-            clear = gr.ClearButton([msg, chatbot])
-            
-        with gr.Column(scale=1):
-            gr.Markdown("### Herramientas Disponibles")
-            tools_info = gr.JSON(label="Herramientas", value=[])
+        msg = gr.Textbox(
+            placeholder="Escribe tu mensaje aquí...",
+            container=False,
+            scale=9,
+            autofocus=True
+        )
+        btn = gr.Button("Enviar", scale=1)
     
-    # Inicializar el agente y cargar herramientas
+    clear = gr.ClearButton([msg, chatbot], value="Limpiar conversación")
+
+    # Inicialización del agente (sin mostrar herramientas)
     async def init_agent():
-        tools = await mcp_agent.initialize()
-        return {"tools": [tool.name for tool in tools.tools]}
-    
-    agent_tab.load(init_agent, outputs=[tools_info])
+        await mcp_agent.initialize()
+        return gr.Info("Agente listo para conversar")
     
-    # Manejar interacción del chat
+    agent_tab.load(init_agent)
+
+    # Manejo de interacciones
     msg.submit(
         chat_with_agent,
         inputs=[msg, chatbot],
         outputs=[chatbot]
-    )
+    ).then(lambda: "", None, [msg])
+    
     btn.click(
         chat_with_agent,
         inputs=[msg, chatbot],