Update app.py

app.py

@@ -1,16 +1,25 @@
 import gradio as gr
 from transformers import pipeline
 import PyPDF2
+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"
+
+# مدل خلاصه‌سازی
 summarizer = pipeline(
     "summarization",
     model="m3hrdadfi/mt5-small-parsinlu-summarization-fa"
 )
 
+# --- توابع ---
 def summarize_text(text):
     if not text.strip():
-        return "متنی وارد نشده است."
+        return "⚠️ لطفاً متن وارد کنید."
     result = summarizer(text, max_length=150, min_length=30)
     return result[0]["summary_text"]
 
@@ -19,25 +28,74 @@ def summarize_pdf(file):
         reader = PyPDF2.PdfReader(file.name)
         text = ""
         for page in reader.pages:
-            text += page.extract_text() + "\n"
+            txt = page.extract_text()
+            if txt:
+                text += txt + "\n"
         return summarize_text(text)
     except Exception as e:
-        return f"خطا در خواندن PDF: {e}"
-
-with gr.Blocks() as demo:
-    gr.Markdown("# 📝 SummarizeX — خلاصه‌ساز متن و PDF")
-
-    with gr.Tab("خلاصه متن"):
-        input_text = gr.Textbox(lines=12, placeholder="متن را اینجا وارد کنید...")
-        output_summary = gr.Textbox(lines=8, label="خلاصه متن")
-        btn1 = gr.Button("خلاصه کن")
-        btn1.click(summarize_text, inputs=input_text, outputs=output_summary)
-
-    with gr.Tab("خلاصه PDF"):
-        pdf_input = gr.File(type="file", file_types=[".pdf"])
-        pdf_output = gr.Textbox(lines=8, label="خلاصه PDF")
-        btn2 = gr.Button("خلاصه PDF")
-        btn2.click(summarize_pdf, inputs=pdf_input, outputs=pdf_output)
+        return f"❌ خطا در خواندن PDF: {e}"
+
+def save_to_pdf(text, summary):
+    filename = f"summary_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    pdf.multi_cell(0, 10, f"📄 متن اصلی:\n\n{text}\n\n---\n\n📝 خلاصه:\n\n{summary}")
+    pdf.output(filename)
+    return filename
+
+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;
+         background: linear-gradient(90deg, #6a4cff, #00c9b7); border-radius: 8px; }
+    .tab { background-color: white; padding: 20px; border-radius: 8px;
+           box-shadow: 0px 2px 8px rgba(0,0,0,0.05); }
+    button { border-radius: 8px !important; font-weight: bold; }
+""") as demo:
+
+    with gr.Row():
+        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")
+
+        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")
+
+        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)
+        excel_btn2.click(lambda f, s: save_to_excel("PDF File", s), inputs=[pdf_input, pdf_summary_output], outputs=file_excel_out2)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)