Update app.py

app.py

@@ -210,33 +210,31 @@ def page_1():
             </text>
         </svg>
     """, unsafe_allow_html=True)
-    
     uploaded_file = st.file_uploader("Upload a PDF file", type=["pdf"])
     if uploaded_file:
         pdf_path = f"/tmp/{uploaded_file.name}"
         with open(pdf_path, "wb") as f:
             f.write(uploaded_file.getbuffer())
 
-        # Normal spinner with Streamlit
+        # Display a spinner while processing
         with st.spinner("Processing PDF... Please wait."):
-            time.sleep(2)  # Simulate processing time
-
-        try:
-            # Step 1: Process images, texts, and videos
-            st.markdown(
-                "<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.")
-        except Exception as e:
-            st.error(f"Error processing PDF: {e}")
+            try:
+                # Simulate processing stages with a delay (this is just an example)
+                time.sleep(1)  # Simulate a step in the processing
+
+                # Step 1: Process images, texts, and videos
+                st.text("Extracting content from PDF...")
+                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 a delay for finalizing (if needed)
+                time.sleep(1)  # Simulate final step
+
+                st.success("PDF processed successfully! Collections saved to session state.")
+            except Exception as e:
+                st.error(f"Error processing PDF: {e}")
 
 def page_2():
     add_background_image("bg2.jpg")