Update app.py

app.py

@@ -5,24 +5,23 @@ import pandas as pd
 from fpdf import FPDF
 from datetime import datetime
 
-# 🎯 مدل عمومی و رایگان برای خلاصه‌سازی فارسی
-# با استفاده از mt5-small و prompt فارسی
+# 🎯 مدل عمومی سبک خلاصه‌سازی فارسی
 summarizer = pipeline(
     "text2text-generation",
-    model="google/mt5-small",
-    tokenizer="google/mt5-small"
+    model="persiannlp/mt5-small-parsinlu-summarization"
 )
 
+# تابع خلاصه‌سازی متن
 def summarize_text(text):
     if not text.strip():
         return "⚠️ لطفاً متن وارد کنید."
-    prompt = f"لطفاً این متن را به صورت خلاصه بیان کن:\n{text}"
-    result = summarizer(prompt, max_length=150, min_length=30, do_sample=False)
+    result = summarizer(text, max_length=150, min_length=30, do_sample=False)
     return result[0]["generated_text"]
 
-def summarize_pdf(file):
+# تابع خلاصه‌سازی PDF
+def summarize_pdf(file_path):
     try:
-        reader = PyPDF2.PdfReader(file.name)
+        reader = PyPDF2.PdfReader(file_path)
         text = ""
         for page in reader.pages:
             txt = page.extract_text()
@@ -32,6 +31,7 @@ def summarize_pdf(file):
     except Exception as e:
         return f"❌ خطا در خواندن PDF: {e}"
 
+# ذخیره به PDF
 def save_to_pdf(text, summary):
     filename = f"summary_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
     pdf = FPDF()
@@ -41,12 +41,14 @@ def save_to_pdf(text, summary):
     pdf.output(filename)
     return filename
 
+# ذخیره به Excel
 def save_to_excel(text, summary):
     filename = f"summary_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx"
     df = pd.DataFrame({"متن اصلی": [text], "خلاصه": [summary]})
     df.to_excel(filename, index=False)
     return filename
 
+# 🎨 رابط کاربری
 with gr.Blocks(css="""
     body { font-family: Vazir, sans-serif; background: #f9fafb; }
     h1 { font-weight: bold; color: white; text-align: center; padding: 20px;
@@ -56,8 +58,7 @@ with gr.Blocks(css="""
     button { border-radius: 8px !important; font-weight: bold; }
 """) as demo:
 
-    with gr.Row():
-        gr.Markdown("<h1>📝 SummarizeX — خلاصه‌ساز متن و PDF (نسخه عمومی)</h1>")
+    gr.Markdown("<h1>📝 SummarizeX — خلاصه‌ساز متن و PDF (نسخه عمومی سبک)</h1>")
 
     with gr.Tab("خلاصه متن"):
         text_input = gr.Textbox(lines=10, placeholder="متن خود را اینجا وارد کنید...")
@@ -74,7 +75,7 @@ with gr.Blocks(css="""
         excel_btn.click(lambda t, s: save_to_excel(t, s), inputs=[text_input, summary_output], outputs=file_excel_out)
 
     with gr.Tab("خلاصه PDF"):
-        pdf_input = gr.File(type="file", file_types=[".pdf"])
+        pdf_input = gr.File(type="filepath", file_types=[".pdf"], label="انتخاب فایل PDF")
         pdf_summary_output = gr.Textbox(lines=8, label="خلاصه PDF")
         btn_pdf_summary = gr.Button("✨ خلاصه PDF")