Upload 31 files

app/component_selection.py

@@ -38,81 +38,93 @@ class ComponentSelection:
         Args:
             session_state: Streamlit session state
         """
+        # Ensure components structure exists
+        if "components" not in session_state:
+            session_state["components"] = {
+                'walls': [],
+                'roofs': [],
+                'floors': [],
+                'windows': [],
+                'doors': []
+            }
+            
         # Initialize pending actions if not exists
         if "pending_actions" not in session_state:
-            session_state["pending_actions"] = {
-                "add_wall": None,
-                "delete_wall": None,
-                "add_roof": None,
-                "delete_roof": None,
-                "add_floor": None,
-                "delete_floor": None,
-                "add_window": None,
-                "delete_window": None,
-                "add_door": None,
-                "delete_door": None
-            }
+            session_state["pending_actions"] = {}
+            
+        # Ensure all required pending action keys exist
+        required_keys = [
+            "add_wall", "delete_wall",
+            "add_roof", "delete_roof",
+            "add_floor", "delete_floor",
+            "add_window", "delete_window",
+            "add_door", "delete_door"
+        ]
+        
+        for key in required_keys:
+            if key not in session_state["pending_actions"]:
+                session_state["pending_actions"][key] = None
         
         # Process add wall action
-        if session_state["pending_actions"]["add_wall"] is not None:
+        if session_state["pending_actions"].get("add_wall") is not None:
             new_wall = session_state["pending_actions"]["add_wall"]
             session_state["components"]["walls"].append(new_wall)
             session_state["pending_actions"]["add_wall"] = None
         
         # Process delete wall action
-        if session_state["pending_actions"]["delete_wall"] is not None:
+        if session_state["pending_actions"].get("delete_wall") is not None:
             wall_index = session_state["pending_actions"]["delete_wall"]
             if 0 <= wall_index < len(session_state["components"]["walls"]):
                 session_state["components"]["walls"].pop(wall_index)
             session_state["pending_actions"]["delete_wall"] = None
         
         # Process add roof action
-        if session_state["pending_actions"]["add_roof"] is not None:
+        if session_state["pending_actions"].get("add_roof") is not None:
             new_roof = session_state["pending_actions"]["add_roof"]
             session_state["components"]["roofs"].append(new_roof)
             session_state["pending_actions"]["add_roof"] = None
         
         # Process delete roof action
-        if session_state["pending_actions"]["delete_roof"] is not None:
+        if session_state["pending_actions"].get("delete_roof") is not None:
             roof_index = session_state["pending_actions"]["delete_roof"]
             if 0 <= roof_index < len(session_state["components"]["roofs"]):
                 session_state["components"]["roofs"].pop(roof_index)
             session_state["pending_actions"]["delete_roof"] = None
         
         # Process add floor action
-        if session_state["pending_actions"]["add_floor"] is not None:
+        if session_state["pending_actions"].get("add_floor") is not None:
             new_floor = session_state["pending_actions"]["add_floor"]
             session_state["components"]["floors"].append(new_floor)
             session_state["pending_actions"]["add_floor"] = None
         
         # Process delete floor action
-        if session_state["pending_actions"]["delete_floor"] is not None:
+        if session_state["pending_actions"].get("delete_floor") is not None:
             floor_index = session_state["pending_actions"]["delete_floor"]
             if 0 <= floor_index < len(session_state["components"]["floors"]):
                 session_state["components"]["floors"].pop(floor_index)
             session_state["pending_actions"]["delete_floor"] = None
         
         # Process add window action
-        if session_state["pending_actions"]["add_window"] is not None:
+        if session_state["pending_actions"].get("add_window") is not None:
             new_window = session_state["pending_actions"]["add_window"]
             session_state["components"]["windows"].append(new_window)
             session_state["pending_actions"]["add_window"] = None
         
         # Process delete window action
-        if session_state["pending_actions"]["delete_window"] is not None:
+        if session_state["pending_actions"].get("delete_window") is not None:
             window_index = session_state["pending_actions"]["delete_window"]
             if 0 <= window_index < len(session_state["components"]["windows"]):
                 session_state["components"]["windows"].pop(window_index)
             session_state["pending_actions"]["delete_window"] = None
         
         # Process add door action
-        if session_state["pending_actions"]["add_door"] is not None:
+        if session_state["pending_actions"].get("add_door") is not None:
             new_door = session_state["pending_actions"]["add_door"]
             session_state["components"]["doors"].append(new_door)
             session_state["pending_actions"]["add_door"] = None
         
         # Process delete door action
-        if session_state["pending_actions"]["delete_door"] is not None:
+        if session_state["pending_actions"].get("delete_door") is not None:
             door_index = session_state["pending_actions"]["delete_door"]
             if 0 <= door_index < len(session_state["components"]["doors"]):
                 session_state["components"]["doors"].pop(door_index)

app/results_display.py

@@ -52,8 +52,16 @@ class ResultsDisplay:
         Args:
             session_state: Streamlit session state containing pending actions
         """
+        # Ensure calculation_results structure exists
+        if "calculation_results" not in session_state:
+            session_state["calculation_results"] = {
+                'cooling': {},
+                'heating': {}
+            }
+            
+        # Initialize pending actions if not exists
         if "pending_actions" not in session_state:
-            return
+            session_state["pending_actions"] = {}
             
         # Process save scenario action
         if "save_scenario" in session_state["pending_actions"]: