Update app.py

app.py

@@ -5,23 +5,20 @@ import pandas as pd
 from fpdf import FPDF
 from datetime import datetime
 
-# رنگ‌های برند
-PRIMARY_GRADIENT = "linear-gradient(90deg, #6a4cff, #00c9b7)"
-HEADER_COLOR = "#1f2937"
-TEXT_COLOR = "#111827"
-
-# مدل خلاصه‌سازی
+# 🎯 مدل عمومی و رایگان برای خلاصه‌سازی فارسی
+# با استفاده از mt5-small و prompt فارسی
 summarizer = pipeline(
-    "summarization",
-    model="m3hrdadfi/mt5-small-parsinlu-summarization-fa"
+    "text2text-generation",
+    model="google/mt5-small",
+    tokenizer="google/mt5-small"
 )
 
-# --- توابع ---
 def summarize_text(text):
     if not text.strip():
         return "⚠️ لطفاً متن وارد کنید."
-    result = summarizer(text, max_length=150, min_length=30)
-    return result[0]["summary_text"]
+    prompt = f"لطفاً این متن را به صورت خلاصه بیان کن:\n{text}"
+    result = summarizer(prompt, max_length=150, min_length=30, do_sample=False)
+    return result[0]["generated_text"]
 
 def summarize_pdf(file):
     try:
@@ -50,7 +47,6 @@ def save_to_excel(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;
@@ -61,37 +57,31 @@ with gr.Blocks(css="""
 """) as demo:
 
     with gr.Row():
-        gr.Markdown("<h1>📝 SummarizeX — خلاصه‌ساز متن و PDF</h1>")
+        gr.Markdown("<h1>📝 SummarizeX — خلاصه‌ساز متن و PDF (نسخه عمومی)</h1>")
 
-    with gr.Tab("خلاصه متن") as tab1:
-        with gr.Row(elem_classes="tab"):
-            text_input = gr.Textbox(lines=10, placeholder="متن خود را اینجا وارد کنید...")
-            summary_output = gr.Textbox(lines=8, label="خلاصه متن")
-        with gr.Row():
-            btn_summary = gr.Button("✨ خلاصه کن", elem_id="btn-summary")
-        with gr.Row():
-            pdf_btn = gr.Button("📄 ذخیره به PDF")
-            excel_btn = gr.Button("📊 ذخیره به Excel")
-        with gr.Row():
-            file_pdf_out = gr.File(label="دانلود PDF")
-            file_excel_out = gr.File(label="دانلود Excel")
+    with gr.Tab("خلاصه متن"):
+        text_input = gr.Textbox(lines=10, placeholder="متن خود را اینجا وارد کنید...")
+        summary_output = gr.Textbox(lines=8, label="خلاصه متن")
+        btn_summary = gr.Button("✨ خلاصه کن")
+        
+        pdf_btn = gr.Button("📄 ذخیره به PDF")
+        excel_btn = gr.Button("📊 ذخیره به Excel")
+        file_pdf_out = gr.File(label="دانلود PDF")
+        file_excel_out = gr.File(label="دانلود Excel")
 
         btn_summary.click(summarize_text, inputs=text_input, outputs=summary_output)
         pdf_btn.click(lambda t, s: save_to_pdf(t, s), inputs=[text_input, summary_output], outputs=file_pdf_out)
         excel_btn.click(lambda t, s: save_to_excel(t, s), inputs=[text_input, summary_output], outputs=file_excel_out)
 
-    with gr.Tab("خلاصه PDF") as tab2:
-        with gr.Row(elem_classes="tab"):
-            pdf_input = gr.File(type="file", file_types=[".pdf"], label="انتخاب فایل PDF")
-            pdf_summary_output = gr.Textbox(lines=8, label="خلاصه PDF")
-        with gr.Row():
-            btn_pdf_summary = gr.Button("✨ خلاصه PDF")
-        with gr.Row():
-            pdf_btn2 = gr.Button("📄 ذخیره به PDF")
-            excel_btn2 = gr.Button("📊 ذخیره به Excel")
-        with gr.Row():
-            file_pdf_out2 = gr.File(label="دانلود PDF")
-            file_excel_out2 = gr.File(label="دانلود Excel")
+    with gr.Tab("خلاصه PDF"):
+        pdf_input = gr.File(type="file", file_types=[".pdf"])
+        pdf_summary_output = gr.Textbox(lines=8, label="خلاصه PDF")
+        btn_pdf_summary = gr.Button("✨ خلاصه PDF")
+        
+        pdf_btn2 = gr.Button("📄 ذخیره به PDF")
+        excel_btn2 = gr.Button("📊 ذخیره به Excel")
+        file_pdf_out2 = gr.File(label="دانلود PDF")
+        file_excel_out2 = gr.File(label="دانلود Excel")
 
         btn_pdf_summary.click(summarize_pdf, inputs=pdf_input, outputs=pdf_summary_output)
         pdf_btn2.click(lambda f, s: save_to_pdf("PDF File", s), inputs=[pdf_input, pdf_summary_output], outputs=file_pdf_out2)