Revert "v4"

This reverts commit 1e24325c1340661d1725b34139dc3e25d56bb089.

app.py

@@ -572,59 +572,61 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
         response_text = str(response)
     
     logger.info(f"Extracted response text: {response_text[:200]}...")
-    
-    # Check if the response contains an SQL query and it truly looks like SQL
-    sql_query = extract_sql_query(response_text)
-    if sql_query and looks_like_sql(sql_query):
-        logger.info(f"Detected SQL query: {sql_query}")
-        # Execute the query and update the response
-        db_connection, _ = setup_database_connection()
-        if db_connection:
-            query_result = execute_sql_query(sql_query, db_connection)
-            
-            # Add the query and its result to the response
-            response_text += f"\n\n### 🔍 Resultado de la consulta:\n```sql\n{sql_query}\n```\n\n{query_result}"
             
-            # Try to generate an interactive chart if the result is tabular
-            try:
-                if isinstance(query_result, str) and '|' in query_result and '---' in query_result:
-                    # Clean up the markdown table
-                    lines = [line.strip() for line in query_result.split('\n') 
-                            if line.strip() and '---' not in line and '|' in line]
-                    if len(lines) > 1:  # At least header + 1 data row
-                        # Get column names from the first line
-                        columns = [col.strip() for col in lines[0].split('|')[1:-1]]
-                        # Get data rows
-                        data = []
-                        for line in lines[1:]:
-                            values = [val.strip() for val in line.split('|')[1:-1]]
-                            if len(values) == len(columns):
-                                data.append(dict(zip(columns, values)))
-                        
-                        if data and len(columns) >= 2:
-                            # 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]
-                            y_col = columns[1]
-
-                            # Coerce numeric values for y
-                            for row in data:
-                                try:
-                                    row[y_col] = float(re.sub(r"[^0-9.\-]", "", str(row[y_col])))
-                                except Exception:
-                                    pass
-
-                            chart_fig = generate_chart(
-                                data=data,
-                                chart_type=desired_type,
-                                x=x_col,
-                                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)}")
+            # Check if the response contains an SQL query and it truly looks like SQL
+            sql_query = extract_sql_query(response_text)
+            if sql_query and looks_like_sql(sql_query):
+                logger.info(f"Detected SQL query: {sql_query}")
+                # Execute the query and update the response
+                db_connection, _ = setup_database_connection()
+                if db_connection:
+                    query_result = execute_sql_query(sql_query, db_connection)
+                    
+                    # Add the query and its result to the response
+                    response_text += f"\n\n### 🔍 Resultado de la consulta:\n```sql\n{sql_query}\n```\n\n{query_result}"
+                    
+                    # Try to generate an interactive chart if the result is tabular
+                    try:
+                        if isinstance(query_result, str) and '|' in query_result and '---' in query_result:
+                            # Convert markdown table to DataFrame
+                            
+                            # Clean up the markdown table
+                            lines = [line.strip() for line in query_result.split('\n') 
+                                    if line.strip() and '---' not in line and '|' in line]
+                            if len(lines) > 1:  # At least header + 1 data row
+                                # Get column names from the first line
+                                columns = [col.strip() for col in lines[0].split('|')[1:-1]]
+                                # Get data rows
+                                data = []
+                                for line in lines[1:]:
+                                    values = [val.strip() for val in line.split('|')[1:-1]]
+                                    if len(values) == len(columns):
+                                        data.append(dict(zip(columns, values)))
+                                
+                                if data and len(columns) >= 2:
+                                    # 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]
+                                    y_col = columns[1]
+
+                                    # Coerce numeric values for y
+                                    for row in data:
+                                        try:
+                                            row[y_col] = float(re.sub(r"[^0-9.\-]", "", str(row[y_col])))
+                                        except Exception:
+                                            pass
+
+                                    chart_fig = generate_chart(
+                                        data=data,
+                                        chart_type=desired_type,
+                                        x=x_col,
+                                        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