Update app.py

app.py

@@ -107,29 +107,33 @@ def update_assignments(assignments_df, creators_df):
     except Exception as e:
         return f"Error updating assignments: {e}", None
 
-def generate_full_schedule():
+def generate_full_schedule(creators_file, overnight_file, day_file, prime_file):
     try:
+        # Read uploaded files directly
         chatter_files = [
-            os.path.join(PROCESSED_FOLDER, "Updated_overnight_file.xlsx"),
-            os.path.join(PROCESSED_FOLDER, "Updated_day_file.xlsx"),
-            os.path.join(PROCESSED_FOLDER, "Updated_prime_file.xlsx"),
+            pd.read_excel(overnight_file),
+            pd.read_excel(day_file),
+            pd.read_excel(prime_file),
         ]
-        creators_file = os.path.join(PROCESSED_FOLDER, "creators_file.xlsx")
+        creators_data = pd.read_excel(creators_file)
 
-        # Debugging: Log file paths and existence
-        for f in chatter_files + [creators_file]:
-            print(f"DEBUG: File path: {f}")
-            print(f"DEBUG: File exists: {os.path.exists(f)}")
+        # 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)
 
-        if not all(os.path.exists(f) for f in chatter_files + [creators_file]):
-            return "Missing required files. Please complete the 'Generate Main Accounts' step first."
+        # Validate data structure
+        required_creator_columns = {"Creator", "ActiveFans"}
+        if not required_creator_columns.issubset(creators_data.columns):
+            return f"Error: Creators file must contain the columns {required_creator_columns}."
 
         # Generate schedule
-        full_schedule = generate_schedule(chatter_files, pd.read_excel(creators_file))
+        full_schedule = generate_schedule(chatter_files, creators_data)
         return full_schedule
     except Exception as e:
         return f"Error generating schedule: {e}"
 
+
 # Gradio Interface
 def app():
     with gr.Blocks() as interface:
@@ -181,12 +185,20 @@ def app():
             )
 
         with gr.Tab("Generate Full Schedule"):
-            generate_schedule_btn = gr.Button("Generate Full Schedule")
-            full_schedule_output = gr.Textbox(label="Full Schedule", interactive=False)
+    creators_file = gr.File(label="Creators File", type="binary")
+    overnight_file = gr.File(label="Overnight File", type="binary")
+    day_file = gr.File(label="Day File", type="binary")
+    prime_file = gr.File(label="Prime File", type="binary")
+    
+    generate_schedule_btn = gr.Button("Generate Full Schedule")
+    full_schedule_output = gr.Textbox(label="Full Schedule", interactive=False)
+
+    generate_schedule_btn.click(
+        generate_full_schedule,
+        inputs=[creators_file, overnight_file, day_file, prime_file],
+        outputs=[full_schedule_output]
+    )
 
-            generate_schedule_btn.click(
-                generate_full_schedule, inputs=[], outputs=[full_schedule_output]
-            )
 
     return interface