Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 from transformers import pipeline
 import pdfplumber
@@ -6,14 +7,30 @@ from docx import Document
 from fpdf import FPDF
 from langdetect import detect
 
-# === Загрузка моделей один раз при старте ===
+# === 1. Подготовка директорий ===
+os.makedirs("/app/models", exist_ok=True)
+FONT_PATH = "DejaVuSans.ttf"
+
+# === 2. Проверяем, есть ли шрифт, если нет — скачиваем ===
+if not os.path.exists(FONT_PATH):
+    import urllib.request
+    print("🪶 Загружаю шрифт DejaVuSans.ttf...")
+    urllib.request.urlretrieve(
+        "https://github.com/dejavu-fonts/dejavu-fonts/raw/master/ttf/DejaVuSans.ttf",
+        FONT_PATH
+    )
+
+# === 3. Подгружаем модели с кэшем ===
+def load_model(task, model_name):
+    return pipeline(task, model=model_name, cache_dir="/app/models")
+
 summarizers = {
-    "en": pipeline("summarization", model="facebook/bart-large-cnn"),
-    "ru": pipeline("summarization", model="IlyaGusev/mbart_ru_sum_gazeta"),
-    "kz": pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum")
+    "en": load_model("summarization", "facebook/bart-large-cnn"),
+    "ru": load_model("summarization", "IlyaGusev/mbart_ru_sum_gazeta"),
+    "kz": load_model("summarization", "csebuetnlp/mT5_multilingual_XLSum")
 }
 
-# === Чтение файла ===
+# === 4. Чтение файла ===
 def read_file(file):
     if not file:
         return ""
@@ -34,7 +51,7 @@ def read_file(file):
         return f"Ошибка при чтении файла: {e}"
     return text.strip()
 
-# === Определение языка ===
+# === 5. Определение языка ===
 def detect_language(text):
     try:
         lang = detect(text)
@@ -47,7 +64,7 @@ def detect_language(text):
     except:
         return "en"
 
-# === Суммаризация текста ===
+# === 6. Суммаризация текста ===
 def summarize_text(text):
     if not text or len(text) < 50:
         return "⚠️ Недостаточно текста для анализа.", "❌", "❌", 0, 0, "❌"
@@ -77,7 +94,7 @@ def summarize_text(text):
 
     return summary, lang_label, model_label, src_len, sum_len, f"{compression}%"
 
-# === Сохранение файлов ===
+# === 7. Сохранение файлов ===
 def save_summary_as_txt(summary_text):
     path = "summary.txt"
     with open(path, "w", encoding="utf-8") as f:
@@ -97,7 +114,7 @@ def save_summary_as_pdf(summary_text):
     pdf = FPDF()
     pdf.add_page()
     try:
-        pdf.add_font('DejaVu', '', 'DejaVuSans.ttf', uni=True)
+        pdf.add_font('DejaVu', '', FONT_PATH, uni=True)
         pdf.set_font('DejaVu', '', 12)
     except:
         pdf.set_font("Arial", size=12)
@@ -105,7 +122,7 @@ def save_summary_as_pdf(summary_text):
     pdf.output(path)
     return path
 
-# === Основная функция ===
+# === 8. Основная функция ===
 def summarize_file(file):
     text = read_file(file)
     if text.startswith("Ошибка"):
@@ -118,7 +135,7 @@ def summarize_file(file):
 
     return summary, lang_label, model_label, src_len, sum_len, compression, txt_path, docx_path, pdf_path
 
-# === Интерфейс ===
+# === 9. Интерфейс Gradio ===
 demo = gr.Interface(
     fn=summarize_file,
     inputs=gr.File(label="📁 Загрузите документ (.pdf, .docx, .txt)"),
@@ -126,15 +143,15 @@ demo = gr.Interface(
         gr.Textbox(label="🧾 Краткое резюме"),
         gr.Textbox(label="🌍 Определённый язык"),
         gr.Textbox(label="🧠 Используемая модель"),
-        gr.Number(label="📄 Длина исходного текста (символов)"),
-        gr.Number(label="📝 Длина резюме (символов)"),
+        gr.Number(label="📄 Длина исходного текста"),
+        gr.Number(label="📝 Длина резюме"),
         gr.Textbox(label="📉 Степень сокращения"),
         gr.File(label="📄 Скачать TXT"),
         gr.File(label="📘 Скачать DOCX"),
         gr.File(label="📕 Скачать PDF"),
     ],
-    title="🧠 Eroha Summarizer PRO",
-    description="⚡ Автоматически определяет язык (🇷🇺 / 🇬🇧 / 🇰🇿), создаёт краткое резюме и сохраняет в TXT, DOCX, PDF. Поддерживает кириллицу в PDF.",
+    title="🧠 Eroha Summarizer PRO (автономная версия)",
+    description="🚀 Работает офлайн. Определяет язык (🇷🇺 / 🇰🇿 / 🇬🇧), создаёт краткое резюме и сохраняет в TXT, DOCX, PDF с поддержкой кириллицы.",
 )
 
 demo.launch(server_name="0.0.0.0", server_port=7860)