Fix hub ui

ui.py

@@ -283,7 +283,6 @@ def build_interface() -> gr.Blocks:
         # --- Event Wiring ---
 
         # 1. Initialization
-        # Note: init_session now returns updates for both model_name_input and org_dropdown
         demo.load(lock_ui, outputs=action_buttons + hub_inputs).then(
             fn=UIController.init_session,
             inputs=None, # Gradio automatically injects profile and token
@@ -321,13 +320,13 @@ def build_interface() -> gr.Blocks:
         
         # 3a. Training
         train_run_event = run_btn.click(
-            fn=lambda: (
-                gr.update(visible=False), 
-                gr.update(interactive=False), # Lock Reload
-                gr.update(interactive=False), # Lock Eval
-                [gr.update(interactive=False) for _ in hub_inputs], # Lock Hub
-                gr.update(visible=True)       # Show Stop
-            ),
+            fn=lambda: [
+                gr.update(visible=False),                     # run_btn
+                gr.update(interactive=False),                 # reload_btn
+                gr.update(interactive=False)                  # eval_btn
+            ] + [gr.update(interactive=False) for _ in hub_inputs] + [ # Unpack Hub Inputs list
+                gr.update(visible=True)                       # stop_btn
+            ],
             outputs=[run_btn, reload_btn, eval_btn, *hub_inputs, stop_btn]
         )
         train_run_event = train_run_event.then(