Update components/session_manager.py

components/session_manager.py

@@ -125,19 +125,53 @@ def load_user_history_from_db():
         print(f"❌ Error loading history from database: {e}")
         st.session_state.user_history = []
 
+# def clear_session():
+#     """Clear session state but preserve user identity and history"""
+#     preserved_keys = ['user_id', 'current_page']  # REMOVED 'user_history' from preserved keys
+#     preserved = {k: st.session_state[k] for k in preserved_keys if k in st.session_state}
+    
+#     st.session_state.clear()
+    
+#     # Restore preserved keys
+#     for k, v in preserved.items():
+#         st.session_state[k] = v
+    
+#     # Re-initialize defaults AND reload history
+#     initialize_session_state()
+
 def clear_session():
-    """Clear session state but preserve user identity and history"""
-    preserved_keys = ['user_id', 'current_page']  # REMOVED 'user_history' from preserved keys
+    """Clear session state but preserve user identity, auth, and navigation"""
+    preserved_keys = [
+        'user_id', 'username', 'fullname', 'authenticated', 
+        'current_page', 'selected_model', 'user_history'
+    ]
+    
     preserved = {k: st.session_state[k] for k in preserved_keys if k in st.session_state}
     
-    st.session_state.clear()
+    # Clear only content-related session state, not auth
+    keys_to_clear = [
+        'user_type', 'original_prompt', 'generated_output', 'feedback_given',
+        'regenerated', 'content_source', 'student_level', 'pdf_export_data',
+        'tutor_topic', 'tutor_content_type', 'feedback_clarity', 'feedback_depth',
+        'feedback_complexity', 'feedback_comments', 'original_filename',
+        'saved_to_history', 'from_history', 'showing_regeneration_prompt',
+        'pending_model_switch', 'previous_model', 'regenerate_with_new_model',
+        'scrolled_to_top', 'regeneration_count', 'regeneration_type',
+        'previous_feedback_given', 'pending_regeneration', 'show_adaptation_message'
+    ]
+    
+    for key in keys_to_clear:
+        if key in st.session_state:
+            del st.session_state[key]
     
     # Restore preserved keys
     for k, v in preserved.items():
         st.session_state[k] = v
     
-    # Re-initialize defaults AND reload history
-    initialize_session_state()
+    # Ensure current page is generator for new content
+    st.session_state.current_page = "generator"
+    
+    print("✅ Session cleared for new content, auth preserved")
 
 def update_session_state(**kwargs):
     # Update all the session state values