Update app.py

app.py

@@ -272,39 +272,39 @@ if __name__ == "__main__":
         # File handling
         pdf_path = os.path.join(tempfile.gettempdir(), f"{patent_number}.pdf")
         if not os.path.isfile(pdf_path):
-            st.write("📥 Downloading patent file...")
-            try:
-                pdf_path = download_pdf(patent_number)
-                st.write(f"✅ File downloaded: {pdf_path}")
-            except Exception as e:
-                st.error(f"Failed to download patent: {e}")
-                st.stop()
+            with st.spinner("📥 Downloading patent file..."):
+                try:
+                    pdf_path = download_pdf(patent_number)
+                    st.write(f"✅ File downloaded: {pdf_path}")
+                except Exception as e:
+                    st.error(f"Failed to download patent: {e}")
+                    st.stop()
         else:
             st.write("✅ File already downloaded.")
 
         # Generate PDF preview only if not already displayed
         if not st.session_state.get("pdf_preview_displayed", False):
-            st.write("🖼️ Generating PDF preview...")
-            preview_image_path = preview_pdf(pdf_path, scale_factor=0.5)  # Added scale_factor
-            if preview_image_path:
-                st.session_state.pdf_preview = preview_image_path
-                st.image(preview_image_path, caption="First Page Preview", use_container_width=False)
-                st.session_state["pdf_preview_displayed"] = True
-            else:
-                st.warning("Failed to generate PDF preview.")
-                st.session_state.pdf_preview = None
+            with st.spinner("🖼️ Generating PDF preview..."):
+                preview_image_path = preview_pdf(pdf_path, scale_factor=0.5)
+                if preview_image_path:
+                    st.session_state.pdf_preview = preview_image_path
+                    st.image(preview_image_path, caption="First Page Preview", use_container_width=False)
+                    st.session_state["pdf_preview_displayed"] = True
+                else:
+                    st.warning("Failed to generate PDF preview.")
+                    st.session_state.pdf_preview = None
 
         # Load the document into the system
-        st.write("🔄 Loading document into the system...")
-        try:
-            st.session_state.chain = load_chain(pdf_path)
-            st.session_state.LOADED_PATENT = patent_number
-            st.session_state.loaded_pdf_path = pdf_path
-            st.session_state.messages = [{"role": "assistant", "content": "Hello! How can I assist you with this patent?"}]
-            st.success("🚀 Document successfully loaded! You can now start asking questions.")
-        except Exception as e:
-            st.error(f"Failed to load the document: {e}")
-            st.stop()
+        with st.spinner("🔄 Loading document into the system..."):
+            try:
+                st.session_state.chain = load_chain(pdf_path)
+                st.session_state.LOADED_PATENT = patent_number
+                st.session_state.loaded_pdf_path = pdf_path
+                st.session_state.messages = [{"role": "assistant", "content": "Hello! How can I assist you with this patent?"}]
+                st.success("🚀 Document successfully loaded! You can now start asking questions.")
+            except Exception as e:
+                st.error(f"Failed to load the document: {e}")
+                st.stop()
 
     # Display previous chat messages
     if st.session_state.messages: