Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,5 @@
 import os
-os.environ["TRANSFORMERS_CACHE"] = "/cache"  # Special directory in Spaces
+os.environ["TRANSFORMERS_CACHE"] = "/cache"  # Hugging Face Spaces cache directory
 
 import streamlit as st
 import fitz  # PyMuPDF
@@ -15,13 +15,13 @@ st.set_page_config(
     }
 )
 
-# Load summarizer model (using a smaller, faster model)
+# Load summarizer model with error handling
 @st.cache_resource
 def load_summarizer():
     try:
         return pipeline(
             "summarization", 
-            model="Falconsai/text_summarization",  # Smaller model
+            model="sshleifer/distilbart-cnn-12-6",
             device=-1  # Use CPU (more reliable in Spaces)
         )
     except Exception as e:
@@ -46,21 +46,21 @@ def extract_text_from_pdf(uploaded_file):
         return ""
 
 # Summarize text in chunks
-def summarize_text(text, summarizer, max_chunk=1024):
-    if not text:
+def summarize_text(text, summarizer, max_chunk_length=2000):
+    if not text or not summarizer:
         return ""
         
     try:
-        chunks = [text[i:i+max_chunk] for i in range(0, len(text), max_chunk)]
+        chunks = [text[i:i+max_chunk_length] for i in range(0, len(text), max_chunk_length)]
         summary = ""
         for chunk in chunks:
             result = summarizer(
                 chunk,
-                max_length=150,
+                max_length=130,
                 min_length=30,
                 do_sample=False
             )
-            summary += result[0]['summary_text'] + " "
+            summary += result[0]['summary_text'] + "\n"
         return summary.strip()
     except Exception as e:
         st.error(f"❌ Summarization failed: {str(e)}")
@@ -71,36 +71,35 @@ summarizer = load_summarizer()
 
 # UI Layout
 st.title("📘 PrepPal - Study Assistant")
-st.markdown("Upload your notes and get an AI-powered summary")
-
-tab1, tab2 = st.tabs(["📄 Summarize Notes", "💬 Feedback"])
+tab1, tab2, tab3 = st.tabs(["📄 Summarize Notes", "❓ Ask a Doubt", "💬 Feedback"])
 
 with tab1:
     st.header("PDF Summarizer")
-    uploaded_file = st.file_uploader(
+    st.write("Upload your class notes in PDF format to receive a summarized version.")
+    
+    uploaded_pdf = st.file_uploader(
         "Choose a PDF file (max 5MB)",
         type=["pdf"],
         accept_multiple_files=False
     )
 
-    if uploaded_file and summarizer:
+    if uploaded_pdf and summarizer:
         with st.spinner("Extracting text..."):
-            text = extract_text_from_pdf(uploaded_file)
+            pdf_text = extract_text_from_pdf(uploaded_pdf)
 
-        if text:
+        if pdf_text:
             st.subheader("Extracted Text Preview")
-            st.text_area("", text[:500] + "...", height=150, disabled=True)
+            st.text_area("", pdf_text[:1000] + "...", height=200, disabled=True)
 
-            if st.button("Generate Summary"):
-                with st.spinner("Summarizing..."):
-                    summary = summarize_text(text, summarizer)
+            if st.button("✂️ Summarize"):
+                with st.spinner("Summarizing... Please wait."):
+                    summary = summarize_text(pdf_text, summarizer)
                 
                 if summary:
-                    st.subheader("AI Summary")
-                    st.write(summary)
-                    
+                    st.subheader("✅ Summary")
+                    st.text_area("Summary Output", summary, height=300)
                     st.download_button(
-                        "Download Summary",
+                        "⬇️ Download Summary",
                         data=summary,
                         file_name="summary.txt",
                         mime="text/plain"
@@ -109,12 +108,15 @@ with tab1:
                     st.warning("No summary generated")
 
 with tab2:
-    st.header("Feedback")
-    st.write("We'd love to hear your thoughts!")
-    feedback = st.text_area("Your feedback")
+    st.header("Ask Questions About Your Notes")
+    st.info("🔧 This feature is coming soon! You'll be able to ask questions about your uploaded notes.")
+
+with tab3:
+    st.header("Help Improve PrepPal")
+    feedback = st.text_area("Your feedback or suggestions")
     if st.button("Submit Feedback"):
-        st.success("Thank you! Your feedback has been recorded.")
+        st.success("Thank you for your feedback! We'll use it to improve PrepPal.")
 
-# Add footer
+# Footer
 st.markdown("---")
 st.caption("PrepPal v1.0 | AI-powered study assistant")