Update helper.py

helper.py

@@ -8,6 +8,9 @@ import pandas as pd
 from sklearn.preprocessing import MinMaxScaler
 
 def assign_main_accounts(creators_file, chatter_files):
+    """
+    Process creators and chatter files to assign main accounts to chatters and prepare data for scheduling.
+    """
     # Load and process creators data
     creators = pd.read_excel(creators_file)
     creators.columns = creators.columns.str.strip()  # Clean up column names
@@ -37,8 +40,15 @@ def assign_main_accounts(creators_file, chatter_files):
     # Sort creators by rank
     creators = creators.sort_values(by="Rank").reset_index(drop=True)
 
+    # Save processed creator file with Creator and ActiveFans only
+    processed_creator_file = os.path.join(PROCESSED_FOLDER, "creators_file.xlsx")
+    creators[["Creator", "Total active fans"]].rename(
+        columns={"Total active fans": "ActiveFans"}
+    ).to_excel(processed_creator_file, index=False)
+    print(f"DEBUG: Processed creators file saved to {processed_creator_file}")
+
     # Assign main accounts to chatters
-    assignments = {}
+    updated_chatter_files = []
     for idx, chatter_file in enumerate(chatter_files):
         shift_name = ["overnight", "day", "prime"][idx]
         chatters = pd.read_excel(chatter_file)
@@ -55,15 +65,11 @@ def assign_main_accounts(creators_file, chatter_files):
         top_creators = creators.iloc[:num_chatters]
         chatters["Main Account"] = top_creators["Creator"].values
 
-        # Save assignment
-        assignments[shift_name] = chatters.to_dict(orient="records")
+        # Append updated chatter data
+        updated_chatter_files.append(chatters)
 
-    # Add processed creators for schedule generation
-    assignments["creators"] = creators[["Creator", "Total active fans"]].rename(
-        columns={"Total active fans": "ActiveFans"}
-    ).to_dict(orient="records")
+    return updated_chatter_files, processed_creator_file
 
-    return assignments