Upload folder using huggingface_hub

app.py

@@ -461,13 +461,33 @@ def _live_task_analysis(model_bundle, task: str, input_text: str) -> str:
         )
 
     if str(task) == "2":
+        # Live diffusion proxy: run same input with multiple step counts and
+        # show semantic drift to final output while task is running.
+        base_steps = int(model_bundle["cfg"]["inference"].get("num_steps", 64))
+        step_grid = sorted(set([max(1, base_steps), max(1, base_steps // 2), max(1, base_steps // 4), 1]), reverse=True)
+        traj = []
+        final_out = None
+        for s in step_grid:
+            out_s = _run_single_prediction(model_bundle, text, {"num_steps": int(s)})
+            if s == 1:
+                final_out = out_s
+            traj.append((s, out_s))
+        if final_out is None and traj:
+            final_out = traj[-1][1]
+        drift_rows = []
+        for s, out_s in traj:
+            d = _compute_cer(out_s, final_out or out_s)
+            drift_rows.append((s, round(d, 4), out_s[:56]))
+
         tfidf = _mini_tfidf_scores(text)
         top = sorted(tfidf.items(), key=lambda kv: kv[1], reverse=True)[:5]
+        traj_txt = "\n".join([f"steps={s:>3d}  drift_to_final={d:.4f}  out={o}" for s, d, o in drift_rows])
         return (
             f"[Live Task2]\n"
             f"Input: {text}\nPrediction: {pred}\n"
             f"Token-length={len(toks)}  unique-ratio={uniq:.3f}\n"
-            f"TF-IDF(top): {top}"
+            f"TF-IDF(top): {top}\n"
+            f"Diffusion trajectory (live):\n{traj_txt}"
         )
 
     if str(task) == "3":
@@ -565,7 +585,7 @@ def start_run_all_background(model_bundle, output_dir, input_text, task4_phase):
 
 def poll_run_all_background(job_id, output_dir):
     if not job_id or job_id not in _BG_JOBS:
-        msg = "No active background job. Start Run All 5 Tasks first."
+        msg = "Background job idle. You can run a single task or start Run All 5 in background."
         empty = refresh_task_outputs(output_dir)
         return msg, msg, *empty
     j = _BG_JOBS[job_id]
@@ -577,6 +597,12 @@ def poll_run_all_background(job_id, output_dir):
     return status, j.get("log", ""), *outputs
 
 
+def run_single_task_and_refresh(model_bundle, task, output_dir, input_text, task4_phase):
+    status, log = run_single_task(model_bundle, task, output_dir, input_text, task4_phase)
+    out = refresh_task_outputs(output_dir)
+    return status, log, *out
+
+
 def run_single_task(model_bundle, task, output_dir, input_text, task4_phase):
     if not model_bundle:
         raise gr.Error("Load a model first.")
@@ -641,11 +667,29 @@ def refresh_task_outputs(output_dir):
 
     task2_drift = _img_or_none(os.path.join(output_dir, "task2_semantic_drift.png"))
     task2_attn = _img_or_none(os.path.join(output_dir, "task2_attn_t0.png"))
+    task2_evolution = _img_or_none(os.path.join(output_dir, "task2_attn_evolution.png"))
+    # Show farthest diffusion step snapshot if available (t=max).
+    task2_tmax = None
+    try:
+        cands = []
+        for name in os.listdir(output_dir):
+            if name.startswith("task2_attn_t") and name.endswith(".png"):
+                step = name.replace("task2_attn_t", "").replace(".png", "")
+                if step.isdigit():
+                    cands.append((int(step), os.path.join(output_dir, name)))
+        if cands:
+            cands.sort(key=lambda x: x[0], reverse=True)
+            task2_tmax = cands[0][1]
+    except Exception:
+        task2_tmax = None
     task3_space = _img_or_none(os.path.join(output_dir, "task3_concept_space.png"))
     task4_plot = _img_or_none(os.path.join(output_dir, "task4_ablation_3d.png"))
     if task4_plot is None:
         task4_plot = _img_or_none(os.path.join(output_dir, "task4_3d.png"))
-    return task1_txt, task2_txt, task2_drift, task2_attn, task3_txt, task3_space, task5_txt, task4_plot
+    return (
+        task1_txt, task2_txt, task2_drift, task2_attn, task2_tmax, task2_evolution,
+        task3_txt, task3_space, task5_txt, task4_plot
+    )
 
 
 CUSTOM_CSS = """
@@ -745,6 +789,9 @@ with gr.Blocks(title="Sanskrit Diffusion Client Demo", css=CUSTOM_CSS) as demo:
                 with gr.Row():
                     task2_drift_img = gr.Image(label="Task2 Drift", type="filepath")
                     task2_attn_img = gr.Image(label="Task2 Attention", type="filepath")
+                with gr.Row():
+                    task2_tmax_img = gr.Image(label="Task2 Attention (t=max)", type="filepath")
+                    task2_evolution_img = gr.Image(label="Task2 Evolution", type="filepath")
                 task3_box = gr.Textbox(label="Task 3 Report", lines=10, interactive=False)
                 task3_img = gr.Image(label="Task3 Concept Space", type="filepath")
                 task5_box = gr.Textbox(label="Task 5 Report", lines=10, interactive=False)
@@ -839,9 +886,22 @@ with gr.Blocks(title="Sanskrit Diffusion Client Demo", css=CUSTOM_CSS) as demo:
     )
 
     run_single_btn.click(
-        fn=run_single_task,
+        fn=run_single_task_and_refresh,
         inputs=[model_state, task_choice, analysis_output_dir, analysis_input, task4_phase],
-        outputs=[task_run_status, task_run_log],
+        outputs=[
+            task_run_status,
+            task_run_log,
+            task1_box,
+            task2_box,
+            task2_drift_img,
+            task2_attn_img,
+            task2_tmax_img,
+            task2_evolution_img,
+            task3_box,
+            task3_img,
+            task5_box,
+            task4_img,
+        ],
     )
     run_all_btn.click(
         fn=start_run_all_background,
@@ -858,6 +918,8 @@ with gr.Blocks(title="Sanskrit Diffusion Client Demo", css=CUSTOM_CSS) as demo:
             task2_box,
             task2_drift_img,
             task2_attn_img,
+            task2_tmax_img,
+            task2_evolution_img,
             task3_box,
             task3_img,
             task5_box,
@@ -872,6 +934,8 @@ with gr.Blocks(title="Sanskrit Diffusion Client Demo", css=CUSTOM_CSS) as demo:
             task2_box,
             task2_drift_img,
             task2_attn_img,
+            task2_tmax_img,
+            task2_evolution_img,
             task3_box,
             task3_img,
             task5_box,
@@ -890,6 +954,8 @@ with gr.Blocks(title="Sanskrit Diffusion Client Demo", css=CUSTOM_CSS) as demo:
             task2_box,
             task2_drift_img,
             task2_attn_img,
+            task2_tmax_img,
+            task2_evolution_img,
             task3_box,
             task3_img,
             task5_box,