Update app/hvac_loads.py

app/hvac_loads.py

@@ -1099,7 +1099,7 @@ def display_hvac_loads_page():
             index=["Fixed Setpoints", "ASHRAE 55 Adaptive Comfort"].index(st.session_state.project_data["indoor_conditions"]["type"])
         )
         st.session_state.project_data["indoor_conditions"]["type"] = indoor_type
-
+        
         if indoor_type == "Fixed Setpoints":
             col1, col2 = st.columns(2)
             with col1:
@@ -1152,6 +1152,123 @@ def display_hvac_loads_page():
                 key="adaptive_acceptability"
             )
             st.session_state.project_data["indoor_conditions"]["adaptive_acceptability"] = acceptability
+        
+        # Internal Loads Conditions Configuration
+        st.subheader("Internal Loads Conditions")
+        col1, col2, col3, col4, col5 = st.columns(5)
+        
+        # Initialize internal_loads_conditions if not present
+        if "internal_loads_conditions" not in st.session_state.project_data:
+            st.session_state.project_data["internal_loads_conditions"] = {
+                "air_velocity": 0.1,
+                "lighting_convective_fraction": 0.5,
+                "lighting_radiative_fraction": 0.5,
+                "equipment_convective_fraction": 0.5,
+                "equipment_radiative_fraction": 0.5
+            }
+        
+        # Retrieve internal loads data
+        internal_loads = st.session_state.project_data.get("internal_loads", {})
+        lighting_systems = internal_loads.get("lighting", [])
+        equipment_systems = internal_loads.get("equipment", [])
+        
+        # Calculate average fractions from lighting systems
+        lighting_convective_total = 0.0
+        lighting_radiative_total = 0.0
+        lighting_count = len(lighting_systems) or 1  # Avoid division by zero
+        for system in lighting_systems:
+            lighting_convective_total += system.get("convective_fraction", 0.5)
+            lighting_radiative_total += system.get("radiative_fraction", 0.5)
+        default_lighting_convective = lighting_convective_total / lighting_count
+        default_lighting_radiative = lighting_radiative_total / lighting_count
+        
+        # Calculate average fractions from equipment systems
+        equipment_convective_total = 0.0
+        equipment_radiative_total = 0.0
+        equipment_count = len(equipment_systems) or 1  # Avoid division by zero
+        for system in equipment_systems:
+            equipment_convective_total += system.get("convective_fraction", 0.5)
+            equipment_radiative_total += system.get("radiative_fraction", 0.5)
+        default_equipment_convective = equipment_convective_total / equipment_count
+        default_equipment_radiative = equipment_radiative_total / equipment_count
+        
+        with col1:
+            air_velocity = st.number_input(
+                "Air Velocity (m/s)",
+                min_value=0.0,
+                max_value=2.0,
+                value=st.session_state.project_data["internal_loads_conditions"].get("air_velocity", 0.1),
+                step=0.01,
+                key="hvac_air_velocity"
+            )
+            if air_velocity < 0.0 or air_velocity > 2.0:
+                st.error("Air velocity must be between 0 and 2 m/s.")
+                air_velocity = max(0.0, min(2.0, air_velocity))
+        
+        with col2:
+            lighting_convective_fraction = st.number_input(
+                "Lighting Convective Fraction",
+                min_value=0.0,
+                max_value=1.0,
+                value=st.session_state.project_data["internal_loads_conditions"].get("lighting_convective_fraction", default_lighting_convective),
+                step=0.01,
+                key="hvac_lighting_convective"
+            )
+        
+        with col3:
+            lighting_radiative_fraction = st.number_input(
+                "Lighting Radiative Fraction",
+                min_value=0.0,
+                max_value=1.0,
+                value=st.session_state.project_data["internal_loads_conditions"].get("lighting_radiative_fraction", default_lighting_radiative),
+                step=0.01,
+                key="hvac_lighting_radiative"
+            )
+            # Validate lighting fractions sum to 1.0
+            if abs(lighting_convective_fraction + lighting_radiative_fraction - 1.0) > 0.01:
+                st.error("Lighting convective and radiative fractions must sum to 1.0.")
+        
+        with col4:
+            equipment_convective_fraction = st.number_input(
+                "Equipment Convective Fraction",
+                min_value=0.0,
+                max_value=1.0,
+                value=st.session_state.project_data["internal_loads_conditions"].get("equipment_convective_fraction", default_equipment_convective),
+                step=0.01,
+                key="hvac_equipment_convective"
+            )
+        
+        with col5:
+            equipment_radiative_fraction = st.number_input(
+                "Equipment Radiative Fraction",
+                min_value=0.0,
+                max_value=1.0,
+                value=st.session_state.project_data["internal_loads_conditions"].get("equipment_radiative_fraction", default_equipment_radiative),
+                step=0.01,
+                key="hvac_equipment_radiative"
+            )
+            # Validate equipment fractions sum to 1.0
+            if abs(equipment_convective_fraction + equipment_radiative_fraction - 1.0) > 0.01:
+                st.error("Equipment convective and radiative fractions must sum to 1.0.")
+        
+        if st.button("Save Internal Loads Conditions"):
+            st.session_state.project_data["internal_loads_conditions"].update({
+                "air_velocity": air_velocity,
+                "lighting_convective_fraction": lighting_convective_fraction,
+                "lighting_radiative_fraction": lighting_radiative_fraction,
+                "equipment_convective_fraction": equipment_convective_fraction,
+                "equipment_radiative_fraction": equipment_radiative_fraction
+            })
+            # Update lighting systems with new fractions
+            for system in lighting_systems:
+                system["convective_fraction"] = lighting_convective_fraction
+                system["radiative_fraction"] = lighting_radiative_fraction
+            # Update equipment systems with new fractions
+            for system in equipment_systems:
+                system["convective_fraction"] = equipment_convective_fraction
+                system["radiative_fraction"] = equipment_radiative_fraction
+            st.success("Internal loads conditions saved successfully.")
+            logger.info("Internal loads conditions updated in session state")
 
         # Calculate HVAC Loads
         if st.button("Calculate HVAC Loads"):