Update src/tasks.py

src/tasks.py

@@ -9,23 +9,18 @@ class TaskContext:
     extra: Dict[str, Any] = None
 
 
-def _diagram_hint(has_image: bool) -> str:
-    if not has_image:
+def _diagram_block(diagram_context: Optional[str]) -> str:
+    if not diagram_context or not diagram_context.strip():
         return ""
-    return (
-        "\nДополнительно: пользователь приложил изображение/фото диаграммы. "
-        "Если на ней есть элементы (сущности, связи, состояния, экраны, потоки), "
-        "вытащи их в текст: список объектов, связей, бизнес-правил и неясностей.\n"
-    )
+    return f"\n\nКонтекст из диаграммы (OCR/описание):\n{diagram_context}\n"
 
 
-def build_analyzer_prompt(requirements_text: str, has_image: bool, ctx: Optional[TaskContext] = None) -> str:
+def build_analyzer_prompt(requirements_text: str, diagram_context: Optional[str] = None, ctx: Optional[TaskContext] = None) -> str:
     ctx = ctx or TaskContext()
     return f"""Ты — аналитик требований к ПО. Проанализируй текст требований и выдай структурированный результат.
-{_diagram_hint(has_image)}
 
 Требования (сырой текст):
-{requirements_text}
+{requirements_text}{_diagram_block(diagram_context)}
 
 Нужно:
 1) Выявить неясности/двусмысленности
@@ -42,17 +37,16 @@ def build_analyzer_prompt(requirements_text: str, has_image: bool, ctx: Optional
 """
 
 
-def build_refactor_prompt(requirements_text: str, analysis_result: str, has_image: bool, ctx: Optional[TaskContext] = None) -> str:
+def build_refactor_prompt(requirements_text: str, analysis_result: str, diagram_context: Optional[str] = None, ctx: Optional[TaskContext] = None) -> str:
     ctx = ctx or TaskContext()
     return f"""Ты — специалист по рефакторингу требований. Перепиши требования так, чтобы они стали:
 - однозначными,
 - тестируемыми,
 - структурированными,
 - без лишних предположений.
-{_diagram_hint(has_image)}
 
 Исходные требования:
-{requirements_text}
+{requirements_text}{_diagram_block(diagram_context)}
 
 Результат анализа:
 {analysis_result}
@@ -60,7 +54,7 @@ def build_refactor_prompt(requirements_text: str, analysis_result: str, has_imag
 Нужно:
 1) Сгенерировать улучшенную версию требований (разделы, нумерация, критерии приемки)
 2) Явно отметить места, где без уточнений нельзя сделать требования однозначными (TODO/Questions)
-3) Добавить минимальный набор acceptance criteria (Given/When/Then или bullet list)
+3) Добавить минимальный набор acceptance criteria
 
 Формат ответа:
 - Рефакторинг требований
@@ -69,13 +63,12 @@ def build_refactor_prompt(requirements_text: str, analysis_result: str, has_imag
 """
 
 
-def build_critic_prompt(requirements_text: str, refactored: str, has_image: bool, ctx: Optional[TaskContext] = None) -> str:
+def build_critic_prompt(requirements_text: str, refactored: str, diagram_context: Optional[str] = None, ctx: Optional[TaskContext] = None) -> str:
     ctx = ctx or TaskContext()
     return f"""Ты — строгий ревьюер требований. Проверь качество рефакторинга.
-{_diagram_hint(has_image)}
 
 Исходные требования:
-{requirements_text}
+{requirements_text}{_diagram_block(diagram_context)}
 
 Рефакторинг:
 {refactored}
@@ -87,9 +80,9 @@ def build_critic_prompt(requirements_text: str, refactored: str, has_image: bool
 4) Структура и читабельность
 
 Также:
-- укажи конкретные проблемы (с примерами)
+- укажи конкретные проблемы
 - предложи правки
-- дай финальный вердикт: можно ли отдавать в разработку сейчас
+- дай вердикт: можно ли отдавать в разработку сейчас
 
 Формат ответа:
 - Оценки