Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,16 +1,17 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
+
 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 with safe temp cache dir
+# Load summarizer model
 @st.cache_resource
 def load_summarizer():
-    temp_cache_dir = tempfile.mkdtemp()
-    return pipeline("summarization", model="t5-small", cache_dir=temp_cache_dir)
+    return pipeline("summarization", model="t5-small")
 
 # Extract text from uploaded PDF
 def extract_text_from_pdf(uploaded_file):
@@ -23,7 +24,7 @@ def extract_text_from_pdf(uploaded_file):
         st.error(f"❌ Error extracting text from PDF: {e}")
     return text
 
-# Summarize large text in manageable chunks
+# Summarize large text
 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 = ""
@@ -32,20 +33,19 @@ def summarize_text(text, summarizer, max_chunk_length=2000):
         summary += result[0]['summary_text'] + "\n"
     return summary.strip()
 
-# Load the summarizer pipeline
+# Load model
 summarizer = load_summarizer()
 
-# Streamlit UI with tabs
+# UI
 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():
@@ -53,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)
@@ -61,12 +61,10 @@ 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.")