Financial

app.py

@@ -9,25 +9,22 @@ from smolagents import CodeAgent, LiteLLMModel, tool
 
 
 # ----------------------------
-# CPU-friendly defaults
+# Config (CPU-friendly defaults)
 # ----------------------------
 OLLAMA_BASE = os.getenv("OLLAMA_URL", "http://127.0.0.1:11434").rstrip("/")
 
-# Одна модель на всё (можно переопределить в Space Variables)
 MODEL_NAME = os.getenv("MODEL_NAME", "qwen2.5-coder:3b")
-
-# Можно отдельно на роли (опционально):
 TRIAGE_MODEL = os.getenv("TRIAGE_MODEL", MODEL_NAME)
 ACTIONS_MODEL = os.getenv("ACTIONS_MODEL", MODEL_NAME)
 WRITER_MODEL = os.getenv("WRITER_MODEL", MODEL_NAME)
 
-# На CPU большие значения = долго/таймауты
 NUM_CTX = int(os.getenv("NUM_CTX", "4096"))
 MAX_TOKENS = int(os.getenv("MAX_TOKENS", "1024"))
 TIMEOUT = int(os.getenv("LITELLM_TIMEOUT", "3600"))
 
 
 def make_model(model_name: str) -> LiteLLMModel:
+    # flatten_messages_as_text=False помогает избежать проблем формата messages
     return LiteLLMModel(
         model_id=f"ollama_chat/{model_name}",
         api_base=OLLAMA_BASE,
@@ -35,13 +32,12 @@ def make_model(model_name: str) -> LiteLLMModel:
         temperature=0.2,
         max_tokens=MAX_TOKENS,
         timeout=TIMEOUT,
-        # помогает избежать некоторых проблем формата messages в связке smolagents+litellm+ollama
         flatten_messages_as_text=False,
     )
 
 
 def _strip_html(raw_html: str) -> str:
-    """Грубая очистка HTML -> текст, чтобы агент не тащил HTML."""
+    """Грубая очистка HTML -> текст."""
     raw_html = ihtml.unescape(raw_html)
     raw_html = re.sub(r"(?is)<(script|style).*?>.*?</\1>", " ", raw_html)
     raw_html = re.sub(r"(?is)<br\s*/?>", "\n", raw_html)
@@ -54,13 +50,13 @@ def _strip_html(raw_html: str) -> str:
 
 @tool
 def web_search(query: str) -> str:
-    """Ищет краткую информацию в интернете по текстовому запросу (DuckDuckGo HTML).
+    """Поиск в интернете (DuckDuckGo HTML), возвращает очищенный текст.
 
     Args:
         query: Текстовый поисковый запрос.
 
     Returns:
-        Очищенный текст выдачи (обрезан до ~4500 символов).
+        Очищенный текст (обрезан до ~4500 символов).
     """
     resp = requests.get(
         "https://duckduckgo.com/html/",
@@ -72,6 +68,56 @@ def web_search(query: str) -> str:
     return _strip_html(resp.text)[:4500]
 
 
+@tool
+def extract_facts(case_description: str) -> str:
+    """Извлекает факты из описания ситуации без домыслов.
+
+    Args:
+        case_description: Текст пользователя.
+
+    Returns:
+        Строка с фактами + списком недостающих уточнений.
+    """
+    text = (case_description or "").strip()
+    low = text.lower()
+
+    # Ключевые признаки
+    flags = []
+    if re.search(r"\b(просроч|просрочка|пропустил платеж)\b", low):
+        flags.append("просрочка/пропуск платежа")
+    if re.search(r"\b(коллектор|взыскател)\b", low):
+        flags.append("контакт/визит коллекторов")
+    if re.search(r"\b(ипотек)\b", low):
+        flags.append("ипотека")
+    if re.search(r"\b(кредит|займ|мфо)\b", low):
+        flags.append("кредит/займ/МФО")
+    if re.search(r"\b(штраф|пени|неустойк)\b", low):
+        flags.append("штрафы/пени")
+
+    # Числа (в т.ч. суммы)
+    nums = re.findall(r"\d[\d\s]{2,}", text)
+    nums = [re.sub(r"\s+", "", n) for n in nums][:8]
+
+    missing = []
+    if "просроч" in low:
+        missing.append("Сколько дней/месяцев просрочка?")
+    if "коллект" in low:
+        missing.append("Кто кредитор/банк/МФО? (без номеров договоров)")
+        missing.append("Поступают ли угрозы/давление/звонки родственникам?")
+    if not nums:
+        missing.append("Какая сумма долга/минимального платежа/просрочки? (можно диапазоном)")
+    missing.append("Какой официальный доход/примерные обязательные расходы? (можно диапазоном)")
+
+    return (
+        "ФАКТЫ ИЗ ОПИСАНИЯ (без домыслов):\n"
+        f"- Исходный текст: {text}\n"
+        f"- Найденные признаки: {', '.join(flags) if flags else 'не выявлено'}\n"
+        f"- Найденные числа: {', '.join(nums) if nums else 'нет'}\n"
+        f"- Чего не хватает (уточнить безопасно): {('; '.join(missing)) if missing else 'ничего'}\n"
+        "Правило: нельзя добавлять факты, которых нет в исходном тексте."
+    )
+
+
 def run_fin_aid_multi_agent(case_description: str, region: str, urgency: str, allow_internet: bool):
     case_description = (case_description or "").strip()
     region = (region or "").strip()
@@ -80,71 +126,43 @@ def run_fin_aid_multi_agent(case_description: str, region: str, urgency: str, al
     if not case_description:
         return "Пожалуйста, заполните поле «Описание ситуации»."
 
-    tools = [web_search] if allow_internet else []
-
-    if allow_internet:
-        web_hint_triage = """
-Интернет доступен ТОЛЬКО через инструмент web_search(query: str).
-В сгенерированном коде НЕ используй import requests и другие сетевые библиотеки.
-
-ОБЯЗАТЕЛЬНО: минимум 2 вызова web_search:
-1) "emergency financial assistance <регион>" или "rent assistance <регион>";
-2) "how to avoid financial aid scams" или "debt relief scams warning".
-
-Не вставляй сырой HTML — только вывод web_search.
-"""
-        web_hint_actions = """
-Интернет доступен ТОЛЬКО через web_search(query: str).
-Используй результаты поиска, чтобы дополнить варианты помощи и анти-мошенничество.
-Не вставляй HTML.
-"""
-        web_hint_writer = """
-Интернет доступен ТОЛЬКО через web_search(query: str).
-Можно уточнить типовые варианты помощи и анти-мошенничество, но не вставляй HTML.
-"""
-    else:
-        web_hint_common = """
-Интернет недоступен, инструмент web_search отсутствует.
-Не пытайся вызывать web_search.
-"""
-        web_hint_triage = web_hint_common
-        web_hint_actions = web_hint_common
-        web_hint_writer = web_hint_common
+    # Всегда даём extract_facts; web_search — только если allow_internet
+    tools = [extract_facts] + ([web_search] if allow_internet else [])
 
     # ----------------------------
-    # Agent 1: Triage
+    # Agent 1: TRIAGE
     # ----------------------------
-    triage = CodeAgent(
+    triage_agent = CodeAgent(
         tools=tools,
         model=make_model(TRIAGE_MODEL),
         add_base_tools=False,
-        max_steps=3,  # на CPU лучше меньше
+        max_steps=2,  # достаточно для одной попытки исправить код, но без лишних итераций
         additional_authorized_imports=[],
     )
 
     triage_prompt = f"""
-Ты агент TRIAGE (финансовая помощь).
-Всегда отвечай на РУССКОМ языке.
-{web_hint_triage}
+Ты агент TRIAGE (финансовая помощь). Всегда отвечай на РУССКОМ.
 
 КРИТИЧЕСКИ ВАЖНО: Ты работаешь как CodeAgent.
 Ты ДОЛЖЕН вернуть ТОЛЬКО Python-код внутри блока <code>...</code>.
 Ничего вне блока не пиши.
 
-Шаблон вывода (обязательно):
-<code>
-triage_text = \"\"\"...\"\"\"
-triage_text
-</code>
+ОБЯЗАТЕЛЬНО закончить через final_answer(triage_text).
+
+Правила:
+- НЕ выдумывай факты. Используй только входные данные и extract_facts().
+- НЕ предполагай взлом/мошенничество/кражу данных, если пользователь этого не писал.
+- НЕ проси чувствительные данные: номера карт, CVV, пароли, SMS-коды, номера документов.
+- Можно просить безопасные данные: суммы диапазонами, сроки, название кредитора без номеров.
 
-Правила безопасности:
-- НЕ проси и НЕ предлагай вводить чувствительные данные: номера карт, CVV, пароли, коды из SMS, номера документов.
-- НЕ придумывай факты (взлом/мошенничество/кражу данных), если пользователь этого не упомянул.
-- Можно просить безопасные категории: диапазоны сумм, сроки, тип дохода, примерные расходы.
+Если web_search доступен (есть в globals), ОБЯЗАТЕЛЬНО сделай 2 вызова:
+1) web_search(f"emergency financial assistance {region}")
+2) web_search("how to avoid financial aid scams")
+Не вставляй большие куски текста — используй только 3–6 коротких тезисов.
 
-Сформируй triage_text со структурой:
+Нужно сформировать triage_text в структуре:
 ## Сводка ситуации
-## Допущения (если нужно — явно пометь как допущения)
+## Допущения (только если явно помечаешь как допущение)
 ## Приоритеты
 - Сегодня (24–72 часа)
 - На неделе
@@ -153,56 +171,74 @@ triage_text
 ## Какие данные подготовить (безопасно)
 ## Вопросы для уточнения (до 8)
 
-Срочность: {urgency}
-Регион: {region}
-Описание ситуации:
-{case_description}
+Входные данные:
+urgency = \"\"\"{urgency}\"\"\"
+region = \"\"\"{region}\"\"\"
+case_description = \"\"\"{case_description}\"\"\"
 """
-    triage_result = triage.run(textwrap.dedent(triage_prompt))
+
+    triage_result = triage_agent.run(textwrap.dedent(triage_prompt))
 
     # ----------------------------
-    # Agent 2: Actions
+    # Agent 2: ACTIONS
     # ----------------------------
     actions_agent = CodeAgent(
         tools=tools,
         model=make_model(ACTIONS_MODEL),
         add_base_tools=False,
-        max_steps=3,
+        max_steps=2,
         additional_authorized_imports=[],
     )
 
     actions_prompt = f"""
-Ты агент ACTIONS (финансовая помощь).
-Всегда отвечай на РУССКОМ языке.
-{web_hint_actions}
+Ты агент ACTIONS (финансовая помощь). Всегда отвечай на РУССКОМ.
 
 КРИТИЧЕСКИ ВАЖНО: Ты работаешь как CodeAgent.
 Ты ДОЛЖЕН вернуть ТОЛЬКО Python-код внутри блока <code>...</code>.
 Ничего вне блока не пиши.
 
-Шаблон вывода (обязательно):
-<code>
-actions_text = \"\"\"...\"\"\"
-actions_text
-</code>
-
-Требования:
-- На основе triage_result сформируй конкретные действия (шаги), сгруппированные по:
-  urgent / short_term / mid_term
-- Составь список "варианты помощи в регионе" (если нет конкретики — категории: соцслужбы/муниципальные программы/НКО/помощь с жильём/коммуналкой/консультации по долгам).
-- Добавь блок анти-мошенничества (проверки/красные флаги).
-- НЕ выдумывай факты, которых нет во входе.
-- НЕ проси чувствительные данные.
-
-Вход (Triage):
+ОБЯЗАТЕЛЬНО закончить через final_answer(actions_text).
+
+Правила:
+- НЕ выдумывай факты. Опирайся на triage_result и, при необходимости, extract_facts(case_description).
+- НЕ проси чувствительные данные: номера карт, CVV, пароли, SMS-коды, номера документов.
+- Дай конкретные действия, учитывая срочность, если есть риск визитов коллекторов/эскалации взыскания.
+
+Если web_search доступен, можно добавить 2–4 уточнённых категории помощи/анти-скам (без длинных вставок).
+
+Сформируй actions_text (Markdown) с блоками:
+## Действия
+### Срочно (urgent)
+- 4–8 пунктов
+### На неделе (short_term)
+- 4–8 пунктов
+### В течение месяца (mid_term)
+- 3–6 пунктов
+
+## Варианты помощи в регионе
+- 6–10 пунктов (категории: соцзащита/муниципалитет/НКО/юристы по долгам/банк/МФО/коммуналка/жильё)
+
+## Анти-мошенничество
+- 6–10 коротких пунктов, релевантных долгам и “помощи за предоплату”
+
+## Уточняющие вопросы (до 8)
+- только по делу
+
+Входные данные:
+urgency = \"\"\"{urgency}\"\"\"
+region = \"\"\"{region}\"\"\"
+case_description = \"\"\"{case_description}\"\"\"
+
+triage_result:
 {triage_result}
 """
+
     actions_result = actions_agent.run(textwrap.dedent(actions_prompt))
 
     # ----------------------------
-    # Agent 3: Writer (final markdown)
+    # Agent 3: WRITER (final)
     # ----------------------------
-    writer = CodeAgent(
+    writer_agent = CodeAgent(
         tools=tools,
         model=make_model(WRITER_MODEL),
         add_base_tools=False,
@@ -211,57 +247,58 @@ actions_text
     )
 
     writer_prompt = f"""
-Ты агент WRITER (финансовая помощь).
-Всегда отвечай на РУССКОМ языке.
-{web_hint_writer}
+Ты агент WRITER (финансовая помощь). Всегда отвечай на РУССКОМ.
 
-КРИТИЧЕСКИ ВАЖНО: Это ФИНАЛЬНЫЙ шаг.
+КРИТИЧЕСКИ ВАЖНО: Это финальный шаг.
 Ты ДОЛЖЕН вернуть ТОЛЬКО Python-код внутри блока <code>...</code>.
 Ничего вне блока не пиши.
 
-Шаблон вывода (обязательно):
-<code>
-report = \"\"\"...\"\"\"
-report
-</code>
+ОБЯЗАТЕЛЬНО закончить через final_answer(report).
 
-Сделай читабельный Markdown-отчёт, используя triage и actions.
-Структура отчёта:
+Правила:
+- НЕ выдумывай факты.
+- НЕ проси номера карт/CVV/пароли/SMS-коды/номера документов.
+- Не делай большой таблицы на 30 дней — только краткий “мини-бюджет” и пример.
+- Для кейса “просрочка + коллекторы” приоритеты должны отражать безопасность, деэскалацию, фиксацию коммуникации и легальные шаги.
+
+Сформируй report (Markdown) строго по структуре:
 
 # План финансовой помощи
 ## Важно
-- дисклеймер: не юр/фин совет
-- не сообщать коды, CVV, пароли, SMS-коды, номера документов
+- не юр/фин совет
+- безопасность: не сообщать коды/пароли/CVV/SMS/номера документов
 ## Сводка ситуации
 ## Приоритеты
 ### Сегодня (24–72 часа)
 ### На неделе
 ### В течение месяца
-## Варианты помощи в регионе
 ## Пошаговый план
-## Мини-бюджет на 30 дней (без большой таблицы; только подход и пример)
+(используй actions_result)
+## Варианты помощи в регионе
+## Мини-бюджет на 30 дней
+- 5–8 правил + 1 короткий пример расчёта (без таблицы на 30 строк)
 ## Анти-мошенничество
 ## Что подготовить
 ## Вопросы для уточнения
 
-Важные требования:
-- НЕ придумывай события (например взлом), если это не сказано пользователем.
-- Если не хватает данных — прямо напиши "не указано" и вынеси в вопросы.
-- НЕ вставляй Python-структуры dict/list в Markdown.
+Входные данные:
+urgency = \"\"\"{urgency}\"\"\"
+region = \"\"\"{region}\"\"\"
+case_description = \"\"\"{case_description}\"\"\"
 
-Вход (Triage):
+triage_result:
 {triage_result}
 
-Вход (Actions):
+actions_result:
 {actions_result}
 """
-    result = writer.run(textwrap.dedent(writer_prompt))
-    final_markdown = str(result).strip() if result is not None else ""
-    return final_markdown
+
+    final_report = writer_agent.run(textwrap.dedent(writer_prompt))
+    return str(final_report).strip() if final_report is not None else ""
 
 
 with gr.Blocks(title="Financial Aid Navigator (CodeAgent + Ollama)") as demo:
-    gr.Markdown("# Financial Aid Navigator (Multi-agent, CodeAgent + Ollama)")
+    gr.Markdown("# Financial Aid Navigator — Multi-agent (CodeAgent + Ollama)")
     gr.Markdown(
         f"- Ollama: `{OLLAMA_BASE}`\n"
         f"- MODEL_NAME: `{MODEL_NAME}`\n"
@@ -283,10 +320,8 @@ with gr.Blocks(title="Financial Aid Navigator (CodeAgent + Ollama)") as demo:
     )
     case_description = gr.Textbox(
         label="Описание ситуации",
-        lines=9,
-        placeholder=(
-            "Опишите: доходы/расходы (можно диапазонами), долги/просрочки, аренда/ипотека, что самое срочное."
-        ),
+        lines=8,
+        placeholder="Например: Просрочил платеж, коллекторы стучатся в дверь. Сумма долга примерно ...",
     )
     allow_internet = gr.Checkbox(
         label="Разрешить агенту поиск в интернете (web_search)",
@@ -294,13 +329,13 @@ with gr.Blocks(title="Financial Aid Navigator (CodeAgent + Ollama)") as demo:
     )
 
     run_btn = gr.Button("Сформировать план помощи (мультиагент)")
-    output = gr.Textbox(label="План (Markdown)", lines=26)
+    output = gr.Textbox(label="Результат (Markdown)", lines=26)
 
     run_btn.click(
         fn=run_fin_aid_multi_agent,
         inputs=[case_description, region, urgency, allow_internet],
         outputs=[output],
-        queue=False,  # совместимо со старыми gradio
+        queue=False,
     )