Update app.py

app.py

@@ -107,32 +107,30 @@ def update_assignments(assignments_df, creators_df):
     except Exception as e:
         return f"Error updating assignments: {e}", None
 
-def generate_full_schedule(creators_file, overnight_file, day_file, prime_file):
+def generate_full_schedule(creators_file_path, overnight_file_path, day_file_path, prime_file_path):
     try:
-        # Read uploaded files directly
-        chatter_files = [
-            pd.read_excel(overnight_file),
-            pd.read_excel(day_file),
-            pd.read_excel(prime_file),
-        ]
-        creators_data = pd.read_excel(creators_file)
-
-        # Debugging: Log the columns of the files
-        print("DEBUG: Creators File Columns:", creators_data.columns)
-        for idx, chatter_df in enumerate(chatter_files):
-            print(f"DEBUG: Chatter File {idx + 1} Columns:", chatter_df.columns)
+        # Log file paths for debugging
+        print(f"DEBUG: Creators File Path: {creators_file_path}")
+        print(f"DEBUG: Overnight File Path: {overnight_file_path}")
+        print(f"DEBUG: Day File Path: {day_file_path}")
+        print(f"DEBUG: Prime File Path: {prime_file_path}")
+
+        # Read files
+        creators_data = pd.read_excel(creators_file_path)
+        overnight_data = pd.read_excel(overnight_file_path)
+        day_data = pd.read_excel(day_file_path)
+        prime_data = pd.read_excel(prime_file_path)
 
-        # Rename "Creator" to "Account" to match the expected column in the schedule function
-        if "Creator" in creators_data.columns:
-            creators_data.rename(columns={"Creator": "Account"}, inplace=True)
+        print("DEBUG: Creators File Columns:", creators_data.columns)
 
-        # Validate data structure
-        required_creator_columns = {"Account", "ActiveFans"}
-        if not required_creator_columns.issubset(creators_data.columns):
-            return f"Error: Creators file must contain the columns {required_creator_columns}."
+        # Validate structure
+        if not {"Creator", "ActiveFans"}.issubset(creators_data.columns):
+            raise KeyError("The account data must contain 'Creator' and 'ActiveFans' columns.")
 
-        # Generate schedule
+        # Pass data to generate_schedule
+        chatter_files = [overnight_data, day_data, prime_data]
         full_schedule = generate_schedule(chatter_files, creators_data)
+
         return full_schedule
     except Exception as e:
         return f"Error generating schedule: {e}"