Update app.py

app.py

@@ -227,6 +227,10 @@ def page_1():
                 "<p style='color: Black; font-weight: bold;'>Extracting content from PDF...</p>",
                 unsafe_allow_html=True
             )
+            image_collection, text_collection, video_collection = process_pdf(pdf_path)
+            st.session_state.image_collection = image_collection
+            st.session_state.text_collection = text_collection
+            st.session_state.video_collection = video_collection
             # Simulate processing function
             time.sleep(1)  # Simulate final step
             st.success("PDF processed successfully! Collections saved to session state.")