DataSense progress bar HTML, replace slider and model jargon

app.py

@@ -91,7 +91,37 @@ CUSTOM_CSS = """
   color: var(--ds-accent);
   margin: 0 0 0.35rem 0;
 }
-#ds-progress-bar input { pointer-events: none; }
+.ds-progress-wrap {
+  margin: 0 0 1rem 0;
+  padding: 0.85rem 1rem;
+  background: #0f1419;
+  border: 1px solid #2a3544;
+  border-radius: 10px;
+}
+.ds-progress-text {
+  font-family: 'IBM Plex Mono', monospace;
+  font-size: 0.82rem;
+  color: #3ecfae;
+  margin-bottom: 0.55rem;
+  line-height: 1.4;
+}
+.ds-progress-text .ds-pct {
+  color: #8b9cb3;
+  font-size: 0.75rem;
+}
+.ds-progress-track {
+  height: 6px;
+  background: #1a2330;
+  border-radius: 999px;
+  overflow: hidden;
+}
+.ds-progress-fill {
+  height: 100%;
+  background: linear-gradient(90deg, #1f6f5c, #3ecfae);
+  border-radius: 999px;
+  transition: width 0.35s ease;
+}
+.ds-progress-wrap.ds-idle .ds-progress-track { display: none; }
 #ds-results .tabitem { padding-top: 0.75rem !important; }
 .ds-answer-card {
   background: linear-gradient(145deg, #122a2a 0%, #151c26 100%);
@@ -245,15 +275,30 @@ def _wrap_trace(trace: str) -> str:
     return f'<div class="ds-trace-wrap">\n\n{trace}\n\n</div>'
 
 
-def _progress_update(pct: int, label: str, trace: str = "", answer: str = ""):
+def _progress_html(pct: int | None, label: str) -> str:
+    safe_label = html.escape(label)
+    if pct is None:
+        return f'<div class="ds-progress-wrap ds-idle"><div class="ds-progress-text">{safe_label}</div></div>'
+    pct = max(0, min(100, pct))
     return (
-        gr.update(value=max(0, min(100, pct))),
-        label,
+        f'<div class="ds-progress-wrap">'
+        f'<div class="ds-progress-text">{safe_label} <span class="ds-pct">{pct}%</span></div>'
+        f'<div class="ds-progress-track"><div class="ds-progress-fill" style="width:{pct}%"></div></div>'
+        f"</div>"
+    )
+
+
+def _progress_update(pct: int | None, label: str, trace: str = "", answer: str = ""):
+    return (
+        _progress_html(pct, label),
         _wrap_trace(trace),
         answer,
     )
 
 
+IDLE_PROGRESS_HTML = _progress_html(None, "Ready — click Run DataSense")
+
+
 @spaces.GPU(duration=180)
 def run_task(
     data_mode: str,
@@ -264,10 +309,10 @@ def run_task(
     progress=gr.Progress(),
 ):
     if not task.strip():
-        yield _progress_update(0, "Waiting for a question", "", '<div class="ds-answer-empty">Enter a task question.</div>')
+        yield _progress_update(None, "Enter a question to run DataSense", "", '<div class="ds-answer-empty">Enter a task question.</div>')
         return
 
-    yield _progress_update(5, "Resolving dataset…")
+    yield _progress_update(5, "Reading your dataset…")
     data_path = _resolve_data_path(data_mode, dataset_name, upload_file)
     if data_path is None:
         msg = "⚠️ Upload a `.csv` or `.xlsx` file first." if data_mode == "Upload your file" else f"⚠️ Dataset not found: {dataset_name}"
@@ -275,11 +320,11 @@ def run_task(
         return
 
     try:
-        yield _progress_update(12, "Loading Gemma-4 + SFT LoRA on GPU…")
-        progress(0.15, desc="Loading model…")
+        yield _progress_update(12, "Loading DataSense on GPU…")
+        progress(0.15, desc="Loading DataSense…")
         model, tokenizer = _load_model()
 
-        yield _progress_update(22, "Starting agent loop…")
+        yield _progress_update(22, "DataSense is starting…")
         agent_stream = run_agent(
             model,
             tokenizer,
@@ -296,7 +341,7 @@ def run_task(
                 pct = int(22 + (73 * step / max(total, 1)))
                 yield _progress_update(
                     pct,
-                    f"Step {step}/{total} — generating code & executing…",
+                    f"DataSense · step {step}/{total}",
                     trace_md,
                     "",
                 )
@@ -305,7 +350,7 @@ def run_task(
                 answer_html = _format_answer_html(result.get("answer", ""), result.get("summary", ""))
                 yield _progress_update(
                     100,
-                    "Complete",
+                    "DataSense finished",
                     result["steps_markdown"],
                     answer_html,
                 )
@@ -390,16 +435,7 @@ def build_ui() -> gr.Blocks:
 
             with gr.Column(scale=7, elem_id="ds-results"):
                 gr.Markdown("### Results")
-                progress_label = gr.Markdown("_Idle — run a task to see live progress._", elem_id="ds-progress-label")
-                progress_bar = gr.Slider(
-                    minimum=0,
-                    maximum=100,
-                    value=0,
-                    step=1,
-                    label="Run progress",
-                    interactive=False,
-                    elem_id="ds-progress-bar",
-                )
+                progress_out = gr.HTML(value=IDLE_PROGRESS_HTML)
                 with gr.Tabs():
                     with gr.Tab("🔍 Execution trace", id="trace_tab"):
                         steps_out = gr.Markdown()
@@ -416,7 +452,7 @@ def build_ui() -> gr.Blocks:
         run_btn.click(
             fn=run_task,
             inputs=[data_mode, dataset, upload, task, max_steps],
-            outputs=[progress_bar, progress_label, steps_out, answer_out],
+            outputs=[progress_out, steps_out, answer_out],
             show_progress="hidden",
         )