Update app.py

app.py

@@ -4,18 +4,20 @@ import pdfplumber
 import docx2txt
 from docx import Document
 from fpdf import FPDF
+from langdetect import detect
 
-# Загружаем модель
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+# Модели суммаризации
+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")
+}
 
 def read_file(file):
-    """Чтение PDF, DOCX и TXT"""
     if file is None:
         return ""
-
     filename = file.name.lower()
     text = ""
-
     try:
         if filename.endswith(".pdf"):
             with pdfplumber.open(file.name) as pdf:
@@ -29,76 +31,74 @@ def read_file(file):
             text = file.read().decode("utf-8", errors="ignore")
     except Exception as e:
         return f"Ошибка при чтении файла: {str(e)}"
-
     return text.strip()
 
+def detect_language(text):
+    try:
+        lang = detect(text)
+        if lang.startswith("ru"):
+            return "ru"
+        elif lang.startswith("kk") or "қ" in text or "ә" in text or "ң" in text:
+            return "kz"
+        else:
+            return "en"
+    except:
+        return "en"
+
 def summarize_text(text):
-    """Суммаризация с безопасным делением на части"""
     if not text or len(text) < 50:
         return "⚠️ Недостаточно текста для суммаризации."
-
-    chunk_size = 2500  # безопасная длина
-    overlap = 200       # перекрытие между кусками
+    lang = detect_language(text)
+    model = summarizers.get(lang, summarizers["en"])
+    chunk_size = 2500
+    overlap = 200
     summaries = []
-
     for i in range(0, len(text), chunk_size - overlap):
         chunk = text[i:i + chunk_size]
         try:
-            result = summarizer(chunk, max_length=180, min_length=40, do_sample=False)
+            result = model(chunk, max_length=180, min_length=40, do_sample=False)
             summaries.append(result[0]['summary_text'])
         except Exception as e:
-            summaries.append(f"[Ошибка в части {len(summaries) + 1}: {str(e)}]")
-
-    final_summary = "\n\n".join(summaries)
-    return final_summary.strip()
+            summaries.append(f"[Ошибка в части {len(summaries)+1}: {str(e)}]")
+    return "\n\n".join(summaries).strip()
 
 def save_summary_as_docx(summary_text):
-    """Сохранение в DOCX"""
+    path = "summary.docx"
     doc = Document()
     doc.add_heading("Резюме документа", level=1)
     doc.add_paragraph(summary_text)
-    path = "summary.docx"
     doc.save(path)
     return path
 
 def save_summary_as_txt(summary_text):
-    """Сохранение в TXT"""
     path = "summary.txt"
     with open(path, "w", encoding="utf-8") as f:
         f.write(summary_text)
     return path
 
 def save_summary_as_pdf(summary_text):
-    """Сохранение в PDF с русским шрифтом"""
     path = "summary.pdf"
     pdf = FPDF()
     pdf.add_page()
-
-    # Используем встроенный шрифт DejaVuSans.ttf, который нужно будет добавить в проект
     try:
         pdf.add_font('DejaVu', '', 'DejaVuSans.ttf', uni=True)
         pdf.set_font('DejaVu', '', 12)
     except:
         pdf.set_font("Arial", size=12)
-
     pdf.multi_cell(0, 10, summary_text)
     pdf.output(path)
     return path
 
 def summarize_file(file):
-    """Главная функция"""
     text = read_file(file)
     if text.startswith("Ошибка"):
         return text, None, None, None
-
     summary = summarize_text(text)
     txt_path = save_summary_as_txt(summary)
     docx_path = save_summary_as_docx(summary)
     pdf_path = save_summary_as_pdf(summary)
-
     return summary, txt_path, docx_path, pdf_path
 
-# Интерфейс Gradio
 demo = gr.Interface(
     fn=summarize_file,
     inputs=gr.File(label="Загрузите документ (.pdf, .docx или .txt)"),
@@ -109,7 +109,7 @@ demo = gr.Interface(
         gr.File(label="📕 Скачать PDF")
     ],
     title="🧠 Eroha Summarizer",
-    description="Загрузите PDF, DOCX или TXT файл. Модель создаст краткое резюме и позволит скачать результат в любом формате.",
+    description="Автоматически определяет язык документа (RU / EN / KZ) и создаёт краткое резюме."
 )
 
 demo.launch(server_name="0.0.0.0", server_port=7860)