Update src/streamlit_app.py

src/streamlit_app.py

@@ -17,7 +17,7 @@ st.set_page_config(
 def load_model():
     return pipeline(
         "summarization",
-        model="Falconsai/text_summarization",  # Small reliable model
+        model="Falconsai/text_summarization",
         device=-1  # CPU only
     )
 
@@ -49,28 +49,29 @@ def main():
         
         if uploaded_file and uploaded_file.size <= 5_000_000:
             if st.button("Generate Summary"):
-                with st.spinner("Processing..."):
-                    try:
-                        text = get_pdf_text(uploaded_file)
-                        if text:
-                            model = load_model()
-                            summary = create_summary(text, model)
-                            st.subheader("Summary")
-                            st.write(summary)
-                    except:
-                        st.info("Completed processing")
+                with st.spinner("Extracting text..."):
+                    text = get_pdf_text(uploaded_file)
+                
+                if text:  # Only proceed if text extraction succeeded
+                    with st.spinner("Loading AI model..."):
+                        model = load_model()
+                    
+                    with st.spinner("Generating summary..."):
+                        summary = create_summary(text, model)
+                    
+                    st.subheader("Summary")
+                    st.write(summary)
+                else:
+                    st.info("Could not extract text from PDF")
     
     with tab2:
         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"):
+        if st.button("Submit Feedback"):
             st.success("Thank you for your feedback!")
 
 if __name__ == "__main__":