Update src/tasks.py

src/tasks.py

@@ -4,20 +4,25 @@ from typing import Dict, Any, Optional
 
 @dataclass
 class TaskContext:
-    """
-    Контекст задачи: сюда позже можно добавить:
-    - домен/компанию/стандарты (BABOK, IEEE 29148 и т.п.)
-    - формат вывода (JSON schema)
-    - чеклисты качества
-    """
-    name: str = "base_requirements_analysis"
+    name: str = "requirements_analysis_and_refactor"
     language: str = "ru"
     extra: Dict[str, Any] = None
 
 
-def build_analyzer_prompt(requirements_text: str, ctx: Optional[TaskContext] = None) -> str:
+def _diagram_hint(has_image: bool) -> str:
+    if not has_image:
+        return ""
+    return (
+        "\nДополнительно: пользователь приложил изображение/фото диаграммы. "
+        "Если на ней есть элементы (сущности, связи, состояния, экраны, потоки), "
+        "вытащи их в текст: список объектов, связей, бизнес-правил и неясностей.\n"
+    )
+
+
+def build_analyzer_prompt(requirements_text: str, has_image: bool, ctx: Optional[TaskContext] = None) -> str:
     ctx = ctx or TaskContext()
     return f"""Ты — аналитик требований к ПО. Проанализируй текст требований и выдай структурированный результат.
+{_diagram_hint(has_image)}
 
 Требования (сырой текст):
 {requirements_text}
@@ -37,13 +42,14 @@ def build_analyzer_prompt(requirements_text: str, ctx: Optional[TaskContext] = N
 """
 
 
-def build_refactor_prompt(requirements_text: str, analysis_result: str, ctx: Optional[TaskContext] = None) -> str:
+def build_refactor_prompt(requirements_text: str, analysis_result: str, has_image: bool, ctx: Optional[TaskContext] = None) -> str:
     ctx = ctx or TaskContext()
     return f"""Ты — специалист по рефакторингу требований. Перепиши требования так, чтобы они стали:
 - однозначными,
 - тестируемыми,
 - структурированными,
 - без лишних предположений.
+{_diagram_hint(has_image)}
 
 Исходные требования:
 {requirements_text}
@@ -52,8 +58,8 @@ def build_refactor_prompt(requirements_text: str, analysis_result: str, ctx: Opt
 {analysis_result}
 
 Нужно:
-1) Сгенерировать улучшенную версию требований (желательно: разделы, нумерация, критерии приемки)
-2) Явно отметить места, где без уточнений нельзя сделать требования однозначными (вставь TODO/Questions)
+1) Сгенерировать улучшенную версию требований (разделы, нумерация, критерии приемки)
+2) Явно отметить места, где без уточнений нельзя сделать требования однозначными (TODO/Questions)
 3) Добавить минимальный набор acceptance criteria (Given/When/Then или bullet list)
 
 Формат ответа:
@@ -63,9 +69,10 @@ def build_refactor_prompt(requirements_text: str, analysis_result: str, ctx: Opt
 """
 
 
-def build_critic_prompt(requirements_text: str, refactored: str, ctx: Optional[TaskContext] = None) -> str:
+def build_critic_prompt(requirements_text: str, refactored: str, has_image: bool, ctx: Optional[TaskContext] = None) -> str:
     ctx = ctx or TaskContext()
     return f"""Ты — строгий ревьюер требований. Проверь качество рефакторинга.
+{_diagram_hint(has_image)}
 
 Исходные требования:
 {requirements_text}
@@ -85,7 +92,7 @@ def build_critic_prompt(requirements_text: str, refactored: str, ctx: Optional[T
 - дай финальный вердикт: можно ли отдавать в разработку сейчас
 
 Формат ответа:
-- Оценки (таблично или списком)
+- Оценки
 - Замечания
 - Рекомендованные правки
 - Вердикт