Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,19 +1,18 @@
-import os
-os.environ["TRANSFORMERS_CACHE"] = "/app/.cache"
-
 import streamlit as st
 import fitz  # PyMuPDF
 from transformers import pipeline
+import tempfile
 
 # Set page config
 st.set_page_config(page_title="PrepPal", page_icon="📘", layout="wide")
 
-# Load summarizer model
+# Load summarizer model with safe temp cache dir
 @st.cache_resource
 def load_summarizer():
-    return pipeline("summarization", model="t5-small")
+    temp_cache_dir = tempfile.mkdtemp()
+    return pipeline("summarization", model="t5-small", cache_dir=temp_cache_dir)
 
-# PDF text extraction
+# Extract text from uploaded PDF
 def extract_text_from_pdf(uploaded_file):
     text = ""
     try:
@@ -21,10 +20,10 @@ 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 extracting text from PDF: {e}")
     return text
 
-# Summarize text in chunks
+# Summarize large text in manageable chunks
 def summarize_text(text, summarizer, max_chunk_length=2000):
     chunks = [text[i:i + max_chunk_length] for i in range(0, len(text), max_chunk_length)]
     summary = ""
@@ -33,17 +32,20 @@ def summarize_text(text, summarizer, max_chunk_length=2000):
         summary += result[0]['summary_text'] + "\n"
     return summary.strip()
 
+# Load the summarizer pipeline
 summarizer = load_summarizer()
 
+# Streamlit UI with tabs
 tab1, tab2, tab3 = st.tabs(["📄 Summarize Notes", "❓ Ask a Doubt", "💬 Feedback"])
 
+# Tab 1: Upload and summarize notes
 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 PDF..."):
             pdf_text = extract_text_from_pdf(uploaded_pdf)
 
         if pdf_text.strip():
@@ -51,7 +53,7 @@ with tab1:
             st.text_area("Raw Text", pdf_text[:1000] + "...", height=200)
 
             if st.button("✂️ Summarize"):
-                with st.spinner("Summarizing... Please wait."):
+                with st.spinner("🤖 Summarizing... Please wait."):
                     summary = summarize_text(pdf_text, summarizer)
                 st.subheader("✅ Summary")
                 st.text_area("Summary Output", summary, height=300)
@@ -59,10 +61,12 @@ with tab1:
         else:
             st.warning("⚠️ No text found in the uploaded PDF.")
 
+# Tab 2: Ask a doubt (placeholder)
 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!")
 
+# Tab 3: Feedback (placeholder)
 with tab3:
     st.header("💬 User Feedback")
     st.info("📬 A feedback form will be added here to collect your thoughts and improve PrepPal.")