Update interim/app.py

interim/app.py

@@ -167,19 +167,25 @@ if __name__ == "__main__":
             st.write(f"✅ File downloaded: {pdf_path}")
 
         st.write("🔄 Loading document into the system...")
-        chain = load_chain(pdf_path)
+
+        # 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:
+            st.session_state.chain = load_chain(pdf_path)
+            st.session_state.loaded_file = 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.")
 
-        # Initialize chat messages
-        if "messages" not in st.session_state:
-            st.session_state["messages"] = [{"role": "assistant", "content": "Hello! How can I assist you with this patent?"}]
+    # Initialize messages if not already done
+    if "messages" not in st.session_state:
+        st.session_state.messages = [{"role": "assistant", "content": "Hello! How can I assist you with this patent?"}]
 
-        # Display previous chat messages
-        for message in st.session_state.messages:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
+    # Display previous chat messages
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
 
-        # Chat Input
+    if "chain" in st.session_state:
         if user_input := st.chat_input("What is your question?"):
             st.session_state.messages.append({"role": "user", "content": user_input})
             with st.chat_message("user"):
@@ -189,15 +195,14 @@ if __name__ == "__main__":
                 message_placeholder = st.empty()
                 full_response = ""
 
-            with st.spinner("Generating response..."):
-                try:
-                    assistant_response = chain({"question": user_input})
-                    full_response = assistant_response["answer"]
-                    message_placeholder.markdown(full_response)
-                except Exception as e:
-                    full_response = f"An error occurred: {e}"
-                    message_placeholder.markdown(full_response)
+                with st.spinner("Generating response..."):
+                    try:
+                        assistant_response = st.session_state.chain({"question": user_input})
+                        full_response = assistant_response["answer"]
+                    except Exception as e:
+                        full_response = f"An error occurred: {e}"
 
-            st.session_state.messages.append({"role": "assistant", "content": full_response})
+                message_placeholder.markdown(full_response)
+                st.session_state.messages.append({"role": "assistant", "content": full_response})
     else:
         st.info("Press the 'Load and Process Patent' button to start processing.")