Update helper.py

helper.py

@@ -14,6 +14,27 @@ def assign_main_accounts(creators_file, chatter_files):
     creators = pd.read_excel(creators_file)
     creators.columns = creators.columns.str.strip()
 
+    # Debugging: Check initial columns
+    print("DEBUG: Initial Columns in Creator File:", creators.columns)
+
+    column_mapping = {
+        "Creator": "Creator",
+        "Total earnings": "Total earnings",
+        "Subscription": "Subscription",
+        "Active Fans": "ActiveFans",
+        "Total active fans": "ActiveFans",
+    }
+    creators.rename(columns={k: v for k, v in column_mapping.items() if k in creators.columns}, inplace=True)
+
+    # Debugging: Check renamed columns
+    print("DEBUG: Renamed Columns in Creator File:", creators.columns)
+
+    required_columns = ["Creator", "Total earnings", "Subscription", "ActiveFans"]
+    missing_columns = [col for col in required_columns if col not in creators.columns]
+    if missing_columns:
+        raise KeyError(f"Missing required columns in creators file: {missing_columns}")
+
+    # Process creators file
     creators["Total earnings"] = creators["Total earnings"].replace("[\$,]", "", regex=True).astype(float)
     creators["Subscription"] = creators["Subscription"].replace("[\$,]", "", regex=True).astype(float)
     creators["ActiveFans"] = pd.to_numeric(creators["ActiveFans"], errors="coerce").fillna(0)
@@ -51,7 +72,6 @@ def assign_main_accounts(creators_file, chatter_files):
 
 
 
-
 def save_processed_files(assignments, output_dir):
     """
     Save processed files for main assignments, ensuring chatter names and main accounts are preserved correctly.