Update app.py

app.py

@@ -1,63 +1,86 @@
 import gradio as gr
 from transformers import pipeline
 from PyPDF2 import PdfReader
-import docx
-from io import BytesIO
+from docx import Document
+from fpdf import FPDF
+import os
 
-# Load summarization model
+# Load the summarizer pipeline
 summarizer = pipeline("summarization")
 
-# Extract text from PDF
+# Helper to extract text from PDF
 def extract_text_from_pdf(file):
     reader = PdfReader(file)
     text = ""
     for page in reader.pages:
-        page_text = page.extract_text()
-        if page_text:
-            text += page_text
+        text += page.extract_text()
     return text
 
-# Extract text from Word document
+# Helper to extract text from DOCX
 def extract_text_from_docx(file):
-    doc = docx.Document(file)
+    doc = Document(file)
     return "\n".join([para.text for para in doc.paragraphs])
 
-# Save summary as a Word file
-def save_summary_to_word(summary_text):
-    doc = docx.Document()
-    doc.add_heading("Summary", level=1)
-    doc.add_paragraph(summary_text)
-    output = BytesIO()
-    doc.save(output)
-    output.seek(0)
-    return output
+# Helper to save summary as DOCX
+def save_as_docx(text):
+    doc = Document()
+    doc.add_heading("Summary", 0)
+    doc.add_paragraph(text)
+    path = "summary.docx"
+    doc.save(path)
+    return path
+
+# Helper to save summary as PDF
+def save_as_pdf(text):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    pdf.multi_cell(0, 10, text)
+    path = "summary.pdf"
+    pdf.output(path)
+    return path
 
 # Main function
-def summarize_file(file):
-    if file.name.endswith(".pdf"):
-        text = extract_text_from_pdf(file)
-    elif file.name.endswith(".docx"):
-        text = extract_text_from_docx(file)
+def summarize_input(text, file, export_type):
+    if file:
+        ext = os.path.splitext(file.name)[-1].lower()
+        if ext == ".pdf":
+            text = extract_text_from_pdf(file)
+        elif ext == ".docx":
+            text = extract_text_from_docx(file)
+        else:
+            return "Unsupported file type", None
+
+    if not text.strip():
+        return "No text to summarize.", None
+
+    summary = summarizer(text, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
+
+    # Save based on export type
+    if export_type == "PDF":
+        file_path = save_as_pdf(summary)
     else:
-        return "Unsupported file type", None
-
-    # Limit length for large documents
-    if len(text) > 3000:
-        text = text[:3000]
-
-    summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
-    summary_text = summary[0]['summary_text']
-    summary_doc = save_summary_to_word(summary_text)
-    return summary_text, ("summary.docx", summary_doc)
-
-# Gradio interface
-gr.Interface(
-    fn=summarize_file,
-    inputs=gr.File(file_types=[".pdf", ".docx"], label="Upload PDF or Word Document"),
-    outputs=[
-        gr.Textbox(label="Summary"),
-        gr.File(label="Download Summary as Word File")
-    ],
-    title="📄 Smart Document Summarizer",
-    description="Upload a PDF or Word file. This app will summarize it using AI and give you a downloadable summary."
-).launch()
+        file_path = save_as_docx(summary)
+
+    return summary, file_path
+
+# UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 📝 Smart Text Summarizer\nUpload a file or enter text. Choose output format and get your summary!")
+    
+    with gr.Row():
+        text_input = gr.Textbox(label="Enter Text", lines=8, placeholder="Or upload a file below instead...")
+        file_input = gr.File(label="Upload PDF or DOCX", file_types=[".pdf", ".docx"])
+    
+    export_type = gr.Radio(["PDF", "Word (DOCX)"], label="Export As", value="PDF")
+    summarize_btn = gr.Button("Summarize")
+
+    output_text = gr.Textbox(label="Summary", lines=8)
+    download_link = gr.File(label="Download Summary")
+
+    summarize_btn.click(fn=summarize_input,
+                        inputs=[text_input, file_input, export_type],
+                        outputs=[output_text, download_link])
+
+# Launch the app
+demo.launch()