Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 # --- Ленивая загрузка модели ---
 summarizer = None
 
-def get_summarizer():
+def load_model():
     global summarizer
     if summarizer is None:
         summarizer = pipeline(
@@ -15,47 +15,69 @@ def get_summarizer():
         )
     return summarizer
 
-# --- Чтение текста из файла ---
-def read_text_from_file(file):
-    if file is None:
-        return ""
+# --- Безопасное чтение PDF ---
+def extract_pdf_text(path):
+    text = ""
+    try:
+        with pdfplumber.open(path) as pdf:
+            for page in pdf.pages:
+                chunk = page.extract_text()
+                if chunk:
+                    text += chunk + "\n"
+    except Exception as e:
+        return "", f"Ошибка при чтении PDF: {e}"
+    return text, None
 
-    filepath = file   # Gradio отдаёт именно путь к временному файлу
-    filepath_lower = filepath.lower()
+# --- Универсальное чтение файла ---
+def read_file(path):
+    if not path:
+        return "", "Файл не передан."
+    path = str(path).strip()
 
-    # PDF
-    if filepath_lower.endswith(".pdf"):
-        text = ""
-        with pdfplumber.open(filepath) as pdf:
-            for page in pdf.pages:
-                text += page.extract_text() or ""
-        return text
+    if path.lower().endswith(".pdf"):
+        return extract_pdf_text(path)
 
-    # TXT / другие
-    with open(filepath, "r", encoding="utf-8", errors="ignore") as f:
-        return f.read()
+    try:
+        with open(path, "r", encoding="utf-8", errors="ignore") as f:
+            return f.read(), None
+    except Exception as e:
+        return "", f"Ошибка при чтении TXT: {e}"
 
 # --- Основная функция ---
-def summarize_file(file):
-    text = read_text_from_file(file)
+def summarize_file(path):
+    text, err = read_file(path)
+    if err:
+        return f"⚠️ {err}"
 
     if not text.strip():
-        return "⚠️ Не удалось извлечь текст из файла."
+        return "⚠️ Не удалось извлечь текст."
+
+    if len(text) < 80:
+        return "⚠️ Слишком мало текста для суммаризации."
 
-    if len(text) < 50:
-        return "⚠️ Слишком короткий текст для суммаризации."
+    model = load_model()
 
-    model = get_summarizer()
-    summary = model(text, max_length=200, min_length=50, do_sample=False)
-    return summary[0]["summary_text"]
+    # Ограничиваем текст для избежания OOM
+    text = text[:4000]
+
+    try:
+        summary = model(
+            text,
+            max_length=180,
+            min_length=60,
+            do_sample=False
+        )
+        return summary[0]["summary_text"]
+    except Exception as e:
+        return f"⚠️ Ошибка суммаризации: {e}"
 
-# --- Интерфейс ---
+# --- Интерфейс Gradio ---
 demo = gr.Interface(
     fn=summarize_file,
-    inputs=gr.File(label="Загрузите файл (.pdf или .txt)", type="filepath"),
+    inputs=gr.File(type="filepath", label="Загрузите файл (.pdf или .txt)"),
     outputs=gr.Textbox(label="Результат суммаризации"),
     title="Eroha Summarizer 🧠",
-    description="Загрузите документ (PDF или TXT), и модель создаст краткое резюме.",
+    description="Загрузите документ (PDF или TXT), и модель создаст краткое резюме."
 )
 
 if __name__ == "__main__":