Update app/internal_loads.py

app/internal_loads.py

@@ -924,6 +924,8 @@ def display_ventilation_infiltration_tab():
             st.rerun()
 
 def display_schedules_tab():
+    import streamlit as st
+
     st.subheader("Schedules Editor")
 
     DEFAULT_STATE = {
@@ -941,7 +943,7 @@ def display_schedules_tab():
     editor_state = st.session_state.schedule_editor
     schedules = st.session_state.project_data["internal_loads"]["schedules"]
 
-    # Form for metadata only
+    # ---------------------- UI FORM for metadata ----------------------
     with st.form("schedule_form"):
         name = st.text_input("Schedule Name", value=editor_state.get("name", ""))
         description = st.text_area("Description", value=editor_state.get("description", ""))
@@ -953,89 +955,80 @@ def display_schedules_tab():
             if editor_state.get("template") in template_options else 0
         )
 
-        # If template changed, update values and sliders
+        # Load selected template into 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]
-                for hour in range(24):
-                    st.session_state[f"weekday_{hour}"] = tpl["weekday"][hour]
-                    st.session_state[f"weekend_{hour}"] = tpl["weekend"][hour]
                 st.session_state.schedule_editor["weekday"] = tpl["weekday"]
                 st.session_state.schedule_editor["weekend"] = tpl["weekend"]
 
         submitted = st.form_submit_button("Save Schedule")
 
-    # Display weekday sliders
-    st.markdown("### Weekday Schedule")
-    cols = st.columns(24)
-    weekday_values = []
-    for hour, col in enumerate(cols):
-        with col:
-            key = f"weekday_{hour}"
-            default_value = st.session_state.get(key)
-            val = vertical_slider(
-                label="",
-                key=key,
-                height=150,
-                step=0.1,
+    weekday_values = st.session_state.schedule_editor.get("weekday", [0.0] * 24)
+    weekend_values = st.session_state.schedule_editor.get("weekend", [0.0] * 24)
+
+    # ---------------------- UI SLIDERS ----------------------
+    st.markdown("### Schedule Profile")
+    col1, col2 = st.columns(2)
+
+    with col1:
+        st.markdown("#### Weekday")
+        weekday_result = []
+        for hour in range(24):
+            val = st.slider(
+                label=f"{hour:02d}:00",
                 min_value=0.0,
                 max_value=1.0,
-                default_value=default_value if isinstance(default_value, (float, int)) else 0.0,
-                slider_color="#4CAF50",
-                track_color="#e0e0e0",
-                thumb_color="#333",
-                value_always_visible=False
+                step=0.1,
+                value=weekday_values[hour],
+                key=f"weekday_slider_{hour}",
+                label_visibility="collapsed",  # cleaner interface
+                help=f"Weekday value for {hour:02d}:00"
             )
-            st.markdown(f"<div style='text-align:center; font-size:10px'>{hour:02d}:00</div>", unsafe_allow_html=True)
-            weekday_values.append(val)
+            weekday_result.append(val)
+            st.markdown(f"<small style='font-size:10px;'>{hour:02d}:00 – {val:.1f}</small>", unsafe_allow_html=True)
 
-    st.markdown("### Weekend Schedule")
-    cols = st.columns(24)
-    weekend_values = []
-    for hour, col in enumerate(cols):
-        with col:
-            key = f"weekend_{hour}"
-            default_value = st.session_state.get(key)
-            val = vertical_slider(
-                label="",
-                key=key,
-                height=150,
-                step=0.1,
+    with col2:
+        st.markdown("#### Weekend")
+        weekend_result = []
+        for hour in range(24):
+            val = st.slider(
+                label=f"{hour:02d}:00",
                 min_value=0.0,
                 max_value=1.0,
-                default_value=default_value if isinstance(default_value, (float, int)) else 0.0,
-                slider_color="#2196F3",
-                track_color="#e0e0e0",
-                thumb_color="#333",
-                value_always_visible=False
+                step=0.1,
+                value=weekend_values[hour],
+                key=f"weekend_slider_{hour}",
+                label_visibility="collapsed",
+                help=f"Weekend value for {hour:02d}:00"
             )
-            st.markdown(f"<div style='text-align:center; font-size:10px'>{hour:02d}:00</div>", unsafe_allow_html=True)
-            weekend_values.append(val)
+            weekend_result.append(val)
+            st.markdown(f"<small style='font-size:10px;'>{hour:02d}:00 – {val:.1f}</small>", unsafe_allow_html=True)
 
-    # Handle saving
+    # ---------------------- Handle Save ----------------------
     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("Schedule already exists.")
-            else:
-                schedules[name] = {
-                    "description": description,
-                    "weekday": weekday_values,
-                    "weekend": weekend_values
-                }
-                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}")
+        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.")
+        else:
+            schedules[name] = {
+                "description": description,
+                "weekday": weekday_result,
+                "weekend": weekend_result
+            }
+
+            # Reset editor to default state
+            st.session_state.schedule_editor = DEFAULT_STATE.copy()
+            st.success(f"Schedule '{name}' saved successfully.")
 
+    # ---------------------- Show Saved ----------------------
     st.markdown("### Saved Schedules")
     if schedules:
         display_schedules_table(schedules)
     else:
-        st.info("No schedules saved.")
+        st.info("No schedules saved yet.")
 
 def is_schedule_in_use(schedule_name: str) -> bool:
     """