Update app.py

app.py

@@ -3,8 +3,9 @@ from transformers import pipeline
 from fastapi import FastAPI
 from summarizer import summarize_via_hf
 from fastapi.middleware.cors import CORSMiddleware
+import os
 
-# 1. Модель суммаризации
+# 1. Модель суммаризации (ленивая загрузка)
 summarizer = None
 
 def summarize_text(text: str):
@@ -16,7 +17,7 @@ def summarize_text(text: str):
     return result[0]["summary_text"]
 
 # 2. FastAPI-приложение (для REST API)
-app = FastAPI()
+app = FastAPI(title="Eroha Summarizer PRO")
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -26,10 +27,20 @@ app.add_middleware(
 
 @app.post("/api/summarize")
 async def summarize_api(data: dict):
-    """REST API для суммаризации"""
+    """REST API для локальной суммаризации"""
     text = data.get("text", "")
     return {"summary": summarize_text(text)}
 
+@app.post("/pipeline/analyze_text")
+async def analyze_text(data: dict):
+    """
+    Этот endpoint принимает текст от пользователя
+    и возвращает краткое резюме через Eroha Summarizer PRO.
+    """
+    text = data.get("text", "")
+    summary = summarize_via_hf(text)
+    return {"summary": summary}
+
 # 3. Gradio интерфейс (веб-интерфейс)
 iface = gr.Interface(
     fn=summarize_text,
@@ -39,16 +50,11 @@ iface = gr.Interface(
     description="AI-инструмент для анализа и суммаризации текста (рус/англ)."
 )
 
-# 4. Запуск сервера (чтобы контейнер Hugging Face не завершался)
+# 4. Автоматический выбор режима
 if __name__ == "__main__":
+    # Если запущено вручную — Gradio
     iface.launch(server_name="0.0.0.0", server_port=7860, share=False)
+else:
+    # Если внутри контейнера Hugging Face — FastAPI
+    app
 
-@app.post("/pipeline/analyze_text")
-async def analyze_text(data: dict):
-    """
-    Этот endpoint принимает текст от пользователя
-    и возвращает краткое резюме через Eroha Summarizer PRO.
-    """
-    text = data.get("text", "")
-    summary = summarize_via_hf(text)
-    return {"summary": summary}