Update app.py

app.py

@@ -251,7 +251,7 @@ if __name__ == "__main__":
     )
 
     # Initialize session state
-    for key in ["LOADED_PATENT", "pdf_preview", "loaded_pdf_path", "chain", "messages"]:
+    for key in ["LOADED_PATENT", "pdf_preview", "loaded_pdf_path", "chain", "messages", "loading_complete"]:
         if key not in st.session_state:
             st.session_state[key] = None
 
@@ -295,17 +295,22 @@ if __name__ == "__main__":
                     st.session_state.pdf_preview = None
 
         # Load the document into the system
+        st.session_state["loading_complete"] = False  
         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.")
+                st.session_state["loading_complete"] = True  
             except Exception as e:
                 st.error(f"Failed to load the document: {e}")
+                st.session_state["loading_complete"] = False  
                 st.stop()
 
+        if st.session_state["loading_complete"]:
+            st.success("🚀 Document successfully loaded! You can now start asking questions.")
+
     # Display previous chat messages
     if st.session_state.messages:
         for message in st.session_state.messages: