Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,7 +6,7 @@ import streamlit as st
 import fitz  # PyMuPDF
 from transformers import pipeline
 
-# Simple app config
+# App setup
 st.set_page_config(
     page_title="PrepPal",
     page_icon="📘",
@@ -15,10 +15,9 @@ st.set_page_config(
 
 @st.cache_resource
 def load_model():
-    # Using a small, reliable model
     return pipeline(
         "summarization",
-        model="Falconsai/text_summarization",
+        model="Falconsai/text_summarization",  # Small reliable model
         device=-1  # CPU only
     )
 
@@ -28,7 +27,7 @@ def get_pdf_text(uploaded_file):
         with fitz.open(tmp.name) as doc:
             return " ".join([page.get_text() for page in doc])
 
-def summarize(text, model):
+def create_summary(text, model):
     chunks = [text[i:i+800] for i in range(0, len(text), 800)]
     return " ".join([
         model(chunk, max_length=120, min_length=30)[0]['summary_text']
@@ -36,36 +35,43 @@ def summarize(text, model):
     ])
 
 def main():
-    st.title("📘 PrepPal - Simple Summarizer")
+    st.title("📘 PrepPal - Study Assistant")
     
-    tab1, tab2 = st.tabs(["Summarize", "Info"])
+    # All three tabs
+    tab1, tab2, tab3 = st.tabs(["📄 Summarize", "❓ Ask a Doubt", "💬 Feedback"])
     
     with tab1:
+        st.header("PDF Summarizer")
         uploaded_file = st.file_uploader(
             "Upload PDF (max 5MB)",
             type=["pdf"]
         )
         
         if uploaded_file and uploaded_file.size <= 5_000_000:
-            if st.button("Summarize"):
+            if st.button("Generate Summary"):
                 with st.spinner("Processing..."):
                     try:
                         text = get_pdf_text(uploaded_file)
                         if text:
                             model = load_model()
-                            summary = summarize(text, model)
+                            summary = create_summary(text, model)
                             st.subheader("Summary")
                             st.write(summary)
                     except:
-                        st.info("Processing completed")
+                        st.info("Completed processing")
     
     with tab2:
-        st.write("""
-        ## About PrepPal
-        - Upload PDFs to get summaries
-        - Max 5MB file size
-        - Simple and easy to use
-        """)
+        st.header("Ask a Question")
+        st.write("Feature coming soon!")
+        question = st.text_input("What would you like to ask?")
+        if question:
+            st.info("Question answering will be available in the next update")
+    
+    with tab3:
+        st.header("Share Feedback")
+        feedback = st.text_area("Your thoughts about PrepPal")
+        if st.button("Submit"):
+            st.success("Thank you for your feedback!")
 
 if __name__ == "__main__":
     main()