UX de respuesta text segunda grafica

app.py

@@ -542,15 +542,27 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
     except Exception as e:
         logger.error(f"Error updating agent memory: {str(e)}", exc_info=True)
     
-    try:
+        try:
         # Add empty assistant message that will be updated
         assistant_message = {"role": "assistant", "content": ""}
         messages.append(assistant_message)
         
         # Execute the agent with proper error handling
         try:
-            # Let the agent use its memory; don't pass raw chat_history
-            response = await agent.ainvoke({"input": question})
+                # If the user is asking for a chart, steer the agent to be concise
+                wants_chart, _desired = detect_chart_preferences(question)
+                input_text = question
+                if wants_chart:
+                    input_text = (
+                        "Instrucciones: Si el usuario pide una visualización y los datos ya "
+                        "están en el historial, no pidas más contexto. Responde brevemente "
+                        "confirmando que generarás/has generado la gráfica y, si necesitas SQL, "
+                        "devuelve solo la consulta en un bloque ```sql``` sin explicaciones.\n\n"
+                        f"Pregunta del usuario: {question}"
+                    )
+
+                # Let the agent use its memory; don't pass raw chat_history
+                response = await agent.ainvoke({"input": input_text})
             logger.info(f"Agent response type: {type(response)}")
             logger.info(f"Agent response content: {str(response)[:500]}...")
             
@@ -757,6 +769,27 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
         else:
             message_content = str(assistant_message)
             
+        # If we generated a chart but the model said it couldn't, replace with a friendly note
+        def looks_unhelpful(msg: str) -> bool:
+            if not isinstance(msg, str):
+                return False
+            patterns = [
+                r"no t(engo|iene) suficiente información",
+                r"i don'?t have enough information",
+                r"need to know what data",
+                r"can you provide more context",
+                r"not enough info",
+            ]
+            text = msg.lower()
+            return any(re.search(p, text) for p in patterns)
+
+        if chart_fig is not None and looks_unhelpful(message_content):
+            message_content = (
+                "He generado la gráfica solicitada utilizando los datos más "
+                "recientes de la conversación. La visualización interactiva se "
+                "muestra abajo."
+            )
+
         # Return the assistant's response and an optional interactive chart figure
         if chart_fig is None:
             logger.info("No chart generated for this turn.")