mm2

app.py

@@ -417,6 +417,40 @@ def execute_sql_query(query, db_connection):
     except Exception as e:
         return f"Error ejecutando la consulta: {str(e)}"
 
+def detect_chart_preferences(question: str) -> Tuple[bool, str]:
+    """Detect whether the user is asking for a chart and infer desired type.
+
+    Returns (wants_chart, chart_type) where chart_type is one of
+    {'bar', 'pie', 'line', 'scatter', 'histogram'}.
+    Defaults to 'bar' when ambiguous.
+    """
+    try:
+        q = (question or "").lower()
+
+        # Broad triggers indicating any chart request
+        chart_triggers = [
+            "grafico", "gráfico", "grafica", "gráfica", "chart", "graph",
+            "visualizacion", "visualización", "plot", "plotly", "diagrama"
+        ]
+        wants_chart = any(k in q for k in chart_triggers)
+
+        # Specific type hints
+        if any(k in q for k in ["pastel", "pie", "circular", "donut", "dona", "anillo"]):
+            return wants_chart or True, "pie"
+        if any(k in q for k in ["linea", "línea", "line", "tendencia"]):
+            return wants_chart or True, "line"
+        if any(k in q for k in ["dispersión", "dispersion", "scatter", "puntos"]):
+            return wants_chart or True, "scatter"
+        if any(k in q for k in ["histograma", "histogram"]):
+            return wants_chart or True, "histogram"
+        if any(k in q for k in ["barra", "barras", "columnas", "column"]):
+            return wants_chart or True, "bar"
+
+        # Default
+        return wants_chart, "bar"
+    except Exception:
+        return False, "bar"
+
 def generate_plot(data, x_col, y_col, title, x_label, y_label):
     """Generate a plot from data and return the file path."""
     plt.figure(figsize=(10, 6))
@@ -563,18 +597,8 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
                                         data.append(dict(zip(columns, values)))
                                 
                                 if data and len(columns) >= 2:
-                                    # Determine chart type from user's question (supports pie chart)
-                                    q_lower = question.lower()
-                                    if any(k in q_lower for k in ["gráfico circular", "grafico circular", "pie", "pastel"]):
-                                        desired_type = 'pie'
-                                    elif any(k in q_lower for k in ["línea", "linea", "line"]):
-                                        desired_type = 'line'
-                                    elif any(k in q_lower for k in ["dispersión", "dispersion", "scatter"]):
-                                        desired_type = 'scatter'
-                                    elif any(k in q_lower for k in ["histograma", "histogram"]):
-                                        desired_type = 'histogram'
-                                    else:
-                                        desired_type = 'bar'
+                                    # Determine chart type from user's question
+                                    _, desired_type = detect_chart_preferences(question)
 
                                     # Choose x/y columns (assume first is category, second numeric)
                                     x_col = columns[0]
@@ -608,8 +632,7 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
             # If we still have no chart but the user clearly wants one,
             # try a second pass to get ONLY a SQL query from the agent and execute it.
             if chart_fig is None:
-                q_lower = question.lower()
-                wants_chart = any(k in q_lower for k in ["gráfico", "grafico", "chart", "graph", "pastel", "pie"])
+                wants_chart, default_type = detect_chart_preferences(question)
                 if wants_chart:
                     try:
                         logger.info("Second pass: asking agent for ONLY SQL query in fenced block.")
@@ -648,7 +671,7 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
                                         except Exception:
                                             continue
                                     if y_col:
-                                        desired_type = 'pie' if any(k in q_lower for k in ["gráfico circular", "grafico circular", "pie", "pastel"]) else 'bar'
+                                        desired_type = default_type
                                         chart_fig = generate_chart(
                                             data=data,
                                             chart_type=desired_type,
@@ -663,11 +686,10 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
                     except Exception as e:
                         logger.error(f"Second-pass SQL synthesis failed: {e}")
             
-            # Fallback: if user asked for a chart (e.g., pie) and we didn't get SQL or chart yet,
+            # Fallback: if user asked for a chart and we didn't get SQL or chart yet,
             # parse the most recent assistant text for lines like "LABEL: NUMBER" (bulleted or plain).
             if chart_fig is None:
-                q_lower = question.lower()
-                wants_chart = any(k in q_lower for k in ["gráfico", "grafico", "chart", "graph", "pastel", "pie"])
+                wants_chart, desired_type = detect_chart_preferences(question)
                 if wants_chart:
                     # Find the most recent assistant message with usable numeric pairs
                     candidate_text = ""
@@ -705,7 +727,6 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
                                 data.append({"label": label, "value": val})
                         logger.info(f"Fallback parse from text: extracted {len(data)} items for potential chart")
                         if len(data) >= 2:
-                            desired_type = 'pie' if any(k in q_lower for k in ["gráfico circular", "grafico circular", "pie", "pastel"]) else 'bar'
                             chart_fig = generate_chart(
                                 data=data,
                                 chart_type=desired_type,
@@ -1013,6 +1034,37 @@ def create_application():
             # Append assistant message back into messages history
             chat_history.append({"role": "assistant", "content": assistant_message})
 
+            # If user asked for a chart but none was produced, try to build one
+            # from the latest assistant text using the same fallback logic.
+            if chart_fig is None:
+                wants_chart, desired_type = detect_chart_preferences(question)
+                if wants_chart and isinstance(assistant_message, str):
+                    candidate_text = assistant_message
+                    raw_lines = candidate_text.split('\n')
+                    norm_lines = []
+                    for l in raw_lines:
+                        s = l.strip().lstrip("•*\t -")
+                        if s:
+                            norm_lines.append(s)
+                    data = []
+                    for l in norm_lines:
+                        m = re.match(r"^(.+?):\s*([0-9][0-9.,]*)$", l)
+                        if m:
+                            label = re.sub(r"[*_`]+", "", m.group(1)).strip()
+                            try:
+                                val = float(m.group(2).replace(',', ''))
+                            except Exception:
+                                continue
+                            data.append({"label": label, "value": val})
+                    if len(data) >= 2:
+                        chart_fig = generate_chart(
+                            data=data,
+                            chart_type=desired_type,
+                            x="label",
+                            y="value",
+                            title="Distribución"
+                        )
+
             logger.info("Response generation complete")
             return chat_history, chart_fig