Update app/materials_library.py

app/materials_library.py

@@ -355,13 +355,47 @@ def get_stable_button_key(prefix: str, name: str, action: str, unique_id: str =
 
 def initialize_session_state():
     """Initialize all required session state variables."""
+    # Ensure project_data exists
     if "project_data" not in st.session_state:
-        st.session_state.project_data = {
-            "materials": {"library": DEFAULT_MATERIALS.copy(), "project": {}, "table": None},
-            "fenestrations": {"library": DEFAULT_FENESTRATIONS.copy(), "project": {}, "table": None}
+        st.session_state.project_data = {}
+    
+    # Ensure materials structure is fully initialized
+    if "materials" not in st.session_state.project_data:
+        st.session_state.project_data["materials"] = {
+            "library": DEFAULT_MATERIALS.copy(),
+            "project": {},
+            "table": None
+        }
+    else:
+        # Ensure all required keys exist in materials
+        if "library" not in st.session_state.project_data["materials"]:
+            st.session_state.project_data["materials"]["library"] = DEFAULT_MATERIALS.copy()
+        if "project" not in st.session_state.project_data["materials"]:
+            st.session_state.project_data["materials"]["project"] = {}
+        if "table" not in st.session_state.project_data["materials"]:
+            st.session_state.project_data["materials"]["table"] = None
+    
+    # Ensure fenestrations structure is fully initialized
+    if "fenestrations" not in st.session_state.project_data:
+        st.session_state.project_data["fenestrations"] = {
+            "library": DEFAULT_FENESTRATIONS.copy(),
+            "project": {},
+            "table": None
         }
-        update_project_materials_table()  # Initialize materials table
-        update_project_fenestrations_table()  # Initialize fenestrations table
+    else:
+        # Ensure all required keys exist in fenestrations
+        if "library" not in st.session_state.project_data["fenestrations"]:
+            st.session_state.project_data["fenestrations"]["library"] = DEFAULT_FENESTRATIONS.copy()
+        if "project" not in st.session_state.project_data["fenestrations"]:
+            st.session_state.project_data["fenestrations"]["project"] = {}
+        if "table" not in st.session_state.project_data["fenestrations"]:
+            st.session_state.project_data["fenestrations"]["table"] = None
+    
+    # Initialize tables if they are None
+    update_project_materials_table()
+    update_project_fenestrations_table()
+    
+    # Initialize other session state variables
     if "material_editor" not in st.session_state:
         reset_material_editor()
     if "fenestration_editor" not in st.session_state:
@@ -374,6 +408,7 @@ def initialize_session_state():
         st.session_state.fenestration_action = {"action": None, "id": None}
     if "current_page" not in st.session_state:
         st.session_state.current_page = "Material Library"
+    
     logger.debug("Session state initialized")
     logger.debug(f"Project materials: {list(st.session_state.project_data['materials']['project'].keys())}")
     logger.debug(f"Project fenestrations: {list(st.session_state.project_data['fenestrations']['project'].keys())}")