Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import os
 import tempfile
 from decimal import Decimal
@@ -14,9 +13,6 @@ from receipt_total_api import extract_total
 
 app = FastAPI()
 
-# ---------- Pydantic-модели под Go-структуры ----------
-
-
 class Transaction(BaseModel):
     date: str
     amount: Decimal
@@ -51,26 +47,17 @@ class ReceiptResponse(BaseModel):
     total: Optional[float]
 
 
-# ---------- Стартуем и заранее грузим модели ----------
-
 advice_tokenizer = None
 advice_model = None
 
 
 @app.on_event("startup")
 def load_models():
-    """
-    Загружаем Qwen один раз при старте сервиса.
-    Donut для чеков грузится в receipt_total_api при первом импорте.
-    """
     global advice_tokenizer, advice_model
     advice_tokenizer, advice_model = advice_load()
 
 
-# ---------- Эндпоинты ----------
-
-
-from datetime import datetime  # вверху файла, если ещё не импортирован
+from datetime import datetime 
 
 @app.post("/forecast", response_model=ForecastResponse)
 def forecast(req: ForecastRequest):
@@ -88,7 +75,6 @@ def forecast(req: ForecastRequest):
     inc_fc = fit_and_forecast(inc, steps, freq, method=method)
     exp_fc = fit_and_forecast(exp, steps, freq, method=method)
 
-    # Преобразуем индексы в RFC3339: "YYYY-MM-DDT00:00:00Z"
     period_end = [
         d.strftime("%Y-%m-%dT%H:%M:%SZ")
         for d in inc_fc.index.to_pydatetime().tolist()
@@ -140,43 +126,22 @@ def advice(req: AdviceRequest):
         },
     ]
 
-    # 1) детерминированный прогон
     raw = advice_gen(messages, advice_tokenizer, advice_model, det=True)
     text = _to_bullets(clean_ru(raw))
 
-    # 2) если мало пунктов — пробуем стохастический прогон
     if text.count("\n") + 1 < 3:
         raw2 = advice_gen(messages, advice_tokenizer, advice_model, det=False)
         text2 = _to_bullets(clean_ru(raw2))
         if text2:
             text = text2
 
-    # 3) Fallback: если всё равно пусто или там явный мусор — жёстко подставляем шаблон
-    bad = not text.strip() or "Ты финансовый помощник" in text or "Мои данные за текущий месяц" in text
-
-    if bad:
-        # можно потом сделать это умнее и завязать на snap, но пока — вменяемый дефолт
-        text = (
-            "• Ограничьте траты на еду до 15–20% от дохода (для вас ≈ 15 000 ₽ в месяц) и ведите учёт покупок в одном приложении.\n"
-            "• Введите недельный лимит на продукты (например, 3 500–4 000 ₽) и не выходите за него, планируя меню заранее.\n"
-            "• Пересмотрите все платные подписки раз в месяц и отмените те, которыми не пользовались 30 дней и более.\n"
-            "• Сгруппируйте подписки по важности: обязательные, полезные, «для развлечения» — и урежьте последнюю группу минимум на 50%.\n"
-            "• Переключите часть подписок на семейные/годовые тарифы и делите стоимость с друзьями или партнёром.\n"
-            "• Перенесите оплату подписок на один день месяца и установите напоминание за 3 дня до списания, чтобы успевать отменять лишнее.\n"
-            "• Освободившиеся деньги (5–10% дохода) автоматически переводите на отдельный накопительный счёт сразу после зарплаты."
-        )
-
     return AdviceResponse(advice=text)
 
 
 
 @app.post("/receipt-total-file", response_model=ReceiptResponse)
 async def receipt_total_file(file: UploadFile = File(...)):
-    """
-    Получает файл чека (multipart/form-data, field "file"),
-    сохраняет во временный файл, вызывает extract_total и возвращает сумму.
-    """
-    # сохраняем во временный файл
+
     suffix = os.path.splitext(file.filename or "")[1] or ".jpg"
     with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as tmp:
         contents = await file.read()
@@ -192,8 +157,6 @@ async def receipt_total_file(file: UploadFile = File(...)):
         except OSError:
             pass
 
-
-# Для локального запуска, на HF Space это не используется
 if __name__ == "__main__":
     import uvicorn