Update app.py

app.py

@@ -49,10 +49,18 @@ def generate_main_accounts():
             creators_file, chatter_files
         )
 
+        # Add the `Shift` column to the combined assignments
+        shifts = ["Overnight", "Day", "Prime"]
+        for idx, chatter_df in enumerate(updated_chatter_files):
+            chatter_df["Shift"] = shifts[idx]
+            updated_chatter_files[idx] = chatter_df
+
+        combined_assignments = pd.concat(updated_chatter_files)
+
         # Save updated files
         for idx, chatter_df in enumerate(updated_chatter_files):
             chatter_df.to_excel(
-                os.path.join(PROCESSED_FOLDER, f"Updated_{['overnight', 'day', 'prime'][idx]}_file.xlsx"),
+                os.path.join(PROCESSED_FOLDER, f"Updated_{shifts[idx].lower()}_file.xlsx"),
                 index=False
             )
         processed_creator_file.to_excel(
@@ -63,8 +71,11 @@ def generate_main_accounts():
     except Exception as e:
         return f"Error during main account generation: {e}", None, None
 
+
 def update_assignments(assignments_df, creators_df):
     try:
+        print("DEBUG: Assignments DataFrame Columns:", assignments_df.columns)
+
         shifts = ["Overnight", "Day", "Prime"]
         for shift in shifts:
             shift_data = assignments_df[assignments_df["Shift"] == shift]
@@ -78,6 +89,7 @@ def update_assignments(assignments_df, creators_df):
     except Exception as e:
         return f"Error updating assignments: {e}"
 
+
 def generate_full_schedule():
     try:
         chatter_files = [