Update app.py

app.py

@@ -3,8 +3,9 @@ from transformers import pipeline
 import pdfplumber
 import docx2txt
 from docx import Document
+from fpdf import FPDF
 
-# Загружаем модель один раз при старте
+# Загружаем модель
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
 def read_file(file):
@@ -12,8 +13,8 @@ def read_file(file):
     if file is None:
         return ""
 
-    text = ""
     filename = file.name.lower()
+    text = ""
 
     try:
         if filename.endswith(".pdf"):
@@ -32,52 +33,70 @@ def read_file(file):
     return text.strip()
 
 def summarize_text(text):
-    """Безопасная суммаризация с разбиением текста"""
+    """Суммаризация с безопасным делением на части"""
     if not text or len(text) < 50:
         return "⚠️ Недостаточно текста для суммаризации."
 
-    # Разделим текст на куски (до 2000 символов)
-    chunk_size = 2000
-    chunks = [text[i:i + chunk_size] for i in range(0, len(text), chunk_size)]
-
+    chunk_size = 2500  # безопасная длина
+    overlap = 200       # перекрытие между кусками
     summaries = []
-    for i, chunk in enumerate(chunks):
+
+    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)
             summaries.append(result[0]['summary_text'])
         except Exception as e:
-            summaries.append(f"[Ошибка в части {i + 1}: {str(e)}]")
+            summaries.append(f"[Ошибка в части {len(summaries) + 1}: {str(e)}]")
 
-    return "\n\n".join(summaries)
+    final_summary = "\n\n".join(summaries)
+    return final_summary.strip()
 
 def save_summary_as_docx(summary_text):
-    """Сохраняем результат в DOCX"""
+    """Сохранение в DOCX"""
     doc = Document()
     doc.add_heading("Резюме документа", level=1)
     doc.add_paragraph(summary_text)
-    output_path = "summary.docx"
-    doc.save(output_path)
-    return output_path
+    path = "summary.docx"
+    doc.save(path)
+    return path
 
 def save_summary_as_txt(summary_text):
-    """Сохраняем результат в TXT"""
-    output_path = "summary.txt"
-    with open(output_path, "w", encoding="utf-8") as f:
+    """Сохранение в TXT"""
+    path = "summary.txt"
+    with open(path, "w", encoding="utf-8") as f:
         f.write(summary_text)
-    return output_path
+    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
+        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)
 
-    # Сохраняем результат в два формата
-    txt_file = save_summary_as_txt(summary)
-    docx_file = save_summary_as_docx(summary)
-
-    return summary, txt_file, docx_file
+    return summary, txt_path, docx_path, pdf_path
 
 # Интерфейс Gradio
 demo = gr.Interface(
@@ -85,11 +104,12 @@ demo = gr.Interface(
     inputs=gr.File(label="Загрузите документ (.pdf, .docx или .txt)"),
     outputs=[
         gr.Textbox(label="Краткое резюме"),
-        gr.File(label="Скачать TXT"),
-        gr.File(label="Скачать DOCX")
+        gr.File(label="📄 Скачать TXT"),
+        gr.File(label="📘 Скачать DOCX"),
+        gr.File(label="📕 Скачать PDF")
     ],
     title="🧠 Eroha Summarizer",
-    description="Загрузите PDF, Word или TXT документ. Модель создаст краткое резюме и позволит скачать его.",
+    description="Загрузите PDF, DOCX или TXT файл. Модель создаст краткое резюме и позволит скачать результат в любом формате.",
 )
 
 demo.launch(server_name="0.0.0.0", server_port=7860)