upgrade graph v1

app.py

@@ -568,6 +568,8 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
                                         y=y_col,
                                         title=f"{y_col} por {x_col}"
                                     )
+                                    if chart_fig is not None:
+                                        logger.info(f"Chart generated from SQL table: type={desired_type}, x={x_col}, y={y_col}, rows={len(data)}")
                     except Exception as e:
                         logger.error(f"Error generating chart: {str(e)}", exc_info=True)
                         # Don't fail the whole request if chart generation fails
@@ -589,6 +591,9 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
                         if len(pair) >= 2 and isinstance(pair[1], str) and pair[1].strip():
                             candidate_text = pair[1]
                             break
+                    # Also consider current response_text as a data source
+                    if not candidate_text and isinstance(response_text, str) and response_text.strip():
+                        candidate_text = response_text
                     if candidate_text:
                         raw_lines = candidate_text.split('\n')
                         # Normalize lines: strip bullets and markdown symbols
@@ -613,6 +618,7 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
                                 except Exception:
                                     continue
                                 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(
@@ -622,6 +628,8 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
                                 y="value",
                                 title="Distribución"
                             )
+                            if chart_fig is not None:
+                                logger.info(f"Chart generated from text fallback: type={desired_type}, items={len(data)}")
             
             # Update the assistant's message with the response
             assistant_message["content"] = response_text
@@ -644,6 +652,10 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
             message_content = str(assistant_message)
             
         # Return the assistant's response and an optional interactive chart figure
+        if chart_fig is None:
+            logger.info("No chart generated for this turn.")
+        else:
+            logger.info("Returning a chart figure to UI.")
         return message_content, chart_fig
         
     except Exception as e: