Fix Gradio Textbox copy-button compatibility in app startup

README.md

@@ -1,10 +1,8 @@
 ---
 title: Math Conjecture Trainer
-emoji: 🧮
 sdk: gradio
 app_file: app.py
 pinned: false
-sdk_version: 6.6.0
 ---
 
 # Math Conjecture Trainer Space
@@ -32,4 +30,4 @@ Set this in Space Settings > Variables and Secrets:
 
 - `configs/deepseek_math_sota.yaml`
 - base model default: `deepseek-ai/deepseek-math-v2`
-- output root: `workspace/runs/math-conjecture-sota`
+- output root: `workspace/runs/math-conjecture-sota`

app.py

@@ -4,6 +4,7 @@
 from __future__ import annotations
 
 import datetime as dt
+import inspect
 import json
 import os
 import shutil
@@ -105,6 +106,16 @@ def stream_subprocess(
     return ret
 
 
+def make_logs_textbox() -> gr.Textbox:
+    textbox_kwargs = {"label": "Logs", "lines": 24, "max_lines": 30}
+    textbox_init_params = inspect.signature(gr.Textbox.__init__).parameters
+    if "buttons" in textbox_init_params:
+        textbox_kwargs["buttons"] = ["copy"]
+    elif "show_copy_button" in textbox_init_params:
+        textbox_kwargs["show_copy_button"] = True
+    return gr.Textbox(**textbox_kwargs)
+
+
 def run_pipeline(
     hf_token: str,
     dataset_repo_id: str,
@@ -249,7 +260,7 @@ with gr.Blocks(title="Math Conjecture Trainer Space") as demo:
         eval_samples = gr.Slider(label="Eval Max Samples", minimum=50, maximum=1000, step=50, value=300)
     run_button = gr.Button("Start Train + Push")
     status = gr.Textbox(label="Status", value="Idle")
-    logs = gr.Textbox(label="Logs", lines=24, max_lines=30, show_copy_button=True)
+    logs = make_logs_textbox()
 
     run_button.click(
         fn=run_pipeline,