Update main.py

main.py

@@ -86,7 +86,37 @@ class BuildSustain:
                     "winter_indoor_design_rh": 50.0,
                     "orientation_angle": 0.0
                 },
-                "climate_data": {},
+                "climate_data": {
+                    "id": "",
+                    "location": {
+                        "city": "",
+                        "state_province": "",
+                        "country": "",
+                        "source": "",
+                        "wmo": "",
+                        "latitude": 0.0,
+                        "longitude": 0.0,
+                        "timezone": 0.0,
+                        "elevation": 0.0
+                    },
+                    "design_conditions": {
+                        "winter_design_temp": 0.0,
+                        "summer_design_temp_db": 30.0,
+                        "summer_design_temp_wb": 25.0,
+                        "heating_degree_days": 0,
+                        "cooling_degree_days": 0,
+                        "monthly_average_temps": [20.0] * 12,
+                        "monthly_average_radiation": [150.0] * 12,
+                        "summer_daily_range": 8.0,
+                        "wind_speed": 3.0,
+                        "pressure": 101325.0
+                    },
+                    "climate_zone": "",
+                    "hourly_data": [],
+                    "epw_filename": "",
+                    "typical_extreme_periods": {},
+                    "ground_temperatures": {}
+                },
                 "materials": {
                     "library": dict(SAMPLE_MATERIALS),
                     "project": {}
@@ -165,7 +195,7 @@ class BuildSustain:
         if 'debug_mode' not in st.session_state:
             st.session_state.debug_mode = False
         
-        # Initialize module-specific rerun flags (for future modules that might need them)
+        # Initialize module-specific rerun flags
         if 'module_rerun_flags' not in st.session_state:
             st.session_state.module_rerun_flags = {}
     
@@ -237,7 +267,7 @@ class BuildSustain:
         # Display the selected page content
         self.display_page(st.session_state.current_page)
         
-        # Handle module-specific reruns (only if needed by specific modules)
+        # Handle module-specific reruns
         self._handle_module_reruns()
     
     def _handle_page_transition(self, previous_page: str, new_page: str):
@@ -247,7 +277,6 @@ class BuildSustain:
         """
         # Clear materials library specific states when leaving that page
         if previous_page == "Material Library":
-            # Only clear materials library editor states
             keys_to_clear = [
                 "material_editor", 
                 "fenestration_editor",
@@ -265,8 +294,20 @@ class BuildSustain:
                 if key in st.session_state:
                     st.session_state.pop(key, None)
         
-        # Add similar cleanup for other modules as needed
-        # This approach ensures each module's cleanup is isolated
+        # Clear climate data specific states when leaving the Climate Data page
+        if previous_page == "Climate Data":
+            keys_to_clear = [
+                "data_source",
+                "projection_country",
+                "projection_state",
+                "location",
+                "rcp",
+                "year",
+                "climate_tab"
+            ]
+            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 = {}
@@ -292,40 +333,28 @@ class BuildSustain:
             page: The name of the page to display
         """
         if page == "Intro":
-            # Call the implemented intro module
             display_intro_page()
         elif page == "Building Information":
-            # Call the implemented building information module
             display_building_info_page()
         elif page == "Climate Data":
-            # Call the implemented climate data module
             display_climate_page()
         elif page == "Material Library":
-            # Call the implemented material library module
             display_materials_page()
         elif page == "Construction":
-            # Call the implemented construction module
             display_construction_page()
         elif page == "Building Components":
-            # Call the implemented components module
             display_components_page()
         elif page == "Internal Loads":
-            # Call the implemented internal loads module
             display_internal_loads_page()
         elif page == "HVAC Loads":
-            # Call the implemented HVAC loads module
             display_hvac_loads_page()
         elif page == "Building Energy":
-            # Call the implemented building energy module
             display_building_energy_page()
         elif page == "Renewable Energy":
-            # Call the implemented renewable energy module
             display_renewable_energy_page()
         elif page == "Embodied Energy":
-            # Call the implemented embodied energy module
             display_embodied_energy_page()
         elif page == "Materials Cost":
-            # Call the implemented materials cost module
             display_materials_cost_page()
         else:
             st.error(f"Unknown page: {page}")