Update app.py

app.py

@@ -36,6 +36,49 @@ 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())
+
+# Use JavaScript to detect when user navigates away and back
+def setup_navigation_detection():
+    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');
+            }});
+        }})();
+        </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