Update app/internal_loads.py

app/internal_loads.py

@@ -25,6 +25,31 @@ from typing import Dict, List, Any, Optional, Tuple, Union
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
+# Custom CSS for vertical sliders
+st.markdown("""
+    <style>
+        .slider-container {
+            display: flex;
+            flex-direction: row;
+            align-items: flex-end;
+            justify-content: center;
+            gap: 8px;
+            padding-top: 30px;
+        }
+        .slider-item {
+            writing-mode: bt-lr;
+            transform: rotate(-90deg);
+            height: 150px;
+            width: 30px;
+        }
+        .hour-label {
+            text-align: center;
+            font-size: 12px;
+            margin-top: 5px;
+        }
+    </style>
+""", unsafe_allow_html=True)
+
 # Define constants
 LOAD_TYPES = ["People", "Lighting", "Equipment", "Ventilation & Infiltration", "Schedules"]
 
@@ -923,16 +948,15 @@ def display_ventilation_infiltration_tab():
             st.rerun()
 
 def display_schedules_tab():
-    """Display the schedules tab content with comprehensive schedule management."""
+    """Display the schedules tab content with responsive vertical slider interface."""
+    st.subheader("Schedule Management")
+    
     # Split the display into two columns
     col1, col2 = st.columns([3, 2])
     
     with col1:
         st.subheader("Saved Schedules")
-        
-        # Get schedules from session state
         schedules = st.session_state.project_data["internal_loads"]["schedules"]
-        
         if schedules:
             display_schedules_table(schedules)
         else:
@@ -941,140 +965,130 @@ def display_schedules_tab():
     with col2:
         st.subheader("Schedule Editor/Creator")
         
-        # Check if we have an editor state for schedules
+        # Initialize editor state
         if "schedule_editor" not in st.session_state:
             st.session_state.schedule_editor = {}
         
-        # Display the schedule editor form
-        with st.form("schedule_editor_form", clear_on_submit=True):
-            editor_state = st.session_state.get("schedule_editor", {})
-            is_edit = editor_state.get("is_edit", False)
-            
-            # Schedule name
-            name = st.text_input(
-                "Schedule Name",
-                value=editor_state.get("name", ""),
-                help="Enter a unique name for this schedule."
-            )
-            
-            # Description
-            description = st.text_area(
-                "Description",
-                value=editor_state.get("description", ""),
-                help="Brief description of this schedule."
+        editor_state = st.session_state.get("schedule_editor", {})
+        is_edit = editor_state.get("is_edit", False)
+        
+        # Schedule name and description
+        name = st.text_input(
+            "Schedule Name",
+            value=editor_state.get("name", ""),
+            help="Enter a unique name for this schedule."
+        )
+        
+        description = st.text_area(
+            "Description",
+            value=editor_state.get("description", ""),
+            help="Brief description of this schedule."
+        )
+        
+        # Template selection for new schedules
+        if not is_edit:
+            template = st.selectbox(
+                "Start from Template",
+                ["Custom"] + list(DEFAULT_SCHEDULE_TEMPLATES.keys()),
+                help="Select a template to start with, or choose Custom for blank schedule."
             )
-            
-            # Template selection for new schedules
-            if not is_edit:
-                template = st.selectbox(
-                    "Start from Template",
-                    ["Custom"] + list(DEFAULT_SCHEDULE_TEMPLATES.keys()),
-                    help="Select a template to start with, or choose Custom for blank schedule."
+        else:
+            template = "Custom"
+        
+        # Initialize schedule values
+        if template != "Custom" and not is_edit:
+            weekday_values = DEFAULT_SCHEDULE_TEMPLATES[template]["weekday"]
+            weekend_values = DEFAULT_SCHEDULE_TEMPLATES[template]["weekend"]
+        else:
+            weekday_values = editor_state.get("weekday", [0.0] * 24)
+            weekend_values = editor_state.get("weekend", [0.0] * 24)
+        
+        # Weekday schedule
+        st.write("**Weekday Schedule**")
+        st.markdown('<div class="slider-container">', unsafe_allow_html=True)
+        weekday_schedule = []
+        for hour in range(24):
+            with st.container():
+                val = st.slider(
+                    label=f"{hour:02d}:00",
+                    min_value=0.0,
+                    max_value=1.0,
+                    step=0.01,
+                    value=float(weekday_values[hour]),
+                    key=f"weekday_{hour}",
+                    help=f"Hour {hour}:00 weekday value",
+                    format="%.2f"
                 )
-            else:
-                template = "Custom"
-            
-            # Schedule values
-            st.write("**Hourly Schedule Values (0.0 - 1.0):**")
-            
-            # Initialize schedule values
-            if template != "Custom" and not is_edit:
-                weekday_values = DEFAULT_SCHEDULE_TEMPLATES[template]["weekday"]
-                weekend_values = DEFAULT_SCHEDULE_TEMPLATES[template]["weekend"]
-            else:
-                weekday_values = editor_state.get("weekday", [0.0] * 24)
-                weekend_values = editor_state.get("weekend", [0.0] * 24)
-            
-            # Weekday schedule
-            st.write("**Weekday Schedule:**")
-            weekday_cols = st.columns(6)
-            weekday_schedule = []
-            
-            for hour in range(24):
-                col_idx = hour % 6
-                with weekday_cols[col_idx]:
-                    value = st.number_input(
-                        f"H{hour:02d}",
-                        min_value=0.0,
-                        max_value=1.0,
-                        value=float(weekday_values[hour]),
-                        format="%.2f",
-                        key=f"weekday_{hour}",
-                        help=f"Hour {hour}:00 weekday value"
-                    )
-                    weekday_schedule.append(value)
-            
-            # Weekend schedule
-            st.write("**Weekend Schedule:**")
-            weekend_cols = st.columns(6)
-            weekend_schedule = []
-            
-            for hour in range(24):
-                col_idx = hour % 6
-                with weekend_cols[col_idx]:
-                    value = st.number_input(
-                        f"H{hour:02d}",
-                        min_value=0.0,
-                        max_value=1.0,
-                        value=float(weekend_values[hour]),
-                        format="%.2f",
-                        key=f"weekend_{hour}",
-                        help=f"Hour {hour}:00 weekend value"
-                    )
-                    weekend_schedule.append(value)
-            
-            # Submit buttons
-            col1, col2 = st.columns(2)
-            with col1:
-                submit_label = "Update Schedule" if is_edit else "Add Schedule"
-                submit = st.form_submit_button(submit_label)
-            
-            with col2:
-                cancel = st.form_submit_button("Cancel")
+                st.markdown(f'<div class="hour-label">{hour:02d}:00</div>', unsafe_allow_html=True)
+                weekday_schedule.append(val)
+        st.markdown('</div>', unsafe_allow_html=True)
         
-        # Handle form submission
-        if submit:
-            # Validate inputs
-            if not name.strip():
-                st.error("Schedule name is required.")
-            elif name in schedules and not is_edit:
-                st.error("A schedule with this name already exists.")
-            else:
-                # Create schedule data
-                schedule_data = {
-                    "description": description,
-                    "weekday": weekday_schedule,
-                    "weekend": weekend_schedule
-                }
-                
-                # Check if editing or adding new
-                if is_edit and st.session_state.schedule_editor.get("original_name"):
-                    # Update existing schedule
-                    original_name = st.session_state.schedule_editor["original_name"]
+        # Weekend schedule
+        st.write("**Weekend Schedule**")
+        st.markdown('<div class="slider-container">', unsafe_allow_html=True)
+        weekend_schedule = []
+        for hour in range(24):
+            with st.container():
+                val = st.slider(
+                    label=f"{hour:02d}:00",
+                    min_value=0.0,
+                    max_value=1.0,
+                    step=0.01,
+                    value=float(weekend_values[hour]),
+                    key=f"weekend_{hour}",
+                    help=f"Hour {hour}:00 weekend value",
+                    format="%.2f"
+                )
+                st.markdown(f'<div class="hour-label">{hour:02d}:00</div>', unsafe_allow_html=True)
+                weekend_schedule.append(val)
+        st.markdown('</div>', unsafe_allow_html=True)
+        
+        # Preview chart
+        st.write("**Schedule Preview**")
+        display_schedule_chart(f"Preview: {name or 'New Schedule'}", {
+            "weekday": weekday_schedule,
+            "weekend": weekend_schedule
+        })
+        
+        # Action buttons
+        col1, col2 = st.columns(2)
+        with col1:
+            submit_label = "Update Schedule" if is_edit else "Add Schedule"
+            if st.button(submit_label):
+                # Validate inputs
+                if not name.strip():
+                    st.error("Schedule name is required.")
+                elif name in schedules and not is_edit:
+                    st.error("A schedule with this name already exists.")
+                else:
+                    # Create schedule data
+                    schedule_data = {
+                        "description": description,
+                        "weekday": weekday_schedule,
+                        "weekend": weekend_schedule
+                    }
                     
-                    # If name changed, remove old and add new
-                    if original_name != name and original_name in schedules:
-                        del schedules[original_name]
+                    # Update or add schedule
+                    if is_edit and st.session_state.schedule_editor.get("original_name"):
+                        original_name = st.session_state.schedule_editor["original_name"]
+                        if original_name != name and original_name in schedules:
+                            del schedules[original_name]
+                        schedules[name] = schedule_data
+                        st.success(f"Schedule '{name}' updated successfully!")
+                    else:
+                        schedules[name] = schedule_data
+                        st.success(f"Schedule '{name}' added successfully!")
                     
-                    schedules[name] = schedule_data
-                    st.success(f"Schedule '{name}' updated successfully!")
-                else:
-                    # Add new schedule
-                    schedules[name] = schedule_data
-                    st.success(f"Schedule '{name}' added successfully!")
-                
-                # Clear editor state
+                    # Clear editor state
+                    st.session_state.schedule_editor = {}
+                    st.session_state.module_rerun_flags["internal_loads"] = True
+                    st.rerun()
+        
+        with col2:
+            if st.button("Cancel"):
                 st.session_state.schedule_editor = {}
-                # [FIX]: Use module_rerun_flags instead of internal_loads_rerun_pending
                 st.session_state.module_rerun_flags["internal_loads"] = True
                 st.rerun()
-        
-        if cancel:
-            # Clear editor state
-            st.session_state.schedule_editor = {}
-            # [FIX]: Use module_rerun_flags instead of internal_loads_rerun_pending
-            st.session_state.module_rerun_flags["internal_loads"] = True
-            st.rerun()
 
 def is_schedule_in_use(schedule_name: str) -> bool:
     """