Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,14 +5,17 @@ import streamlit as st
 import fitz  # PyMuPDF
 from transformers import pipeline
 
-# Set page config
-st.set_page_config(page_title="PrepPal", page_icon="📘", layout="wide")
+# Page configuration
+st.set_page_config(page_title="StudyGenie 📘", layout="centered")
 
-# Load summarizer model
-@st.cache_resource
-def load_summarizer():
-    return pipeline("summarization", model="Falconsai/text_summarization")
+# Title
+st.title("📄 StudyGenie - Summarize Your Notes")
+st.write("Upload your PDF notes below and get a smart AI-generated summary instantly.")
 
+# Load model only when needed
+@st.cache_resource(show_spinner=False)
+def load_summarizer():
+    return pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 
 # Extract text from uploaded PDF
 def extract_text_from_pdf(uploaded_file):
@@ -22,7 +25,7 @@ def extract_text_from_pdf(uploaded_file):
             for page in doc:
                 text += page.get_text()
     except Exception as e:
-        st.error(f"❌ Error extracting text from PDF: {e}")
+        st.error(f"❌ Error reading PDF: {e}")
     return text
 
 # Summarize large text
@@ -34,38 +37,38 @@ def summarize_text(text, summarizer, max_chunk_length=2000):
         summary += result[0]['summary_text'] + "\n"
     return summary.strip()
 
-# Load model
-summarizer = load_summarizer()
-
-# UI
+# Tabs for multiple functions
 tab1, tab2, tab3 = st.tabs(["📄 Summarize Notes", "❓ Ask a Doubt", "💬 Feedback"])
 
+# Tab 1: Summarize
 with tab1:
-    st.header("📄 Upload Notes & Get Summary")
-    st.write("Upload your class notes in PDF format to receive a summarized version.")
     uploaded_pdf = st.file_uploader("Upload your PDF notes (PDF only)", type=["pdf"])
 
     if uploaded_pdf:
-        with st.spinner("Extracting text from PDF..."):
+        with st.spinner("🔍 Extracting text from your PDF..."):
             pdf_text = extract_text_from_pdf(uploaded_pdf)
 
         if pdf_text.strip():
-            st.subheader("📘 Extracted Text (Preview)")
-            st.text_area("Raw Text", pdf_text[:1000] + "...", height=200)
+            st.success("✅ Text extracted successfully!")
+            st.subheader("🔹 Preview of Extracted Text")
+            st.text_area("Extracted Text", pdf_text[:1000] + "...", height=200)
 
-            if st.button("✂️ Summarize"):
-                with st.spinner("Summarizing... Please wait."):
+            if st.button("✂️ Summarize Notes"):
+                with st.spinner("🧠 Summarizing..."):
+                    summarizer = load_summarizer()
                     summary = summarize_text(pdf_text, summarizer)
-                st.subheader("✅ Summary")
+                st.subheader("📘 Summary")
                 st.text_area("Summary Output", summary, height=300)
                 st.download_button("⬇️ Download Summary", summary, file_name="summary.txt")
         else:
             st.warning("⚠️ No text found in the uploaded PDF.")
 
+# Tab 2: Ask a Doubt
 with tab2:
     st.header("❓ Ask a Doubt")
-    st.info("🔧 This feature is under development. You’ll soon be able to chat with your notes using AI!")
+    st.info("🔧 This feature is coming soon! You’ll be able to chat with your notes using AI.")
 
+# Tab 3: Feedback
 with tab3:
     st.header("💬 User Feedback")
-    st.info("📬 A feedback form will be added here to collect your thoughts and improve PrepPal.")
+    st.info("📬 A feedback form will be added here to help us improve StudyGenie.")