Update app.py

app.py

@@ -385,49 +385,51 @@ mcp_agent = MCPAgent(sse_url, HF_TOKEN, llm)
 chat_history = []
 
 def format_tool_call(tool_name, arguments):
-    return f"🔧 **Llamando a herramienta**: `{tool_name}`\n```json\n{arguments}\n```"
+    message = f"🔧 **Llamando a herramienta**: `{tool_name}`\n```json\n{arguments}\n```"
+    print(message)  # Solo imprime en consola
+    return ""  # Devuelve cadena vacía para no mostrar en chat
 
 def format_tool_result(tool_name, result):
-    return f"✅ **Resultado de `{tool_name}`**\n```\n{result}\n```"
+    message = f"✅ **Resultado de `{tool_name}`**\n```\n{result}\n```"
+    print(message)  # Solo imprime en consola
+    return ""  # Devuelve cadena vacía para no mostrar en chat
 
 async def chat_with_agent(message, history):
     history = history or []
     
-    # Agregar solo el mensaje del usuario al historial
-    history.append((message, ""))
-    yield history  # Mostrar inmediatamente el mensaje del usuario
+    # Agregar mensaje del usuario y marcador de "escribiendo"
+    history.append((message, None))  # None activa el indicador de typing
+    yield history
     
-    # Variables para rastrear el último evento mostrado
-    last_tool_call = None
-    last_tool_result = None
+    # Variables para seguimiento
+    processing_message = None
     
-    # Procesar con el agente
-    full_response = ""
-    async for event in mcp_agent.stream_response(message):
-        if event["type"] == "tool_call":
-            if last_tool_call != (event["tool_name"], event["arguments"]):
-                tool_msg = format_tool_call(event["tool_name"], event["arguments"])
-                history.append((None, tool_msg))
-                last_tool_call = (event["tool_name"], event["arguments"])
-                yield history
+    try:
+        # Procesar con el agente
+        full_response = ""
+        async for event in mcp_agent.stream_response(message):
+            if event["type"] == "tool_call":
+                format_tool_call(event["tool_name"], event["arguments"])
                 
-        elif event["type"] == "tool_result":
-            if last_tool_result != (event["tool_name"], event["result"]):
-                result_msg = format_tool_result(event["tool_name"], event["result"])
-                history.append((None, result_msg))
-                last_tool_result = (event["tool_name"], event["result"])
-                yield history
+            elif event["type"] == "tool_result":
+                format_tool_result(event["tool_name"], event["result"])
                 
-        elif event["type"] == "final_response":
-            full_response = event["content"]
-    
-    # Actualizar solo si hay una respuesta final
-    if full_response:
-        # Buscar y reemplazar el último mensaje vacío
-        for i in reversed(range(len(history))):
-            if history[i][1] == "":
-                history[i] = (history[i][0], full_response)
-                break
+            elif event["type"] == "final_response":
+                full_response = event["content"]
+                
+            # Actualizar indicador de actividad periódicamente
+            if not processing_message or len(history[-1][1] or "") > len(processing_message):
+                processing_message = "✍️ Generando respuesta..."
+                history[-1] = (history[-1][0], processing_message)
+                yield history
+        
+        # Mostrar respuesta final
+        if full_response:
+            history[-1] = (history[-1][0], full_response)
+            yield history
+            
+    except Exception as e:
+        history[-1] = (history[-1][0], f"⚠️ Error: {str(e)}")
         yield history