Update app.py

app.py

@@ -6,22 +6,20 @@ import docx2txt
 from docx import Document
 from fpdf import FPDF
 from langdetect import detect
+import urllib.request
 
-# === 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):
+    print(f"🔹 Загружается модель: {model_name}")
     return pipeline(task, model=model_name, cache_dir="/app/models")
 
 summarizers = {
@@ -30,7 +28,7 @@ summarizers = {
     "kz": load_model("summarization", "csebuetnlp/mT5_multilingual_XLSum")
 }
 
-# === 4. Чтение файла ===
+# === Чтение файлов ===
 def read_file(file):
     if not file:
         return ""
@@ -48,10 +46,10 @@ def read_file(file):
         else:
             text = file.read().decode("utf-8", errors="ignore")
     except Exception as e:
-        return f"Ошибка при чтении файла: {e}"
+        return f"⚠️ Ошибка при чтении файла: {e}"
     return text.strip()
 
-# === 5. Определение языка ===
+# === Определение языка ===
 def detect_language(text):
     try:
         lang = detect(text)
@@ -64,7 +62,7 @@ def detect_language(text):
     except:
         return "en"
 
-# === 6. Суммаризация текста ===
+# === Суммаризация ===
 def summarize_text(text):
     if not text or len(text) < 50:
         return "⚠️ Недостаточно текста для анализа.", "❌", "❌", 0, 0, "❌"
@@ -73,8 +71,8 @@ def summarize_text(text):
     model = summarizers.get(lang, summarizers["en"])
     flags = {"ru": "🇷🇺 Русский", "kz": "🇰🇿 Қазақ тілі", "en": "🇬🇧 English"}
     lang_label = flags.get(lang, "🌍 Unknown")
-
     model_label = model.model.name_or_path if hasattr(model.model, "name_or_path") else "Custom"
+
     chunk_size = 2500
     overlap = 200
     summaries = []
@@ -94,7 +92,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:
@@ -113,45 +111,48 @@ def save_summary_as_pdf(summary_text):
     path = "summary.pdf"
     pdf = FPDF()
     pdf.add_page()
-    try:
-        pdf.add_font('DejaVu', '', FONT_PATH, uni=True)
-        pdf.set_font('DejaVu', '', 12)
-    except:
-        pdf.set_font("Arial", size=12)
+    pdf.add_font('DejaVu', '', FONT_PATH, uni=True)
+    pdf.set_font('DejaVu', '', 12)
     pdf.multi_cell(0, 10, summary_text)
     pdf.output(path)
     return path
 
-# === 8. Основная функция ===
+# === Главная функция ===
 def summarize_file(file):
     text = read_file(file)
-    if text.startswith("Ошибка"):
+    if text.startswith("⚠️"):
         return text, "❌", "❌", 0, 0, "❌", None, None, None
 
-    summary, lang_label, model_label, src_len, sum_len, compression = summarize_text(text)
+    with gr.Progress(track_tqdm=True) as progress:
+        progress(0, desc="🧠 Анализ текста...")
+        summary, lang_label, model_label, src_len, sum_len, compression = summarize_text(text)
+        progress(1, desc="✅ Готово!")
+
     txt_path = save_summary_as_txt(summary)
     docx_path = save_summary_as_docx(summary)
     pdf_path = save_summary_as_pdf(summary)
 
     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)"),
+    inputs=gr.File(label="📂 Загрузите документ (.pdf, .docx, .txt)"),
     outputs=[
         gr.Textbox(label="🧾 Краткое резюме"),
         gr.Textbox(label="🌍 Определённый язык"),
         gr.Textbox(label="🧠 Используемая модель"),
         gr.Number(label="📄 Длина исходного текста"),
         gr.Number(label="📝 Длина резюме"),
-        gr.Textbox(label="📉 Степень сокращения"),
-        gr.File(label="📄 Скачать TXT"),
-        gr.File(label="📘 Скачать DOCX"),
-        gr.File(label="📕 Скачать PDF"),
+        gr.Textbox(label="📉 Сжатие"),
+        gr.File(label="�� TXT"),
+        gr.File(label="📘 DOCX"),
+        gr.File(label="📕 PDF"),
     ],
     title="🧠 Eroha Summarizer PRO (автономная версия)",
-    description="🚀 Работает офлайн. Определяет язык (🇷🇺 / 🇰🇿 / 🇬🇧), создаёт краткое резюме и сохраняет в TXT, DOCX, PDF с поддержкой кириллицы.",
+    description="🚀 Определяет язык (🇷🇺 / 🇰🇿 / 🇬🇧), создаёт краткое резюме и сохраняет в TXT, DOCX, PDF.",
+    theme="soft",
+    allow_flagging="never"
 )
 
 demo.launch(server_name="0.0.0.0", server_port=7860)