Update main.py

main.py

@@ -14,6 +14,7 @@ import logging
 import os
 import sys
 from typing import Dict, List, Any, Optional, Tuple
+from datetime import datetime
 
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -119,9 +120,16 @@ class BuildSustain:
                     "climate_zone": "",
                     "hourly_data": [],
                     "epw_filename": "",
-                    "typical_extreme_periods": {},
+                    "typical_extreme_periods": {
+                        "summer_extreme": {"start": {"month": 7, "day": 1}, "end": {"month": 7, "day": 7}},
+                        "summer_typical": {"start": {"month": 6, "day": 1}, "end": {"month": 6, "day": 7}},
+                        "winter_extreme": {"start": {"month": 1, "day": 1}, "end": {"month": 1, "day": 7}},
+                        "winter_typical": {"start": {"month": 12, "day": 1}, "end": {"month": 12, "day": 7}}
+                    },
                     "ground_temperatures": {
-                        "0.5": [20.0] * 12  # Added default monthly ground temperatures for 0.5m depth
+                        "0.5": [20.0] * 12,
+                        "2": [18.0] * 12,
+                        "4": [16.0] * 12
                     },
                     "ground_reflectivity": 0.2
                 },
@@ -152,6 +160,13 @@ class BuildSustain:
                     "ventilation": [],
                     "infiltration": []
                 },
+                "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
+                },
                 "hvac_loads": {
                     "cooling": {
                         "hourly": [],
@@ -165,8 +180,10 @@ class BuildSustain:
                             "conduction": 0.0,
                             "solar": 0.0,
                             "internal": 0.0,
-                            "ventilation": 0.0,
-                            "infiltration": 0.0
+                            "ventilation_sensible": 0.0,
+                            "ventilation_latent": 0.0,
+                            "infiltration_sensible": 0.0,
+                            "infiltration_latent": 0.0
                         }
                     },
                     "heating": {
@@ -191,8 +208,8 @@ class BuildSustain:
                 },
                 "sim_period": {
                     "type": "Full Year",
-                    "start_date": None,
-                    "end_date": None,
+                    "start_date": datetime(2025, 1, 1),
+                    "end_date": datetime(2025, 12, 31),
                     "base_temp": 18.3
                 },
                 "indoor_conditions": {
@@ -409,8 +426,14 @@ class BuildSustain:
                 "hvac_cooling_rh",
                 "hvac_heating_temp",
                 "hvac_heating_rh",
-                "adaptive_acceptability"
-            ]
+                "adaptive_acceptability",
+                "hvac_air_velocity",
+                "hvac_lighting_convective",
+                "hvac_lighting_radiative",
+                "hvac_equipment_convective",
+                "hvac_equipment_radiative",
+                "ground_temp_depth"
+            ] + [f"ground_temp_{depth}_{month}" for depth in ["0.5", "2", "4"] for month in ["Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"]]
             for key in keys_to_clear:
                 if key in st.session_state:
                     st.session_state.pop(key, None)