nicer display, restart on select column

app.py

@@ -44,30 +44,17 @@ def create_app() -> gr.Blocks:
                 gr.Markdown("## Job Results")
                 results_table = gr.Dataframe(
                     value=get_results_table(),
-                    interactive=False,
+                    interactive=True,
+                    show_search="search",
+                    show_copy_button=True,
+                    show_fullscreen_button=True,
                     wrap=True,
-                    datatype=["str", "str", "str", "str", "str", "str", "html"]
+                    static_columns=list(range(7)),
+                    datatype=["str", "str", "str", "str", "str", "str", "html", "str"],
+                    elem_id="results_table"
                 )
                 refresh_btn = gr.Button("Refresh Results")
 
-        with gr.Row():
-            with gr.Column():
-                gr.Markdown("## Relaunch Individual Job")
-
-                # Load model-provider combinations
-
-                relaunch_model = gr.Dropdown(
-                    label="Model",
-                    choices=[],
-                    interactive=True
-                )
-                relaunch_provider = gr.Dropdown(
-                    label="Provider",
-                    choices=[],
-                    interactive=True
-                )
-                relaunch_btn = gr.Button("Relaunch Job", variant="secondary")
-
         def update_model_choices() -> gr.update:
             models_providers = load_models_providers(globals.LOCAL_CONFIG_FILE)
             model_choices = sorted(list(set([mp[0] for mp in models_providers])))
@@ -91,11 +78,6 @@ def create_app() -> gr.Blocks:
             outputs=output
         )
 
-        init_btn.click(
-            fn=update_model_choices,
-            outputs=relaunch_model
-        )
-
         launch_btn.click(
             fn=launch_jobs,
             outputs=output
@@ -106,16 +88,26 @@ def create_app() -> gr.Blocks:
             outputs=results_table
         )
 
-        # Update provider dropdown when model is selected
-        relaunch_model.change(
-            fn=update_provider_choices,
-            inputs=relaunch_model,
-            outputs=relaunch_provider
-        )
+        # Handle dataframe cell selection for relaunch
+        def handle_table_select(evt: gr.SelectData):
+            """Handle when a cell in the results table is clicked."""
+            print(f"[Relaunch] Cell selected - Row: {evt.index[0]}, Col: {evt.index[1]}, Value: {evt.value}")
+
+            # Check if the Actions column (index 7) was clicked
+            if evt.index[1] == 7:
+                # Get the full row data from the dataframe
+                df = get_results_table()
+                row_data = df.data.iloc[evt.index[0]]
+
+                model = row_data['Model']
+                provider = row_data['Provider']
+                print(f"[Relaunch] Relaunching job - Model: {model}, Provider: {provider}")
+
+                return run_single_job(model, provider, globals.TASKS)
 
-        relaunch_btn.click(
-            fn=run_single_job,
-            inputs=[relaunch_model, relaunch_provider],
+        results_table.select(
+            fn=handle_table_select,
+            inputs=[],
             outputs=output
         )
 

requirements.txt

@@ -1,4 +1,4 @@
-gradio>=4.0.0
+gradio>=5.49.0
 huggingface-hub>=0.20.0
 apscheduler>=3.10.0
 datasets>=2.14.0

utils/io.py

@@ -70,22 +70,21 @@ def load_models_providers(file_path: str = globals.LOCAL_CONFIG_FILE) -> List[Tu
 def save_results() -> None:
     """Persist job results to HuggingFace dataset."""
     try:
-        with globals.results_lock:
-            if not globals.job_results:
-                print("No results to save")
-                return
-
-            records = list(globals.job_results.values())
-            df = pd.DataFrame(records)
-            dataset = Dataset.from_pandas(df)
-
-            # Push to HuggingFace Hub
-            dataset.push_to_hub(
-                globals.RESULTS_DATASET_NAME,
-                token=os.getenv("HF_TOKEN"),
-                private=False
-            )
-            print(f"Saved {len(records)} results to dataset")
+        if not globals.job_results:
+            print("No results to save")
+            return
+
+        records = list(globals.job_results.values())
+        df = pd.DataFrame(records)
+        dataset = Dataset.from_pandas(df)
+
+        # Push to HuggingFace Hub
+        dataset.push_to_hub(
+            globals.RESULTS_DATASET_NAME,
+            token=os.getenv("HF_TOKEN"),
+            private=False
+        )
+        print(f"Saved {len(records)} results to dataset")
 
     except Exception as e:
         print(f"Error saving results to dataset: {e}")
@@ -132,41 +131,46 @@ def style_status(val):
 
 def get_results_table():
     """Return job results as a styled pandas DataFrame for Gradio DataFrame."""
-    with globals.results_lock:
-        if not globals.job_results:
-            return pd.DataFrame(columns=["Model", "Provider", "Last Run", "Status", "Current Score", "Previous Score", "Latest Job Id"])
-
-        table_data = []
-        for key, info in globals.job_results.items():
-            current_score = info.get("current_score", "N/A")
-            if current_score is not None and isinstance(current_score, (int, float)):
-                current_score = f"{current_score:.4f}"
-
-            previous_score = info.get("previous_score", "N/A")
-            if previous_score is not None and isinstance(previous_score, (int, float)):
-                previous_score = f"{previous_score:.4f}"
-
-            job_id = info.get("job_id", "N/A")
-            # Create a clickable link for the job ID
-            if job_id != "N/A":
-                job_url = f"https://hf.co/jobs/{globals.NAMESPACE}/{job_id}"
-                job_link = f'{job_id}: <a href="{job_url}" target="_blank">📄</a> '
-            else:
-                job_link = job_id
-
-            table_data.append([
-                info["model"],
-                info["provider"],
-                info["last_run"],
-                info["status"],
-                current_score,
-                previous_score,
-                job_link
-            ])
-
-        df = pd.DataFrame(table_data, columns=["Model", "Provider", "Last Run", "Status", "Current Score", "Previous Score", "Job Id and Logs"])
-
-        # Apply styling to the Status column
-        styled_df = df.style.map(style_status, subset=['Status'])
-        return styled_df
+    if not globals.job_results:
+        return pd.DataFrame(columns=["Model", "Provider", "Last Run", "Status", "Current Score", "Previous Score", "Latest Job Id"])
+
+    table_data = []
+    for key, info in globals.job_results.items():
+        current_score = info.get("current_score", "N/A")
+        if current_score is not None and isinstance(current_score, (int, float)):
+            current_score = f"{current_score:.4f}"
+
+        previous_score = info.get("previous_score", "N/A")
+        if previous_score is not None and isinstance(previous_score, (int, float)):
+            previous_score = f"{previous_score:.4f}"
+
+        job_id = info.get("job_id", "N/A")
+        # Create a clickable link for the job ID
+        if job_id != "N/A":
+            job_url = f"https://hf.co/jobs/{globals.NAMESPACE}/{job_id}"
+            job_link = f'{job_id}: <a href="{job_url}" target="_blank">📄</a> '
+        else:
+            job_link = job_id
+
+        # Create relaunch link with data attributes for JavaScript to access
+        model = info["model"]
+        provider = info["provider"]
+        relaunch_link = '🔄 Relaunch'
+
+        table_data.append([
+            model,
+            provider,
+            info["last_run"],
+            info["status"],
+            current_score,
+            previous_score,
+            job_link,
+            relaunch_link
+        ])
+
+    df = pd.DataFrame(table_data, columns=["Model", "Provider", "Last Run", "Status", "Current Score", "Previous Score", "Job Id and Logs", "Actions"])
+
+    # Apply styling to the Status column
+    styled_df = df.style.map(style_status, subset=['Status'])
+    return styled_df
 

utils/jobs.py

@@ -57,7 +57,7 @@ def extract_score_from_job(job_id: str) -> Optional[float]:
         return None
 
 
-def run_single_job(model: str, provider: str, tasks: str) -> Optional[str]:
+def run_single_job(model: str, provider: str, tasks: str = globals.TASKS) -> Optional[str]:
     """Run a single job for a model-provider combination."""
 
     if not model or not provider:
@@ -72,12 +72,11 @@ def run_single_job(model: str, provider: str, tasks: str) -> Optional[str]:
 
     # Check if job is already running
     key = globals.get_model_provider_key(model, provider)
-    with globals.results_lock:
-        if key in globals.job_results:
-            current_status = globals.job_results[key].get("status")
-            if current_status == "RUNNING":
-                print( f"Job for {model} on {provider} is already running. Please wait for it to complete.")
-                return -1
+    if key in globals.job_results:
+        current_status = globals.job_results[key].get("status")
+        if current_status == "RUNNING":
+            print( f"Job for {model} on {provider} is already running. Please wait for it to complete.")
+            return -1
 
     print(f"Starting job for model={model}, provider={provider}")
 
@@ -145,13 +144,11 @@ def launch_jobs(tasks: str = globals.TASKS, config_file: str = globals.LOCAL_CON
 def update_job_statuses() -> None:
     """Check and update the status of active jobs."""
     try:
-        with globals.results_lock:
-            keys = list(globals.job_results.keys())
+        keys = list(globals.job_results.keys())
 
         for key in keys:
             try:
-                with globals.results_lock:
-                    job_id = globals.job_results[key]["job_id"]
+                job_id = globals.job_results[key]["job_id"]
 
                 job_info = inspect_job(job_id=job_id, namespace=globals.NAMESPACE)
                 new_status = job_info.status.stage