Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,18 +5,15 @@ import streamlit as st
 import fitz  # PyMuPDF
 from transformers import pipeline
 
-# Page configuration
-st.set_page_config(page_title="StudyGenie 📘", layout="centered")
+# Set page config
+st.set_page_config(page_title="PrepPal", page_icon="📘", layout="wide")
 
-# 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)
+# Load summarizer model
+@st.cache_resource
 def load_summarizer():
     return pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 
+
 # Extract text from uploaded PDF
 def extract_text_from_pdf(uploaded_file):
     text = ""
@@ -25,7 +22,7 @@ def extract_text_from_pdf(uploaded_file):
             for page in doc:
                 text += page.get_text()
     except Exception as e:
-        st.error(f"❌ Error reading PDF: {e}")
+        st.error(f"❌ Error extracting text from PDF: {e}")
     return text
 
 # Summarize large text
@@ -37,38 +34,38 @@ def summarize_text(text, summarizer, max_chunk_length=2000):
         summary += result[0]['summary_text'] + "\n"
     return summary.strip()
 
-# Tabs for multiple functions
+# Load model
+summarizer = load_summarizer()
+
+# UI
 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 your PDF..."):
+        with st.spinner("Extracting text from PDF..."):
             pdf_text = extract_text_from_pdf(uploaded_pdf)
 
         if pdf_text.strip():
-            st.success("✅ Text extracted successfully!")
-            st.subheader("🔹 Preview of Extracted Text")
-            st.text_area("Extracted Text", pdf_text[:1000] + "...", height=200)
+            st.subheader("📘 Extracted Text (Preview)")
+            st.text_area("Raw Text", pdf_text[:1000] + "...", height=200)
 
-            if st.button("✂️ Summarize Notes"):
-                with st.spinner("🧠 Summarizing..."):
-                    summarizer = load_summarizer()
+            if st.button("✂️ Summarize"):
+                with st.spinner("Summarizing... Please wait."):
                     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 coming soon! You’ll be able to chat with your notes using AI.")
+    st.info("🔧 This feature is under development. You’ll soon 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 help us improve StudyGenie.")
+    st.info("📬 A feedback form will be added here to collect your thoughts and improve PrepPal.")