Upload 2 files

api.py

@@ -0,0 +1,81 @@
+!pip install transformers accelerate torch PyPDF2 gradio --quiet
+
+import gradio as gr
+from transformers import pipeline
+import PyPDF2
+
+# ===== مدل‌ها =====
+# مدل خلاصه‌سازی فارسی (عمومی)
+persian_summarizer = pipeline(
+    "summarization",
+    model="m3hrdadfi/bert2bert-fa-summarization",
+    tokenizer="m3hrdadfi/bert2bert-fa-summarization"
+)
+
+# مدل خلاصه‌سازی انگلیسی (عمومی و سبک)
+english_summarizer = pipeline(
+    "summarization",
+    model="facebook/bart-large-cnn"
+)
+
+# ===== توابع =====
+def extract_text_from_pdf(pdf_file):
+    """استخراج متن از فایل PDF"""
+    pdf_reader = PyPDF2.PdfReader(pdf_file)
+    text = ""
+    for page in pdf_reader.pages:
+        page_text = page.extract_text()
+        if page_text:
+            text += page_text + "\n"
+    return text
+
+def summarize_document(language, text, pdf):
+    """خلاصه‌سازی متن یا PDF بر اساس زبان انتخابی"""
+    if pdf is not None:
+        text = extract_text_from_pdf(pdf)
+
+    if not text or len(text.strip()) < 50:
+        return "❗ لطفاً یک متن یا فایل PDF معتبر و حداقل ۵۰ کاراکتر وارد کنید."
+
+    # انتخاب مدل بر اساس زبان
+    if language == "Persian":
+        summarizer = persian_summarizer
+        max_chunk = 500
+    else:
+        summarizer = english_summarizer
+        max_chunk = 1024
+
+    # شکستن متن به بخش‌های کوچکتر
+    text = text.strip().replace("\n", " ")
+    chunks = [text[i:i+max_chunk] for i in range(0, len(text), max_chunk)]
+
+    summary = ""
+    for chunk in chunks:
+        result = summarizer(
+            chunk,
+            max_length=150,
+            min_length=40,
+            do_sample=False
+        )[0]['summary_text']
+        summary += result + "\n"
+
+    return summary.strip()
+
+# ===== رابط Gradio =====
+with gr.Blocks() as demo:
+    gr.Markdown("## 📄 SummarizeX – AI Text & PDF Summarizer 🌐")
+    gr.Markdown("خلاصه‌ساز هوش مصنوعی برای متن و PDF – پشتیبانی از فارسی و انگلیسی")
+
+    with gr.Row():
+        language = gr.Radio(["Persian", "English"], label="زبان خلاصه‌سازی", value="Persian")
+    
+    with gr.Row():
+        pdf_input = gr.File(label="آپلود PDF", file_types=[".pdf"])
+        text_input = gr.Textbox(label="یا متن را وارد کنید", lines=8, placeholder="متن را اینجا وارد کنید...")
+
+    summarize_button = gr.Button("خلاصه کن / Summarize")
+    output_box = gr.Textbox(label="خلاصه نهایی", lines=8)
+
+    summarize_button.click(summarize_document, inputs=[language, text_input, pdf_input], outputs=output_box)
+
+demo.launch(share=True)

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+transformers
+pypdf2
+torch