Update app.py

app.py

@@ -2,7 +2,6 @@ import streamlit as st
 import fitz  # PyMuPDF
 import os
 import re
-from transformers import pipeline
 
 # Configure Streamlit page
 st.set_page_config(page_title="PDF Extractor", layout="centered")
@@ -45,11 +44,11 @@ def extract_text_from_pdf(pdf_path):
 
     return text.strip()
 
-# Function to generate a summary using an AI model
-def summarize_text(text):
-    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-    summary = summarizer(text, max_length=200, min_length=50, do_sample=False)
-    return summary[0]['summary_text']
+# Function for simple text summarization (basic method)
+def simple_summarize(text):
+    sentences = text.split(". ")
+    summary = ". ".join(sentences[:5])  # Take the first 5 sentences as a simple summary
+    return summary + "..." if len(sentences) > 5 else summary
 
 # Function to extract only key points, MCQs, and important questions
 def extract_relevant_info(text):
@@ -97,7 +96,7 @@ if uploaded_file:
             # Perform selected task
             if task == "Summarize PDF":
                 st.subheader("📖 Summary")
-                summary = summarize_text(extracted_text)
+                summary = simple_summarize(extracted_text)
                 st.write(summary)
             
             elif task == "Generate MCQs, Key Points, and Important Questions":