Update app/hvac_loads.py

app/hvac_loads.py

@@ -872,10 +872,10 @@ def display_hvac_loads_page():
             "Country": climate_data.get("location", {}).get("country", ""),
             "City": climate_data.get("location", {}).get("city", ""),
             "State/Province": climate_data.get("location", {}).get("state_province", ""),
-            "Latitude": climate_data.get("latitude", 0.0),
-            "Longitude": climate_data.get("longitude", 0.0),
-            "Elevation": climate_data.get("elevation", 0.0),
-            "Time Zone": climate_data.get("timezone", 0.0),
+            "Latitude": climate_data.get("location", {}).get("latitude", 0.0),
+            "Longitude": climate_data.get("location", {}).get("longitude", 0.0),
+            "Elevation": climate_data.get("location", {}).get("elevation", 0.0),
+            "Time Zone": climate_data.get("location", {}).get("timezone", 0.0),
             "Ground Reflectivity": climate_data.get("ground_reflectivity", 0.2)
         }
         
@@ -939,15 +939,13 @@ def display_hvac_loads_page():
             st.session_state.project_data["climate_data"]["location"].update({
                 "country": country,
                 "city": city,
-                "state_province": state_province
-            })
-            st.session_state.project_data["climate_data"].update({
+                "state_province": state_province,
                 "latitude": latitude,
                 "longitude": longitude,
                 "elevation": elevation,
-                "timezone": timezone,
-                "ground_reflectivity": ground_reflectivity
+                "timezone": timezone
             })
+            st.session_state.project_data["climate_data"]["ground_reflectivity"] = ground_reflectivity
             st.success("Location information saved successfully.")
             logger.info("Location information updated in session state")