Update app.py

app.py

@@ -168,10 +168,8 @@ if __name__ == "__main__":
         st.session_state.loaded_pdf_path = None
     if "chain" not in st.session_state:
         st.session_state.chain = None
-    if "messages" not in st.session_state:
-        st.session_state.messages = [{"role": "assistant", "content": "Hello! How can I assist you with this patent?"}]
 
-    # Load and process patent
+    # Button to load and process patent
     if st.button("Load and Process Patent"):
         if not patent_link:
             st.warning("Please enter a valid Google patent link.")
@@ -194,33 +192,35 @@ if __name__ == "__main__":
         else:
             st.write("✅ File already downloaded.")
 
-        # Generate preview if not already stored
-        if st.session_state.LOADED_PATENT != patent_number:
-            st.write("🖼️ Generating PDF preview...")
-            preview_image_path = preview_pdf(pdf_path)
-            if preview_image_path:
-                st.session_state.pdf_preview = preview_image_path
-            else:
-                st.warning("Failed to generate PDF preview.")
-                st.session_state.pdf_preview = None
-
-            # Load the chain and store metadata
-            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!")
+        # Generate PDF preview
+        st.write("🖼️ Generating PDF preview...")
+        preview_image_path = preview_pdf(pdf_path)
+        if preview_image_path:
+            st.session_state.pdf_preview = preview_image_path
+            st.image(preview_image_path, caption="First Page Preview", use_container_width=True)
         else:
-            st.info("✅ This patent is already loaded.")
+            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...")
+        st.session_state.chain = load_chain(pdf_path)
+        st.session_state.LOADED_PATENT = patent_number
+        st.session_state.loaded_pdf_path = pdf_path
+
+        # Initialize messages AFTER processing
+        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.")
 
     # Display the PDF preview if available
     if st.session_state.pdf_preview:
         st.image(st.session_state.pdf_preview, caption="First Page Preview", use_container_width=True)
 
     # Display previous chat messages
-    for message in st.session_state.messages:
-        with st.chat_message(message["role"]):
-            st.markdown(message["content"])
+    if "messages" in st.session_state:
+        for message in st.session_state.messages:
+            with st.chat_message(message["role"]):
+                st.markdown(message["content"])
 
     # User input for questions
     if st.session_state.chain: