Update app.py

app.py

@@ -53,15 +53,29 @@ def generate_main_accounts(updated_data=None):
 
     try:
         if updated_data is not None:
-            # If updated data is provided, save it back as the new assignments
+            # Ensure updated_data is in the correct format
+            if isinstance(updated_data, pd.DataFrame):
+                updated_data = updated_data.to_dict(orient="records")
+            
+            # Validate updated data structure
+            for record in updated_data:
+                if not isinstance(record, dict):
+                    raise ValueError(f"Invalid record in updated data: {record}")
+
+            # Group updated data by shift
             assignments = {
-                shift: pd.DataFrame([record for record in updated_data if record["Shift"] == shift.capitalize()])
+                shift.lower(): [
+                    {key: record[key] for key in record if key != "Shift"}  # Remove the Shift column
+                    for record in updated_data if record["Shift"] == shift.capitalize()
+                ]
                 for shift in ["overnight", "day", "prime"]
             }
+
+            # Save updated assignments
             save_processed_files(assignments, PROCESSED_FOLDER)
-            return "Assignments updated successfully!", pd.concat(assignments.values())
+            return "Assignments updated successfully!", pd.DataFrame(updated_data)
 
-        # Generate assignments for preview
+        # Generate new assignments if no updated data is provided
         main_assignments = assign_main_accounts(creators_file, chatter_files)
         save_processed_files(main_assignments, PROCESSED_FOLDER)
 
@@ -80,6 +94,7 @@ def generate_main_accounts(updated_data=None):
 
 
 
+
 def generate_full_schedule():
     chatter_files = [
         os.path.join(PROCESSED_FOLDER, "Updated_overnight_file.xlsx"),
@@ -153,6 +168,7 @@ def app():
 
 
 
+
 # Launch Gradio App
 if __name__ == "__main__":
     app().launch()