Update app.py

app.py

@@ -36,49 +36,44 @@ if 'page_loaded' not in st.session_state:
     st.session_state.last_query = ""
     st.session_state.last_answer = None
 
-# Add new session state variable for detecting page reloads
-if 'page_visit_id' not in st.session_state:
-    import uuid
-    # Generate a unique ID for this page load
-    st.session_state.page_visit_id = str(uuid.uuid4())
+# Add a flag to track if we need to check for navigation
+if 'check_navigation' not in st.session_state:
+    st.session_state.check_navigation = True
 
-# Use JavaScript to detect when user navigates away and back
-def setup_navigation_detection():
+# Check if we need to reset state on page load
+if st.session_state.check_navigation:
+    # Reset key query-related state variables
+    st.session_state.last_query = ""
+    st.session_state.last_answer = None
+    st.session_state.submit_clicked = False
+    # If you have any other state to reset, do it here
+    
+    # Mark that we've checked navigation for this page load
+    st.session_state.check_navigation = False
+    
+    # Add JavaScript to reset the flag when navigating away
     components.html(
-        f"""
+        """
         <script>
-        (function() {{
-            // Store the current page visit ID
-            const currentVisitId = "{st.session_state.page_visit_id}";
-            
-            // Check if we have a stored ID from before
-            const previousVisitId = localStorage.getItem('lastVisitId');
-            
-            // If we have a previous ID and it's different, we've returned after navigation
-            if (previousVisitId && previousVisitId !== currentVisitId) {{
-                console.log('Returned to app after navigation');
-                
-                // Force a reload of the page to reset all state
-                window.location.reload();
-            }}
-            
-            // Store current ID for next time
-            localStorage.setItem('lastVisitId', currentVisitId);
-            
-            // Listen for navigation away from the page
-            window.addEventListener('beforeunload', function() {{
-                // When navigating away, we want to be able to detect return
-                localStorage.setItem('navigatedAway', 'true');
-            }});
-        }})();
+        // Set up beforeunload event to prepare for next page load
+        window.addEventListener('beforeunload', function() {
+            // This will be executed when the user navigates away
+            // We'll use sessionStorage which persists across page loads
+            sessionStorage.setItem('resetOnReturn', 'true');
+        });
+        
+        // Check if we need to reload on return
+        if (sessionStorage.getItem('resetOnReturn') === 'true') {
+            // Clear the flag
+            sessionStorage.removeItem('resetOnReturn');
+            // Force page reload to reset state
+            window.location.reload();
+        }
         </script>
         """,
         height=0
     )
 
-# Call this function early in your script
-setup_navigation_detection()
-
 # THEN: Import your modules
 from rag_engine import process_query, load_model, cached_load_data_files
 from utils import setup_all_auth