Update app.py

app.py

@@ -71,7 +71,6 @@ def generate_main_accounts():
     except Exception as e:
         return f"Error during main account generation: {e}", None, None
 
-
 def update_assignments(assignments_df, creators_df):
     try:
         print("DEBUG: Assignments DataFrame Columns:", assignments_df.columns)
@@ -85,10 +84,9 @@ def update_assignments(assignments_df, creators_df):
         creators_file_path = os.path.join(PROCESSED_FOLDER, "creators_file.xlsx")
         creators_df.to_excel(creators_file_path, index=False)
 
-        return "Assignments and Creator File updated successfully!"
+        return "Assignments and Creator File updated successfully!", PROCESSED_FOLDER
     except Exception as e:
-        return f"Error updating assignments: {e}"
-
+        return f"Error updating assignments: {e}", None
 
 def generate_full_schedule():
     try:
@@ -108,12 +106,11 @@ def generate_full_schedule():
             return "Missing required files. Please complete the 'Generate Main Accounts' step first."
 
         # Generate schedule
-        full_schedule = generate_schedule(chatter_files, creators_file)
+        full_schedule = generate_schedule(chatter_files, pd.read_excel(creators_file))
         return full_schedule
     except Exception as e:
         return f"Error generating schedule: {e}"
 
-
 # Gradio Interface
 def app():
     with gr.Blocks() as interface:
@@ -149,7 +146,8 @@ def app():
             creators_preview = gr.Dataframe(label="Processed Creator File Preview", interactive=True)
 
             generate_main_btn = gr.Button("Generate Main Accounts")
-            update_btn = gr.Button("Update Assignments")
+            update_btn = gr.Button("Update and Download Files")
+            download_zip_btn = gr.File(label="Download All Processed Files")
 
             generate_main_btn.click(
                 generate_main_accounts,
@@ -160,7 +158,7 @@ def app():
             update_btn.click(
                 update_assignments,
                 inputs=[assignments_preview, creators_preview],
-                outputs=[generate_main_status],
+                outputs=[generate_main_status, download_zip_btn],
             )
 
         with gr.Tab("Generate Full Schedule"):
@@ -178,3 +176,4 @@ if __name__ == "__main__":
 
 
 
+