Update app.py

app.py

@@ -106,6 +106,27 @@ def update_assignments(assignments_df, creators_df):
         return "Assignments and Creator File updated successfully!", ZIP_FILE
     except Exception as e:
         return f"Error updating assignments: {e}", None
+        
+def update_and_download_schedule(overnight_df, day_df, prime_df):
+    try:
+        # Save updated schedules to the processed folder
+        overnight_path = os.path.join(PROCESSED_FOLDER, "Updated_Overnight_Schedule.xlsx")
+        day_path = os.path.join(PROCESSED_FOLDER, "Updated_Day_Schedule.xlsx")
+        prime_path = os.path.join(PROCESSED_FOLDER, "Updated_Prime_Schedule.xlsx")
+        
+        overnight_df.to_excel(overnight_path, index=False)
+        day_df.to_excel(day_path, index=False)
+        prime_df.to_excel(prime_path, index=False)
+
+        # Zip the updated files for download
+        with zipfile.ZipFile(ZIP_FILE, "w") as zipf:
+            zipf.write(overnight_path, arcname="Updated_Overnight_Schedule.xlsx")
+            zipf.write(day_path, arcname="Updated_Day_Schedule.xlsx")
+            zipf.write(prime_path, arcname="Updated_Prime_Schedule.xlsx")
+
+        return "Schedules updated and saved successfully!", ZIP_FILE
+    except Exception as e:
+        return f"Error updating schedules: {e}", None
 
 def generate_full_schedule(creators_file_path, overnight_file_path, day_file_path, prime_file_path):
     try:
@@ -203,15 +224,27 @@ def app():
             prime_file = gr.File(label="Prime File", type="binary")
 
             generate_schedule_btn = gr.Button("Generate Full Schedule")
-            overnight_schedule_output = gr.Dataframe(label="Overnight Schedule", interactive=False)
-            day_schedule_output = gr.Dataframe(label="Day Schedule", interactive=False)
-            prime_schedule_output = gr.Dataframe(label="Prime Schedule", interactive=False)
+            overnight_schedule_output = gr.Dataframe(label="Overnight Schedule", interactive=True)
+            day_schedule_output = gr.Dataframe(label="Day Schedule", interactive=True)
+            prime_schedule_output = gr.Dataframe(label="Prime Schedule", interactive=True)
+
+            update_schedule_btn = gr.Button("Update and Download Schedule")
+            download_schedule_btn = gr.File(label="Download Updated Schedules", value=ZIP_FILE)
+            update_status = gr.Textbox(label="Update Status", interactive=False)
 
+            # Generate Schedule
             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]
+            )
+
+            # Update and Download Schedule
+            update_schedule_btn.click(
+                update_and_download_schedule,
+                inputs=[overnight_schedule_output, day_schedule_output, prime_schedule_output],
+                outputs=[update_status, download_schedule_btn]
+            )
 
     return interface
 
@@ -224,3 +257,4 @@ if __name__ == "__main__":
 
 
 
+