plotly

.idea/ChatWB.iml

@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="jdk" jdkName="Python 3.12" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+  <component name="PyDocumentationSettings">
+    <option name="format" value="PLAIN" />
+    <option name="myDocStringFormat" value="Plain" />
+  </component>
+</module>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/material_theme_project_new.xml

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="MaterialThemeProjectNewConfig">
+    <option name="metadata">
+      <MTProjectMetadataState>
+        <option name="userId" value="-312bc485:19d5d08a8f6:-7fe5" />
+      </MTProjectMetadataState>
+    </option>
+  </component>
+</project>

.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/ChatWB.iml" filepath="$PROJECT_DIR$/.idea/ChatWB.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,7 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>

.idea/workspace.xml

@@ -0,0 +1,72 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ChangeListManager">
+    <list default="true" id="d8567411-95a5-4448-96bf-d0f9581df121" name="Changes" comment="">
+      <change beforePath="$PROJECT_DIR$/app.py" beforeDir="false" afterPath="$PROJECT_DIR$/app.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/prompts.py" beforeDir="false" afterPath="$PROJECT_DIR$/prompts.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/requirements.txt" beforeDir="false" afterPath="$PROJECT_DIR$/requirements.txt" afterDir="false" />
+    </list>
+    <option name="SHOW_DIALOG" value="false" />
+    <option name="HIGHLIGHT_CONFLICTS" value="true" />
+    <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
+    <option name="LAST_RESOLUTION" value="IGNORE" />
+  </component>
+  <component name="Git.Settings">
+    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
+  </component>
+  <component name="ProjectColorInfo"><![CDATA[{
+  "associatedIndex": 4
+}]]></component>
+  <component name="ProjectId" id="3BvvXYBsEELfdGehPTtVAqCklWA" />
+  <component name="ProjectViewState">
+    <option name="hideEmptyMiddlePackages" value="true" />
+    <option name="showLibraryContents" value="true" />
+  </component>
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "ModuleVcsDetector.initialDetectionPerformed": "true",
+    "RunOnceActivity.ShowReadmeOnStart": "true",
+    "RunOnceActivity.TerminalTabsStorage.copyFrom.TerminalArrangementManager.252": "true",
+    "RunOnceActivity.git.unshallow": "true",
+    "RunOnceActivity.typescript.service.memoryLimit.init": "true",
+    "ai.playground.ignore.import.keys.banner.in.settings": "true",
+    "com.intellij.ml.llm.matterhorn.ej.ui.settings.DefaultModelSelectionForGA.v1": "true",
+    "git-widget-placeholder": "main",
+    "junie.onboarding.icon.badge.shown": "true",
+    "last_opened_file_path": "/home/levin/Документы/dev/AI_WB/ChatWB",
+    "nodejs_package_manager_path": "npm",
+    "to.speed.mode.migration.done": "true",
+    "vue.rearranger.settings.migration": "true"
+  }
+}]]></component>
+  <component name="SharedIndexes">
+    <attachedChunks>
+      <set>
+        <option value="bundled-js-predefined-d6986cc7102b-9b0f141eb926-JavaScript-PY-253.31033.139" />
+        <option value="bundled-python-sdk-2653e85de345-6d6dccd035ac-com.jetbrains.pycharm.pro.sharedIndexes.bundled-PY-253.31033.139" />
+      </set>
+    </attachedChunks>
+  </component>
+  <component name="TaskManager">
+    <task active="true" id="Default" summary="Default task">
+      <changelist id="d8567411-95a5-4448-96bf-d0f9581df121" name="Changes" comment="" />
+      <created>1775382398674</created>
+      <option name="number" value="Default" />
+      <option name="presentableId" value="Default" />
+      <updated>1775382398674</updated>
+      <workItem from="1775382400134" duration="142000" />
+    </task>
+    <servers />
+  </component>
+  <component name="TypeScriptGeneratedFilesManager">
+    <option name="version" value="3" />
+  </component>
+  <component name="github-copilot-workspace">
+    <instructionFileLocations>
+      <option value=".github/instructions" />
+    </instructionFileLocations>
+    <promptFileLocations>
+      <option value=".github/prompts" />
+    </promptFileLocations>
+  </component>
+</project>

app.py

@@ -1,9 +1,4 @@
 import chainlit as cl
-import ast
-import json
-import re
-
-import plotly.graph_objects as go
 
 from graph import workflow
 from database import init_storage
@@ -11,177 +6,6 @@ import database
 
 _graph = None
 
-GRAPH_KEYWORDS = ("график", "графика", "тренд", "динамик", "plot", "chart")
-REFUSAL_PATTERNS = (
-    "не могу вызывать внешние библиотеки",
-    "не могу вызывать plotly",
-    "нет возможности выполнять python код",
-    "могу только предоставить данные",
-)
-
-
-def _normalize_message_content(content) -> str:
-    if isinstance(content, str):
-        return content
-    if isinstance(content, list):
-        parts = []
-        for item in content:
-            if isinstance(item, dict) and item.get("type") == "text":
-                parts.append(item.get("text", ""))
-            elif isinstance(item, str):
-                parts.append(item)
-        return "\n".join(p for p in parts if p)
-    return str(content)
-
-
-def _is_chart_spec(obj: dict) -> bool:
-    required = {"chart_type", "x_field", "y_field", "points"}
-    return isinstance(obj, dict) and required.issubset(obj.keys()) and isinstance(obj.get("points"), list)
-
-
-def _extract_chart_spec(text: str):
-    tag_pattern = re.compile(r"<chart_spec>\s*(\{[\s\S]*?\})\s*</chart_spec>", re.IGNORECASE)
-    code_pattern = re.compile(r"```json\s*(\{[\s\S]*?\})\s*```", re.IGNORECASE)
-
-    for pattern in (tag_pattern, code_pattern):
-        match = pattern.search(text)
-        if not match:
-            continue
-        payload = match.group(1)
-        try:
-            spec = json.loads(payload)
-        except json.JSONDecodeError:
-            continue
-        if _is_chart_spec(spec):
-            cleaned = (text[:match.start()] + text[match.end():]).strip()
-            return spec, cleaned
-
-    try:
-        raw_obj = json.loads(text)
-        if _is_chart_spec(raw_obj):
-            return raw_obj, ""
-    except json.JSONDecodeError:
-        pass
-
-    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")
-    y_field = spec.get("y_field")
-    chart_type = str(spec.get("chart_type", "line")).lower()
-
-    if not points or not x_field or not y_field:
-        return None
-
-    x_values = [point.get(x_field) for point in points if isinstance(point, dict)]
-    y_values = [point.get(y_field) for point in points if isinstance(point, dict)]
-    if not x_values or not y_values:
-        return None
-
-    title = spec.get("title") or f"{y_field} по {x_field}"
-    fig = go.Figure()
-
-    if chart_type == "bar":
-        fig.add_trace(go.Bar(x=x_values, y=y_values, name=y_field))
-    else:
-        fig.add_trace(go.Scatter(x=x_values, y=y_values, mode="lines+markers", name=y_field))
-
-    fig.update_layout(
-        title=title,
-        xaxis_title=x_field,
-        yaxis_title=y_field,
-        template="plotly_white",
-        margin=dict(l=40, r=20, t=50, b=40),
-    )
-    return fig
-
 @cl.on_chat_start
 async def start():
     global _graph
@@ -208,23 +32,5 @@ async def main(message: cl.Message):
         pass
 
     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)
-        if figure:
-            await cl.Message(
-                content=cleaned_text or "Построил график по полученным данным.",
-                elements=[cl.Plotly(name="chart", figure=figure, display="inline")],
-            ).send()
-            return
-
-    await cl.Message(content=cleaned_text if cleaned_text else last_msg).send()
+    last_msg = final_state.values["messages"][-1].content
+    await cl.Message(content=last_msg).send()

prompts.py

@@ -141,10 +141,8 @@ ADS_ANALYST_PROMPT = """
 
     Уверенность (score) ставь адекватно: 0.5 (слабая гипотеза) до 1.0 (очевидный слив бюджета).
 
-    Если пользователь просит график, обязательно вызывай тренд-инструмент и возвращай структуру для визуализации
-    в отдельном JSON-блоке ```json``` с объектом chart_spec.
-    Обязательные поля: chart_type (line/bar), x_field, y_field, points.
-    В points используй только сырые точки из tools (например, [{"date": "2026-04-01", "ad_spend_total": 12345.67}]).
+    Если пользователь просит график, обязательно вызывай тренд-инструмент и возвращай данные по точкам
+    (например: дата и значение метрики) для последующего построения графика во внешнем интерфейсе.
 
     Обработка ошибок: Если инструмент вернул словарь с ключом "error" (например, {"error": "Товар не найден"}), не выдумывай данные. Прямо сообщи пользователю, что информацию найти не удалось, опираясь на текст ошибки.
 """
@@ -288,9 +286,7 @@ AI-Рекомендации:
     Если инструмент вернул ошибку или пустой ответ — честно сообщи об этом пользователю.
 
     Если пользователь просит график, вызови соответствующий инструмент тренда (например, get_sales_trend)
-    и верни структуру для визуализации в отдельном JSON-блоке ```json``` с объектом chart_spec.
-    Обязательные поля: chart_type (line/bar), x_field, y_field, points.
-    Не подменяй график текстовым описанием, если есть данные для построения.
+    и верни данные по точкам (ось X и метрика Y) для последующего построения графика во внешнем интерфейсе.
 """
 
 LOGISTICS_ANALYST_PROMPT = """
@@ -505,9 +501,8 @@ LOGISTICS_ANALYST_PROMPT = """
     Оценивай уверенность (score) от 0.5 (есть сомнения) до 1.0 (проблема критичная, действовать немедленно).
 
     Если пользователь просит график или динамику, обязательно вызывай тренд-инструмент
-    (например, get_logistics_trend) и возвращай структуру для визуализации
-    в отдельном JSON-блоке ```json``` с объектом chart_spec.
-    Обязательные поля: chart_type (line/bar), x_field, y_field, points.
+    (например, get_logistics_trend) и возвращай данные по точкам (ось X и метрика Y)
+    для последующего построения графика во внешнем интерфейсе.
 
     Если инструмент API возвращает ошибку, честно скажи об этом.
 """
@@ -703,8 +698,7 @@ FINANCE_ANALYST_PROMPT = """
     Уверенность (score) при сохранении инсайта ставь от 0.5 (гипотеза) до 1.0 (критическая проблема).
 
     Если пользователь просит график, обязательно вызывай get_finance_trend и возвращай
-    структуру для визуализации в отдельном JSON-блоке ```json``` с объектом chart_spec.
-    Обязательные поля: chart_type (line/bar), x_field, y_field, points.
+    данные по точкам (ось X и метрика Y) для последующего построения графика во внешнем интерфейсе.
 
     Вопросы по остаткам и складам -> перенаправляй к агенту "Логист / Остатки".
 

requirements.txt

@@ -26,4 +26,3 @@ asyncpg
 sqlalchemy
 pandas
 numpy
-plotly