Update app.py

app.py

@@ -131,9 +131,20 @@ def generate_full_schedule(creators_file_path, overnight_file_path, day_file_pat
         chatter_files = [overnight_data, day_data, prime_data]
         full_schedule = generate_schedule(chatter_files, creators_data)
 
-        return full_schedule
+        # Extract individual schedules
+        overnight_schedule = full_schedule["Overnight"]
+        day_schedule = full_schedule["Day"]
+        prime_schedule = full_schedule["Prime"]
+
+        print("DEBUG: Overnight Schedule:", overnight_schedule.head())
+        print("DEBUG: Day Schedule:", day_schedule.head())
+        print("DEBUG: Prime Schedule:", prime_schedule.head())
+
+        return overnight_schedule, day_schedule, prime_schedule
     except Exception as e:
-        return f"Error generating schedule: {e}"
+        print(f"DEBUG: Error in generate_full_schedule: {e}")
+        return f"Error generating schedule: {e}", None, None
+
 
 # Gradio Interface
 def app():
@@ -197,10 +208,10 @@ def app():
             prime_schedule_output = gr.Dataframe(label="Prime Schedule", interactive=False)
 
             generate_schedule_btn.click(
-                generate_full_schedule,
-                inputs=[creators_file, overnight_file, day_file, prime_file],
-                outputs=[overnight_schedule_output, day_schedule_output, prime_schedule_output]
-            )
+              generate_full_schedule,
+              inputs=[creators_file, overnight_file, day_file, prime_file],
+              outputs=[overnight_schedule_output, day_schedule_output, prime_schedule_output]
+)
 
     return interface