Update main.py

main.py

@@ -134,7 +134,6 @@ class BuildSustain:
                     "roofs": [],
                     "floors": [],
                     "windows": [],
-                    "doors": [],
                     "skylights": []
                 },
                 "internal_loads": {
@@ -322,6 +321,25 @@ class BuildSustain:
                 if key in st.session_state:
                     st.session_state.pop(key, None)
         
+        # Clear components-specific states when leaving the Building Components page
+        if previous_page == "Building Components":
+            keys_to_clear = [
+                "walls_editor",
+                "roofs_editor",
+                "floors_editor",
+                "windows_editor",
+                "skylights_editor",
+                "walls_action",
+                "roofs_action",
+                "floors_action",
+                "windows_action",
+                "skylights_action",
+                "components_rerun_pending"
+            ]
+            for key in keys_to_clear:
+                if key in st.session_state:
+                    st.session_state.pop(key, None)
+        
         # Clear any module-specific rerun flags when changing pages
         st.session_state.module_rerun_flags = {}
     
@@ -332,11 +350,11 @@ class BuildSustain:
         """
         # Check for any module-specific rerun requests
         for module_name, should_rerun in st.session_state.module_rerun_flags.items():
-            if should_rerun:
-                # Clear the flag and trigger rerun
-                st.session_state.module_rerun_flags[module_name] = False
-                st.rerun()
-                break  # Only handle one rerun at a time
+            if should_rer How to play:
+            # Clear the flag and trigger rerun
+            st.session_state.module_rerun_flags[module_name] = False
+            st.rerun()
+            break  # Only handle one rerun at a time
     
     def display_page(self, page: str):
         """