Update app.py

app.py

@@ -1,68 +1,80 @@
-import gradio as gr
+import streamlit as st
 from transformers import pipeline
 from PyPDF2 import PdfReader
-from docx import Document
+import docx2txt
 from fpdf import FPDF
-import os
+from io import BytesIO
+from docx import Document
 
-# Load summarization pipeline
-summarizer = pipeline("summarization")
+# Load summarizer
+summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 
-# Function to read text from different file types
-def extract_text(file):
+# Helper: Extract text from PDF
+def extract_text_from_pdf(file):
+    pdf = PdfReader(file)
     text = ""
-    if file.name.endswith(".pdf"):
-        reader = PdfReader(file.name)
-        for page in reader.pages:
-            text += page.extract_text()
-    elif file.name.endswith(".docx"):
-        doc = Document(file.name)
-        for para in doc.paragraphs:
-            text += para.text + "\n"
-    else:
-        text = file.read().decode("utf-8")
+    for page in pdf.pages:
+        text += page.extract_text()
     return text
 
-# Function to summarize and return as PDF or Word
-def summarize_file(file, output_format):
-    text = extract_text(file)
-    if not text.strip():
-        return None, "File is empty or could not be read."
-    
-    summarized = summarizer(text, max_length=150, min_length=40, do_sample=False)[0]["summary_text"]
+# Helper: Save summary to DOCX
+def generate_docx(text):
+    doc = Document()
+    doc.add_heading("Summary", 0)
+    doc.add_paragraph(text)
+    buffer = BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return buffer
+
+# Helper: Save summary to PDF
+def generate_pdf(text):
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", size=12)
+    for line in text.split("\n"):
+        pdf.multi_cell(0, 10, line)
+    buffer = BytesIO()
+    pdf.output(buffer)
+    buffer.seek(0)
+    return buffer
+
+# App UI
+st.set_page_config(page_title="Smart Text Summarizer", layout="centered")
+st.title("📄 Smart Text Summarizer")
+
+uploaded_file = st.file_uploader("Upload a PDF or Word file", type=["pdf", "docx"])
+
+if uploaded_file:
+    file_type = uploaded_file.name.split(".")[-1].lower()
     
-    output_path = "/tmp/summary_output"
-    if output_format == "PDF":
-        pdf = FPDF()
-        pdf.add_page()
-        pdf.set_auto_page_break(auto=True, margin=15)
-        pdf.set_font("Arial", size=12)
-        for line in summarized.split("\n"):
-            pdf.multi_cell(0, 10, line)
-        output_file = f"{output_path}.pdf"
-        pdf.output(output_file)
+    # Extract text
+    if file_type == "pdf":
+        text = extract_text_from_pdf(uploaded_file)
+    elif file_type == "docx":
+        text = docx2txt.process(uploaded_file)
     else:
-        doc = Document()
-        doc.add_heading("Summary", 0)
-        doc.add_paragraph(summarized)
-        output_file = f"{output_path}.docx"
-        doc.save(output_file)
+        st.error("Unsupported file format.")
+        st.stop()
+
+    if not text:
+        st.error("No readable text found in the file.")
+        st.stop()
+
+    st.subheader("🔍 Extracted Text")
+    st.text_area("Preview", text[:1000] + "...", height=200)
 
-    return output_file, "Success"
+    # Summarize
+    if st.button("Summarize"):
+        with st.spinner("Summarizing..."):
+            summary = summarizer(text, max_length=150, min_length=30, do_sample=False)[0]['summary_text']
+            st.success("Summary Generated ✅")
+            st.subheader("📝 Summary")
+            st.write(summary)
 
-# Gradio interface
-iface = gr.Interface(
-    fn=summarize_file,
-    inputs=[
-        gr.File(label="Upload a text, Word, or PDF file"),
-        gr.Radio(["PDF", "Word"], label="Choose output format")
-    ],
-    outputs=[
-        gr.File(label="Download Summary"),
-        gr.Textbox(label="Status")
-    ],
-    title="Smart Text Summarizer",
-    description="Upload a .txt, .docx, or .pdf file and get the summary back as a Word or PDF file."
-)
+            # Download options
+            docx_file = generate_docx(summary)
+            pdf_file = generate_pdf(summary)
 
-iface.launch()
+            st.download_button("📥 Download as DOCX", docx_file, file_name="summary.docx")
+            st.download_button("📥 Download as PDF", pdf_file, file_name="summary.pdf")