Split testing and tuning

ui.py

@@ -144,7 +144,7 @@ def _render_dataset_tab(engine_state):
     }
 
 def _render_training_tab(engine_state):
-    with gr.TabItem("2. Training"):
+    with gr.TabItem("2. Training & Eval"):
         gr.Markdown("### 🚀 Fine-Tuning Configuration")
         with gr.Group():
             gr.Markdown("**Hyperparameters**")
@@ -171,6 +171,7 @@ def _render_training_tab(engine_state):
 
     return {
         "params": [param_epochs, param_lr, param_test_size, param_shuffle, param_model],
+        "eval_params": [param_test_size, param_shuffle, param_model],
         "buttons": [run_training_btn, stop_training_btn, clear_reload_btn, run_eval_btn],
         "outputs": [output_display, loss_plot],
         "model_input": param_model # specifically needed for initialization
@@ -219,7 +220,8 @@ def build_interface() -> gr.Blocks:
         # Helpers for UI State
         # 'action_buttons' now ONLY contains buttons that should always be enabled after a process
         # Zip and Push buttons are excluded here because their state depends on has_model_tuned
-        action_buttons = [train_ui["buttons"][2], train_ui["buttons"][0]] # [Reload, Run]
+        run_btn, stop_btn, reload_btn, eval_btn = train_ui["buttons"]
+        action_buttons = [reload_btn, run_btn, eval_btn] 
         
         repo_input = export_ui["hub_controls"][0]
         push_btn = export_ui["hub_controls"][1]
@@ -269,17 +271,18 @@ def build_interface() -> gr.Blocks:
             outputs=[data_ui["import_status"]]
         )
 
-        # 3. Training Tab
-        run_btn, stop_btn, reload_btn = train_ui["buttons"]
+        # 3. Training & Eval Tab
         
+        # 3a. Training
         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), # Lock Zip 
                 gr.update(visible=True)       # Show Stop
             ),
-            outputs=[run_btn, reload_btn, zip_btn, stop_btn]
+            outputs=[run_btn, reload_btn, eval_btn, zip_btn, stop_btn]
         ).then(
             fn=UIController.run_training,
             inputs=[engine_state, *train_ui["params"]],
@@ -288,9 +291,10 @@ def build_interface() -> gr.Blocks:
             fn=lambda: (
                 gr.update(visible=True), 
                 gr.update(interactive=True), 
+                gr.update(interactive=True),
                 gr.update(visible=False)
             ),
-            outputs=[run_btn, reload_btn, stop_btn]
+            outputs=[run_btn, reload_btn, eval_btn, stop_btn]
         ).then(
             # Final check determines if Zip/Push should unlock
             fn=UIController.update_hub_interactive,
@@ -298,6 +302,29 @@ def build_interface() -> gr.Blocks:
             outputs=[repo_input, push_btn, zip_btn] 
         )
 
+        # 3b. Evaluation
+        eval_btn.click(
+            fn=lambda: (
+                gr.update(interactive=False), # Lock Run
+                gr.update(interactive=False), # Lock Reload
+                gr.update(visible=False),     # Hide self (optional, or lock)
+                gr.update(visible=True)       # Show Stop
+            ),
+            outputs=[run_btn, reload_btn, eval_btn, stop_btn]
+        ).then(
+            fn=UIController.run_evaluation,
+            inputs=[engine_state, *train_ui["eval_params"]],
+            outputs=[train_ui["outputs"][0]] # Output only to log, not plot
+        ).then(
+            fn=lambda: (
+                gr.update(interactive=True), 
+                gr.update(interactive=True), 
+                gr.update(visible=True),
+                gr.update(visible=False)
+            ),
+            outputs=[run_btn, reload_btn, eval_btn, stop_btn]
+        )
+
         stop_btn.click(fn=UIController.stop_process, inputs=[engine_state])
 
         reload_btn.click(lock_ui, outputs=action_buttons + [push_btn, zip_btn]).then(
@@ -337,4 +364,4 @@ def build_interface() -> gr.Blocks:
             outputs=[repo_input, push_btn, zip_btn]
         )
 
-    return demo
+    return demo