Update app/components.py

app/components.py

@@ -71,7 +71,7 @@ def display_components_page():
             st.rerun()
 
 def initialize_components():
-    """Initialize components in session state if not present."""
+    """Initialize components in session state if not present and migrate existing components."""
     if "components" not in st.session_state.project_data:
         st.session_state.project_data["components"] = {
             "walls": [],
@@ -80,6 +80,15 @@ def initialize_components():
             "windows": [],
             "skylights": []
         }
+    migrate_component_types()
+
+def migrate_component_types():
+    """Add 'type' key to existing components for backward compatibility."""
+    for comp_type in st.session_state.project_data["components"]:
+        for component in st.session_state.project_data["components"][comp_type]:
+            if "type" not in component:
+                component["type"] = comp_type
+                logger.info(f"Added type '{comp_type}' to component '{component['name']}'")
 
 def display_component_tab(comp_type: str):
     """
@@ -295,11 +304,19 @@ def display_component_tab(comp_type: str):
                 st.error("Component name must be unique.")
                 return
             
+            # Validate component type
+            valid_types = set(COMPONENT_TYPES)  # {'Walls', 'Roofs', 'Floors', 'Windows', 'Skylights'}
+            if comp_type not in [t.lower() for t in valid_types]:
+                st.error(f"Invalid component type '{comp_type}'. Please select a valid type.")
+                logger.error(f"Invalid component type '{comp_type}' for component '{name}'.")
+                return
+            
             # Create component data
             component_data = {
                 "id": str(uuid.uuid4()),
                 "name": name,
-                "area": area
+                "area": area,
+                "type": comp_type,  # Explicitly set the component type
             }
             
             if comp_type in ["walls", "windows", "roofs", "skylights"]: