Update app.py

app.py

@@ -155,7 +155,7 @@ if __name__ == "__main__":
     # Fetch query parameters safely
     query_params = st.query_params
     default_patent_link = query_params.get("patent_link", "https://patents.google.com/patent/US8676427B1/en")
-    
+
     # Input for Google Patent Link
     patent_link = st.text_area("Enter Google Patent Link:", value=default_patent_link, height=100)
 
@@ -194,13 +194,14 @@ if __name__ == "__main__":
         # Load the document into the system
         st.write("🔄 Loading document into the system...")
 
-        # Persist the chain in session state to prevent reloading
-        if "chain" not in st.session_state or st.session_state.get("loaded_file") != pdf_path:
+        # Check if the patent is already loaded in session state
+        if "LOADED_PATENT" not in st.session_state or st.session_state.get("LOADED_PATENT") != patent_number:
             st.session_state.chain = load_chain(pdf_path)
-            st.session_state.loaded_file = pdf_path
+            st.session_state.LOADED_PATENT = patent_number  # Store the current patent number
             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.success("🚀 Document successfully loaded! You can now start asking questions.")
+        else:
+            st.success("✅ This patent is already loaded. You can continue asking questions.")
 
     # Initialize messages if not already done
     if "messages" not in st.session_state: