Update app.py

app.py

@@ -441,57 +441,70 @@ def nurse_scheduling_app(
 
 # Create Gradio interface
 def create_interface():
-    with gr.Blocks(title="Advanced Nurse Scheduling System") as demo:
-        gr.Markdown("# Advanced Nurse Scheduling System")
+    with gr.Blocks() as interface:
+        gr.Markdown("# Staff Scheduling Optimizer")
         
         with gr.Row():
             with gr.Column():
-                csv_file = gr.File(label="Upload Demand Data CSV")
+                # Left panel for inputs
+                gr.Markdown("### Clinic Parameters")  # Use Markdown instead of Group label
+                with gr.Group():  # Group without label
+                    csv_input = gr.File(label="Upload CSV")
+                    beds_per_staff = gr.Number(label="Beds per Staff", value=3)
+                    max_hours_per_staff = gr.Number(label="Maximum monthly hours", value=160)
+                    hours_per_cycle = gr.Number(label="Hours per Cycle", value=4)
+                    rest_days_per_week = gr.Number(label="Rest Days per Week", value=2)
                 
-                with gr.Group(label="Clinic Parameters"):
-                    clinic_start = gr.Textbox(label="Clinic Start Time (HH:MM)", value="08:00")
-                    clinic_end = gr.Textbox(label="Clinic End Time (HH:MM)", value="20:00")
-                    cycle_duration = gr.Number(label="Cycle Duration (hours)", value=5)
-                
-                with gr.Group(label="Staff Constraints"):
-                    beds_per_staff = gr.Slider(label="Beds Per Staff", minimum=1, maximum=10, value=3, step=1)
-                    max_shift_duration = gr.Slider(label="Maximum Shift Duration (hours)", minimum=4, maximum=12, value=12, step=1)
-                    handover_overlap = gr.Slider(label="Handover Overlap (minutes)", minimum=0, maximum=60, value=30, step=5)
-                    max_monthly_hours = gr.Slider(label="Maximum Monthly Hours", minimum=100, maximum=300, value=234, step=1)
-                
-                submit_btn = gr.Button("Generate Schedule")
+                gr.Markdown("### Time Parameters")  # Use Markdown for section headers
+                with gr.Group():
+                    clinic_start = gr.Dropdown(
+                        label="Clinic Start Hour",
+                        choices=am_pm_times,
+                        value="08:00 AM"
+                    )
+                    clinic_end = gr.Dropdown(
+                        label="Clinic End Hour",
+                        choices=am_pm_times,
+                        value="08:00 PM"
+                    )
+                    overlap_time = gr.Number(label="Overlap Time", value=0)
+                    max_start_time_change = gr.Number(label="Max Start Time Change", value=2)
             
             with gr.Column():
-                summary_output = gr.Markdown(label="Summary")
-                gantt_output = gr.Plot(label="Gantt Chart")
-                schedule_output = gr.Dataframe(label="Schedule")
-                csv_output = gr.File(label="Download Schedule CSV")
+                # Right panel for outputs
+                with gr.Tabs():
+                    with gr.TabItem("Schedule"):
+                        schedule_output = gr.Dataframe()
+                    with gr.TabItem("Visualization"):
+                        gantt_chart = gr.Plot()
+                    with gr.TabItem("Statistics"):
+                        stats_output = gr.Markdown()
+        
+        optimize_btn = gr.Button("Optimize Schedule", variant="primary")
         
-        submit_btn.click(
+        # Connect the button to your optimization function
+        optimize_btn.click(
             fn=nurse_scheduling_app,
             inputs=[
-                csv_file, 
-                beds_per_staff, 
-                max_shift_duration, 
-                handover_overlap, 
-                max_monthly_hours,
+                csv_input,
+                beds_per_staff,
+                max_hours_per_staff,
+                hours_per_cycle,
+                rest_days_per_week,
                 clinic_start,
                 clinic_end,
-                cycle_duration
+                overlap_time,
+                max_start_time_change
             ],
-            outputs=[
-                summary_output,
-                gantt_output,
-                schedule_output,
-                csv_output
-            ]
+            outputs=[schedule_output, gantt_chart, stats_output]
         )
-        
-    return demo
+    
+    return interface
 
-# Create and launch the interface
-interface = create_interface()
+# Define your time options
+am_pm_times = [f"{i:02d}:00 AM" for i in range(1, 13)] + [f"{i:02d}:00 PM" for i in range(1, 13)]
 
-# For local testing
+# Launch the interface
 if __name__ == "__main__":
-    interface.launch()
+    interface = create_interface()
+    interface.launch(share=True)