Update data/climate_data.py

data/climate_data.py

@@ -20,7 +20,6 @@ from io import StringIO
 # Define paths
 DATA_DIR = os.path.dirname(os.path.abspath(__file__))
 
-
 @dataclass
 class ClimateLocation:
     """Class representing a climate location with ASHRAE 169 data."""
@@ -63,7 +62,6 @@ class ClimateLocation:
             "monthly_humidity": self.monthly_humidity
         }
 
-
 class ClimateData:
     """Class for managing ASHRAE 169 climate data."""
     
@@ -229,32 +227,37 @@ class ClimateData:
                         monthly_humidity[month] = st.number_input(f"{month} Humidity (%)", min_value=0.0, max_value=100.0, value=50.0, step=5.0, key=f"hum_{month}")
                 
                 if st.form_submit_button("Save Climate Data"):
-                    # Generate ID internally using country and city from session_state
-                    generated_id = f"{session_state.building_info['country'][:2].upper()}-{session_state.building_info['city'][:3].upper()}"
-                    location = ClimateLocation(
-                        id=generated_id,
-                        country=session_state.building_info["country"],
-                        state_province="N/A",  # Default since input removed
-                        city=session_state.building_info["city"],
-                        latitude=latitude,
-                        longitude=longitude,
-                        elevation=elevation,
-                        climate_zone=climate_zone,
-                        heating_degree_days=hdd,
-                        cooling_degree_days=cdd,
-                        winter_design_temp=winter_design_temp,
-                        summer_design_temp_db=summer_design_temp_db,
-                        summer_design_temp_wb=summer_design_temp_wb,
-                        summer_daily_range=summer_daily_range,
-                        monthly_temps=monthly_temps,
-                        monthly_humidity=monthly_humidity
-                    )
-                    self.add_location(location)
-                    st.success("Climate data saved manually!")
-                    self.display_design_conditions(location)
-                    self.visualize_data(location, epw_data=None)
+                    try:
+                        # Generate ID internally using country and city from session_state
+                        generated_id = f"{session_state.building_info['country'][:2].upper()}-{session_state.building_info['city'][:3].upper()}"
+                        location = ClimateLocation(
+                            id=generated_id,
+                            country=session_state.building_info["country"],
+                            state_province="N/A",  # Default since input removed
+                            city=session_state.building_info["city"],
+                            latitude=latitude,
+                            longitude=longitude,
+                            elevation=elevation,
+                            climate_zone=climate_zone,
+                            heating_degree_days=hdd,
+                            cooling_degree_days=cdd,
+                            winter_design_temp=winter_design_temp,
+                            summer_design_temp_db=summer_design_temp_db,
+                            summer_design_temp_wb=summer_design_temp_wb,
+                            summer_daily_range=summer_daily_range,
+                            monthly_temps=monthly_temps,
+                            monthly_humidity=monthly_humidity
+                        )
+                        self.add_location(location)
+                        session_state["climate_data"] = location.to_dict()  # Save to session state
+                        st.success("Climate data saved manually!")
+                        st.write(f"Debug: Saved climate data for {location.city}: {location.to_dict()}")  # Debug
+                        self.display_design_conditions(location)
+                        self.visualize_data(location, epw_data=None)
+                    except Exception as e:
+                        st.error(f"Error saving climate data: {str(e)}. Please check inputs and try again.")
 
-        # EPW Upload Tab (unchanged)
+        # EPW Upload Tab
         with tab2:
             uploaded_file = st.file_uploader("Upload EPW File", type=["epw"])
             if uploaded_file:
@@ -326,7 +329,9 @@ class ClimateData:
                         monthly_humidity=monthly_humidity
                     )
                     self.add_location(location)
+                    session_state["climate_data"] = location.to_dict()  # Save to session state
                     st.success("Climate data extracted from EPW file with calculated Wet Bulb Temperature!")
+                    st.write(f"Debug: Saved climate data for {location.city}: {location.to_dict()}")  # Debug
                     self.display_design_conditions(location)
                     self.visualize_data(location, epw_data=epw_data)
                 except Exception as e:
@@ -341,6 +346,11 @@ class ClimateData:
             else:
                 st.button("Continue to Building Components", disabled=True)
 
+        # Display saved session state data (if any)
+        if "climate_data" in session_state and session_state["climate_data"]:
+            st.subheader("Saved Climate Data")
+            st.json(session_state["climate_data"])  # Display as JSON for clarity
+
     def display_design_conditions(self, location: ClimateLocation):
         """Display a table of design conditions including additional parameters for HVAC calculations."""
         st.subheader("Design Conditions for HVAC Calculations")