Update data/calculation.py

data/calculation.py

@@ -24,12 +24,17 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 logger = logging.getLogger(__name__)
 
 # Initialize session_state for climate data
-    if "climate_data" not in st.session_state:
-        st.session_state["climate_data"] = {
-            "latitude": 0.0,
-            "longitude": 0.0,
-            "timezone": 0.0
-        }
+    building_info["climate_data"] = st.session_state.get("climate_data", {
+        "latitude": 0.0,
+        "longitude": 0.0,
+        "timezone": 0.0
+    })
+if "climate_data" not in st.session_state:
+    st.session_state["climate_data"] = {
+        "latitude": 0.0,
+        "longitude": 0.0,
+        "timezone": 0.0
+    }
 
 class SolarCalculations:
     """Class for performing solar radiation and angle calculations based on ASHRAE methodologies."""