plotly

app.py

@@ -1,4 +1,5 @@
 import chainlit as cl
+import ast
 import json
 import re
 
@@ -10,6 +11,14 @@ import database
 
 _graph = None
 
+GRAPH_KEYWORDS = ("график", "графика", "тренд", "динамик", "plot", "chart")
+REFUSAL_PATTERNS = (
+    "не могу вызывать внешние библиотеки",
+    "не могу вызывать plotly",
+    "нет возможности выполнять python код",
+    "могу только предоставить данные",
+)
+
 
 def _normalize_message_content(content) -> str:
     if isinstance(content, str):
@@ -57,6 +66,91 @@ def _extract_chart_spec(text: str):
     return None, text
 
 
+def _has_graph_intent(text: str) -> bool:
+    lowered = text.lower()
+    return any(keyword in lowered for keyword in GRAPH_KEYWORDS)
+
+
+def _is_plot_refusal(text: str) -> bool:
+    lowered = text.lower()
+    return any(pattern in lowered for pattern in REFUSAL_PATTERNS)
+
+
+def _parse_structured_content(text: str):
+    try:
+        return json.loads(text)
+    except (json.JSONDecodeError, TypeError):
+        pass
+
+    try:
+        return ast.literal_eval(text)
+    except (ValueError, SyntaxError, TypeError):
+        return None
+
+
+def _is_points_list(value) -> bool:
+    return isinstance(value, list) and value and all(isinstance(item, dict) for item in value)
+
+
+def _extract_points_from_payload(payload):
+    if _is_points_list(payload):
+        return payload
+    if isinstance(payload, dict):
+        for key in ("points", "data", "items", "rows", "result"):
+            candidate = payload.get(key)
+            if _is_points_list(candidate):
+                return candidate
+    return None
+
+
+def _infer_axes(points):
+    if not points:
+        return None, None
+
+    keys = list(points[0].keys())
+    if not keys:
+        return None, None
+
+    x_candidates = ("date", "day", "dt", "timestamp", "period", "period_start")
+    x_field = next((key for key in x_candidates if key in keys), keys[0])
+
+    y_field = None
+    for key in keys:
+        if key == x_field:
+            continue
+        values = [row.get(key) for row in points if isinstance(row, dict)]
+        numeric_values = [val for val in values if isinstance(val, (int, float)) and not isinstance(val, bool)]
+        if numeric_values:
+            y_field = key
+            break
+
+    return x_field, y_field
+
+
+def _chart_spec_from_state_messages(messages):
+    for msg in reversed(messages):
+        if getattr(msg, "type", "") != "tool":
+            continue
+        parsed = _parse_structured_content(_normalize_message_content(getattr(msg, "content", "")))
+        points = _extract_points_from_payload(parsed)
+        if not points:
+            continue
+
+        x_field, y_field = _infer_axes(points)
+        if not x_field or not y_field:
+            continue
+
+        return {
+            "chart_type": "line",
+            "x_field": x_field,
+            "y_field": y_field,
+            "points": points,
+            "title": f"{y_field} по {x_field}",
+        }
+
+    return None
+
+
 def _build_plotly_figure(spec: dict):
     points = spec.get("points", [])
     x_field = spec.get("x_field")
@@ -116,6 +210,13 @@ async def main(message: cl.Message):
     final_state = await _graph.aget_state(config)
     last_msg = _normalize_message_content(final_state.values["messages"][-1].content)
     chart_spec, cleaned_text = _extract_chart_spec(last_msg)
+    graph_requested = _has_graph_intent(message.content or "")
+
+    if not chart_spec and graph_requested:
+        chart_spec = _chart_spec_from_state_messages(final_state.values["messages"])
+
+    if graph_requested and _is_plot_refusal(last_msg):
+        cleaned_text = "Построил график по последним данным из инструментов."
 
     if chart_spec:
         figure = _build_plotly_figure(chart_spec)
@@ -126,4 +227,4 @@ async def main(message: cl.Message):
             ).send()
             return
 
-    await cl.Message(content=last_msg).send()
+    await cl.Message(content=cleaned_text if cleaned_text else last_msg).send()