Update app.py

app.py

@@ -92,21 +92,37 @@ def update_assignments(assignments_df, creators_df):
 
 def generate_full_schedule():
     try:
+        # Load chatter files as DataFrames
         chatter_files = [
-            os.path.join(PROCESSED_FOLDER, "Updated_overnight_file.xlsx"),
-            os.path.join(PROCESSED_FOLDER, "Updated_day_file.xlsx"),
-            os.path.join(PROCESSED_FOLDER, "Updated_prime_file.xlsx"),
+            pd.read_excel(os.path.join(PROCESSED_FOLDER, "Updated_overnight_file.xlsx")),
+            pd.read_excel(os.path.join(PROCESSED_FOLDER, "Updated_day_file.xlsx")),
+            pd.read_excel(os.path.join(PROCESSED_FOLDER, "Updated_prime_file.xlsx")),
         ]
-        creators_file = os.path.join(PROCESSED_FOLDER, "creators_file.xlsx")
 
-        if not all(os.path.exists(f) for f in chatter_files + [creators_file]):
-            return "Missing required files. Please complete the 'Generate Main Accounts' step first."
+        # Load processed creators file as a DataFrame
+        creators_file_path = os.path.join(PROCESSED_FOLDER, "creators_file.xlsx")
+        if not os.path.exists(creators_file_path):
+            return "Processed creator file is missing. Please complete the 'Generate Main Accounts' step first."
+
+        account_data = pd.read_excel(creators_file_path)
+
+        # Validate the required columns in creators file
+        if not {"Creator", "ActiveFans"}.issubset(account_data.columns):
+            raise KeyError("The processed creators file must contain 'Creator' and 'ActiveFans' columns.")
 
-        full_schedule = generate_schedule(chatter_files, creators_file)
-        return full_schedule
+        # Generate the schedule
+        full_schedule = generate_schedule(chatter_files, account_data)
+
+        # Optionally save the generated schedules for each shift
+        for shift_name, schedule_data in full_schedule.items():
+            shift_file = os.path.join(PROCESSED_FOLDER, f"Final_Schedule_{shift_name}.xlsx")
+            pd.DataFrame(schedule_data).to_excel(shift_file, index=False)
+
+        return "Full schedules generated successfully!"
     except Exception as e:
         return f"Error generating schedule: {e}"
 
+
 # Gradio Interface
 def app():
     with gr.Blocks() as interface: