Jeremy Live
commited on
Commit
·
8204547
1
Parent(s):
cdd4766
Revert "m1"
Browse filesThis reverts commit 26e76ca731e37afe51f754cc66b6ecbca1f2ddc8.
app.py
CHANGED
@@ -389,33 +389,6 @@ def execute_sql_query(query, db_connection):
|
|
389 |
except Exception as e:
|
390 |
return f"Error ejecutando la consulta: {str(e)}"
|
391 |
|
392 |
-
def parse_markdown_table(markdown_text: str) -> Optional[List[Dict[str, Any]]]:
|
393 |
-
"""Parse a GitHub-style markdown table string to a list of dicts.
|
394 |
-
|
395 |
-
Returns None if parsing fails or no rows are found.
|
396 |
-
"""
|
397 |
-
if not isinstance(markdown_text, str):
|
398 |
-
return None
|
399 |
-
try:
|
400 |
-
# Keep only table lines with pipes and drop the separator row
|
401 |
-
lines = [
|
402 |
-
line.strip() for line in markdown_text.split('\n')
|
403 |
-
if line.strip() and '|' in line and set(line.strip()) != {'-','|',':',' '}
|
404 |
-
]
|
405 |
-
if len(lines) < 2:
|
406 |
-
return None
|
407 |
-
# First line is header
|
408 |
-
columns = [col.strip() for col in lines[0].split('|')[1:-1]]
|
409 |
-
data: List[Dict[str, Any]] = []
|
410 |
-
for line in lines[1:]:
|
411 |
-
values = [val.strip() for val in line.split('|')[1:-1]]
|
412 |
-
if len(values) != len(columns):
|
413 |
-
continue
|
414 |
-
data.append(dict(zip(columns, values)))
|
415 |
-
return data if data else None
|
416 |
-
except Exception:
|
417 |
-
return None
|
418 |
-
|
419 |
def generate_plot(data, x_col, y_col, title, x_label, y_label):
|
420 |
"""Generate a plot from data and return the file path."""
|
421 |
plt.figure(figsize=(10, 6))
|
@@ -633,18 +606,24 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
|
|
633 |
db_connection, _ = setup_database_connection()
|
634 |
if db_connection:
|
635 |
query_result = execute_sql_query(sql_query2, db_connection)
|
636 |
-
#
|
637 |
-
|
638 |
-
|
639 |
-
|
640 |
-
|
641 |
-
|
642 |
-
|
643 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
644 |
y_col = None
|
645 |
-
for col in columns[1:]:
|
646 |
try:
|
647 |
-
pd.to_numeric(
|
648 |
y_col = col
|
649 |
break
|
650 |
except Exception:
|
@@ -652,15 +631,15 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
|
|
652 |
if y_col:
|
653 |
desired_type = 'pie' if any(k in q_lower for k in ["gráfico circular", "grafico circular", "pie", "pastel"]) else 'bar'
|
654 |
chart_fig = generate_chart(
|
655 |
-
data=
|
656 |
chart_type=desired_type,
|
657 |
x=x_col,
|
658 |
y=y_col,
|
659 |
title=f"{y_col} por {x_col}"
|
660 |
)
|
661 |
if chart_fig is not None:
|
662 |
-
logger.info("Chart generated from second-pass SQL execution
|
663 |
-
chart_state = {"data":
|
664 |
else:
|
665 |
logger.info("No DB connection on second pass; skipping.")
|
666 |
except Exception as e:
|
@@ -683,8 +662,7 @@ async def stream_agent_response(question: str, chat_history: List[List[str]]) ->
|
|
683 |
if not candidate_text and isinstance(response_text, str) and response_text.strip():
|
684 |
candidate_text = response_text
|
685 |
if candidate_text:
|
686 |
-
|
687 |
-
raw_lines = re.split(r"\n|,\s+(?=[^,]+:\s*\d)" , candidate_text)
|
688 |
# Normalize lines: strip bullets and markdown symbols
|
689 |
norm_lines = []
|
690 |
for l in raw_lines:
|
|
|
389 |
except Exception as e:
|
390 |
return f"Error ejecutando la consulta: {str(e)}"
|
391 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
392 |
def generate_plot(data, x_col, y_col, title, x_label, y_label):
|
393 |
"""Generate a plot from data and return the file path."""
|
394 |
plt.figure(figsize=(10, 6))
|
|
|
606 |
db_connection, _ = setup_database_connection()
|
607 |
if db_connection:
|
608 |
query_result = execute_sql_query(sql_query2, db_connection)
|
609 |
+
# Try to parse table-like text into DataFrame if possible
|
610 |
+
data = None
|
611 |
+
if isinstance(query_result, str):
|
612 |
+
try:
|
613 |
+
import pandas as pd
|
614 |
+
df = pd.read_csv(io.StringIO(query_result), sep="|")
|
615 |
+
data = df
|
616 |
+
except Exception:
|
617 |
+
pass
|
618 |
+
# As a fallback, don't rely on text table; just skip charting here
|
619 |
+
if data is not None and hasattr(data, "empty") and not data.empty:
|
620 |
+
# Heuristics: choose first column as x and second as y if numeric
|
621 |
+
x_col = data.columns[0]
|
622 |
+
# pick first numeric column different to x
|
623 |
y_col = None
|
624 |
+
for col in data.columns[1:]:
|
625 |
try:
|
626 |
+
pd.to_numeric(data[col])
|
627 |
y_col = col
|
628 |
break
|
629 |
except Exception:
|
|
|
631 |
if y_col:
|
632 |
desired_type = 'pie' if any(k in q_lower for k in ["gráfico circular", "grafico circular", "pie", "pastel"]) else 'bar'
|
633 |
chart_fig = generate_chart(
|
634 |
+
data=data,
|
635 |
chart_type=desired_type,
|
636 |
x=x_col,
|
637 |
y=y_col,
|
638 |
title=f"{y_col} por {x_col}"
|
639 |
)
|
640 |
if chart_fig is not None:
|
641 |
+
logger.info("Chart generated from second-pass SQL execution.")
|
642 |
+
chart_state = {"data": data, "x_col": x_col, "y_col": y_col, "title": f"{y_col} por {x_col}", "chart_type": desired_type}
|
643 |
else:
|
644 |
logger.info("No DB connection on second pass; skipping.")
|
645 |
except Exception as e:
|
|
|
662 |
if not candidate_text and isinstance(response_text, str) and response_text.strip():
|
663 |
candidate_text = response_text
|
664 |
if candidate_text:
|
665 |
+
raw_lines = candidate_text.split('\n')
|
|
|
666 |
# Normalize lines: strip bullets and markdown symbols
|
667 |
norm_lines = []
|
668 |
for l in raw_lines:
|