Update app/internal_loads.py

app/internal_loads.py

@@ -923,28 +923,28 @@ def display_ventilation_infiltration_tab():
             st.session_state.module_rerun_flags["internal_loads"] = True
             st.rerun()
 
-import streamlit as st
-from streamlit_vertical_slider import vertical_slider
-
 def display_schedules_tab():
     st.subheader("Schedules Editor")
 
+    DEFAULT_STATE = {
+        "name": "",
+        "description": "",
+        "template": "None",
+        "weekday": [0.0] * 24,
+        "weekend": [0.0] * 24,
+        "is_edit": False
+    }
+
     if "schedule_editor" not in st.session_state:
-        st.session_state.schedule_editor = {
-            "name": "",
-            "description": "",
-            "template": "None",
-            "weekday": [0.0]*24,
-            "weekend": [0.0]*24,
-            "is_edit": False
-        }
+        st.session_state.schedule_editor = DEFAULT_STATE.copy()
 
     editor_state = st.session_state.schedule_editor
     schedules = st.session_state.project_data["internal_loads"]["schedules"]
 
+    # Form for metadata only
     with st.form("schedule_form"):
-        name = st.text_input("Schedule Name", value=editor_state["name"])
-        description = st.text_area("Description", value=editor_state["description"])
+        name = st.text_input("Schedule Name", value=editor_state.get("name", ""))
+        description = st.text_area("Description", value=editor_state.get("description", ""))
         template_options = list(DEFAULT_SCHEDULE_TEMPLATES.keys())
         selected_template = st.selectbox(
             "Select Template",
@@ -953,8 +953,8 @@ def display_schedules_tab():
             if editor_state.get("template") in template_options else 0
         )
 
-        # Load template values and write into st.session_state directly
-        if selected_template != editor_state["template"]:
+        # If template changed, update values and sliders
+        if selected_template != editor_state.get("template", "None"):
             st.session_state.schedule_editor["template"] = selected_template
             if selected_template != "None":
                 tpl = DEFAULT_SCHEDULE_TEMPLATES[selected_template]
@@ -964,56 +964,74 @@ def display_schedules_tab():
                 st.session_state.schedule_editor["weekday"] = tpl["weekday"]
                 st.session_state.schedule_editor["weekend"] = tpl["weekend"]
 
-        st.markdown("### Weekday Schedule")
-        cols = st.columns(24)
-        weekday_values = []
-        for hour in range(24):
-            with cols[hour]:
-                val = st.slider(
-                    label=f"{hour:02d}",
-                    min_value=0.0,
-                    max_value=1.0,
-                    value=st.session_state.get(f"weekday_{hour}", 0.0),
-                    step=0.1,
-                    key=f"weekday_{hour}"
-                )
-                weekday_values.append(val)
+        submitted = st.form_submit_button("Save Schedule")
 
-        st.markdown("### Weekend Schedule")
-        cols = st.columns(24)
-        weekend_values = []
-        for hour in range(24):
-            with cols[hour]:
-                val = st.slider(
-                    label=f"{hour:02d}",
-                    min_value=0.0,
-                    max_value=1.0,
-                    value=st.session_state.get(f"weekend_{hour}", 0.0),
-                    step=0.1,
-                    key=f"weekend_{hour}"
-                )
-                weekend_values.append(val)
+    # Display weekday sliders
+    st.markdown("### Weekday Schedule")
+    cols = st.columns(24)
+    weekday_values = []
+    for hour, col in enumerate(cols):
+        with col:
+            val = vertical_slider(
+                label="",
+                key=f"weekday_{hour}",
+                height=150,
+                step=0.1,
+                min_value=0.0,
+                max_value=1.0,
+                default_value=st.session_state.get(f"weekday_{hour}", 0.0),
+                slider_color="#4CAF50",
+                track_color="#e0e0e0",
+                thumb_color="#333",
+                value_always_visible=False
+            )
+            st.markdown(f"<div style='text-align:center; font-size:10px'>{hour:02d}:00</div>", unsafe_allow_html=True)
+            weekday_values.append(val)
 
-        submitted = st.form_submit_button("Save Schedule")
-        if submitted:
+    st.markdown("### Weekend Schedule")
+    cols = st.columns(24)
+    weekend_values = []
+    for hour, col in enumerate(cols):
+        with col:
+            val = vertical_slider(
+                label="",
+                key=f"weekend_{hour}",
+                height=150,
+                step=0.1,
+                min_value=0.0,
+                max_value=1.0,
+                default_value=st.session_state.get(f"weekend_{hour}", 0.0),
+                slider_color="#2196F3",
+                track_color="#e0e0e0",
+                thumb_color="#333",
+                value_always_visible=False
+            )
+            st.markdown(f"<div style='text-align:center; font-size:10px'>{hour:02d}:00</div>", unsafe_allow_html=True)
+            weekend_values.append(val)
+
+    # Handle saving
+    if submitted:
+        try:
             if not name.strip():
                 st.error("Schedule name is required.")
             elif name in schedules and not editor_state.get("is_edit"):
-                st.error("A schedule with this name already exists.")
+                st.error("Schedule already exists.")
             else:
                 schedules[name] = {
                     "description": description,
                     "weekday": weekday_values,
                     "weekend": weekend_values
                 }
-                st.session_state.schedule_editor = {}
-                st.success(f"Schedule '{name}' saved successfully!")
+                st.session_state.schedule_editor = DEFAULT_STATE.copy()
+                st.success(f"Schedule '{name}' saved successfully.")
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
 
-    st.write("### Saved Schedules")
+    st.markdown("### Saved Schedules")
     if schedules:
         display_schedules_table(schedules)
     else:
-        st.info("No schedules saved yet.")
+        st.info("No schedules saved.")
 
 def is_schedule_in_use(schedule_name: str) -> bool:
     """