Update app.py

app.py

@@ -121,46 +121,59 @@ import gradio as gr
 
 def app():
     with gr.Blocks() as interface:
+        with gr.Tab("Login"):
+            username = gr.Textbox(label="Username")
+            password = gr.Textbox(label="Password", type="password")
+            login_btn = gr.Button("Login")
+            login_status = gr.Textbox(label="Login Status", interactive=False)
+
+            login_btn.click(login, inputs=[username, password], outputs=[login_status])
+
+        with gr.Tab("Upload Files"):
+            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")
+            upload_btn = gr.Button("Upload Files")
+            upload_status = gr.Textbox(label="Upload Status", interactive=False)
+
+            upload_btn.click(
+                upload_files,
+                inputs=[creators_file, overnight_file, day_file, prime_file],
+                outputs=[upload_status],
+            )
+
         with gr.Tab("Generate Main Accounts"):
-            status = gr.Textbox(label="Status", interactive=False)
+            generate_status = gr.Textbox(label="Status", interactive=False)
             assignments_preview = gr.Dataframe(label="Main Account Assignments Preview", interactive=True)
-            creators_preview = gr.Dataframe(label="Creators File Preview", interactive=True)
+            creators_preview = gr.Dataframe(label="Processed Creator File Preview", interactive=True)
 
             generate_btn = gr.Button("Generate Main Accounts")
             update_btn = gr.Button("Update Assignments")
 
-            # Generate Main Accounts Logic
-            def generate_main():
-                status, assignments, creators = generate_main_accounts()
-                return status, assignments, creators
-
-            # Bind Generate Button
             generate_btn.click(
                 generate_main,
                 inputs=[],
-                outputs=[status, assignments_preview, creators_preview],
+                outputs=[generate_status, assignments_preview, creators_preview],
             )
-
-            # Update Assignments Logic
-            def update_assignments(assignments, creators):
-                # Save updated assignments and creators back to processed files
-                for idx, shift in enumerate(["Overnight", "Day", "Prime"]):
-                    shift_data = assignments[assignments["Shift"] == shift]
-                    shift_data.to_excel(os.path.join(PROCESSED_FOLDER, f"Updated_{shift.lower()}_file.xlsx"), index=False)
-
-                creators.to_excel(os.path.join(PROCESSED_FOLDER, "creators_file.xlsx"), index=False)
-                return "Assignments and Creator File updated successfully!"
-
-            # Bind Update Button
             update_btn.click(
                 update_assignments,
                 inputs=[assignments_preview, creators_preview],
-                outputs=[status],
+                outputs=[generate_status],
+            )
+
+        with gr.Tab("Generate Full Schedule"):
+            generate_schedule_btn = gr.Button("Generate Full Schedule")
+            full_schedule_output = gr.Dataframe(label="Full Schedule", interactive=False)
+
+            generate_schedule_btn.click(
+                generate_full_schedule, inputs=[], outputs=[full_schedule_output]
             )
 
     return interface
 
+
+# Run App
 if __name__ == "__main__":
     app().launch()
 
-