Upload folder using huggingface_hub

app.py

@@ -645,7 +645,24 @@ def _live_task_analysis(model_bundle, task: str, input_text: str, task5_cfg: dic
     return _live_input_summary(model_bundle, text)
 
 
-def _bg_worker(job_id: str, model_bundle, output_dir: str, input_text: str, task4_phase: str, task5_cfg: dict):
+def _run_quick_task(task, model_bundle, input_text, task5_cfg):
+    log = (
+        f"[Quick Mode] Task {task}\n"
+        f"Heavy analysis runner skipped for speed.\n\n"
+        f"{_live_task_analysis(model_bundle, task, input_text, task5_cfg)}"
+    )
+    return 0, log, False
+
+
+def _bg_worker(
+    job_id: str,
+    model_bundle,
+    output_dir: str,
+    input_text: str,
+    task4_phase: str,
+    task5_cfg: dict,
+    quick_mode: bool,
+):
     tasks = ["1", "2", "3", "4", "5"]
     failures = 0
     logs = []
@@ -661,14 +678,17 @@ def _bg_worker(job_id: str, model_bundle, output_dir: str, input_text: str, task
             }
         )
         try:
-            code, log, used_bundled = _run_analysis_cmd(
-                task,
-                model_bundle["ckpt_path"],
-                output_dir,
-                input_text,
-                task4_phase,
-                task5_cfg.get("samples", 50),
-            )
+            if quick_mode:
+                code, log, used_bundled = _run_quick_task(task, model_bundle, input_text, task5_cfg)
+            else:
+                code, log, used_bundled = _run_analysis_cmd(
+                    task,
+                    model_bundle["ckpt_path"],
+                    output_dir,
+                    input_text,
+                    task4_phase,
+                    task5_cfg.get("samples", 50),
+                )
             logs.append(f"\n\n{'='*22} TASK {task} {'='*22}\n{log}")
             if code != 0:
                 failures += 1
@@ -740,7 +760,7 @@ def _bg_worker(job_id: str, model_bundle, output_dir: str, input_text: str, task
     )
 
 
-def start_run_all_background(model_bundle, output_dir, input_text, task4_phase, task5_cfg):
+def start_run_all_background(model_bundle, output_dir, input_text, task4_phase, task5_cfg, quick_mode):
     if not model_bundle:
         raise gr.Error("Load a model first.")
     os.makedirs(output_dir, exist_ok=True)
@@ -758,12 +778,19 @@ def start_run_all_background(model_bundle, output_dir, input_text, task4_phase,
     }
     th = threading.Thread(
         target=_bg_worker,
-        args=(job_id, model_bundle, output_dir, input_text, task4_phase, task5_cfg),
+        args=(job_id, model_bundle, output_dir, input_text, task4_phase, task5_cfg, bool(quick_mode)),
         daemon=True,
     )
     th.start()
     flow = _build_flow_markdown(model_loaded=True, inference_ready=True, task_states=_BG_JOBS[job_id]["task_states"])
-    return f"Background run started. Job ID: {job_id}", f"Job {job_id} queued...", job_id, _BG_JOBS[job_id]["task_states"], flow
+    mode = "Quick" if quick_mode else "Full"
+    return (
+        f"Background run started ({mode} Mode). Job ID: {job_id}",
+        f"Job {job_id} queued...",
+        job_id,
+        _BG_JOBS[job_id]["task_states"],
+        flow,
+    )
 
 
 def poll_run_all_background(job_id, output_dir):
@@ -782,19 +809,24 @@ def poll_run_all_background(job_id, output_dir):
     return status, j.get("log", ""), j.get("task_states", {}), flow, *outputs
 
 
-def run_single_task_and_refresh(model_bundle, task, output_dir, input_text, task4_phase, task5_cfg):
-    status, log, task_states, flow = run_single_task(model_bundle, task, output_dir, input_text, task4_phase, task5_cfg)
+def run_single_task_and_refresh(model_bundle, task, output_dir, input_text, task4_phase, task5_cfg, quick_mode):
+    status, log, task_states, flow = run_single_task(
+        model_bundle, task, output_dir, input_text, task4_phase, task5_cfg, quick_mode
+    )
     out = refresh_task_outputs(output_dir)
     return status, log, task_states, flow, *out
 
 
-def run_single_task(model_bundle, task, output_dir, input_text, task4_phase, task5_cfg):
+def run_single_task(model_bundle, task, output_dir, input_text, task4_phase, task5_cfg, quick_mode):
     if not model_bundle:
         raise gr.Error("Load a model first.")
     t0 = time.perf_counter()
-    code, log, used_bundled = _run_analysis_cmd(
-        task, model_bundle["ckpt_path"], output_dir, input_text, task4_phase, task5_cfg.get("samples", 50)
-    )
+    if quick_mode:
+        code, log, used_bundled = _run_quick_task(task, model_bundle, input_text, task5_cfg)
+    else:
+        code, log, used_bundled = _run_analysis_cmd(
+            task, model_bundle["ckpt_path"], output_dir, input_text, task4_phase, task5_cfg.get("samples", 50)
+        )
     task_states = {k: "pending" for k in ["1", "2", "3", "4", "5"]}
     task_states[str(task)] = "running"
     elapsed = (time.perf_counter() - t0) * 1000.0
@@ -836,16 +868,19 @@ def run_single_task(model_bundle, task, output_dir, input_text, task4_phase, tas
     return status, log, task_states, flow
 
 
-def run_all_tasks(model_bundle, output_dir, input_text, task4_phase, task5_cfg):
+def run_all_tasks(model_bundle, output_dir, input_text, task4_phase, task5_cfg, quick_mode):
     if not model_bundle:
         raise gr.Error("Load a model first.")
     logs = []
     failures = 0
     used_bundled_any = False
     for task in ["1", "2", "3", "4", "5"]:
-        code, log, used_bundled = _run_analysis_cmd(
-            task, model_bundle["ckpt_path"], output_dir, input_text, task4_phase, task5_cfg.get("samples", 50)
-        )
+        if quick_mode:
+            code, log, used_bundled = _run_quick_task(task, model_bundle, input_text, task5_cfg)
+        else:
+            code, log, used_bundled = _run_analysis_cmd(
+                task, model_bundle["ckpt_path"], output_dir, input_text, task4_phase, task5_cfg.get("samples", 50)
+            )
         logs.append(f"\n\n{'='*22} TASK {task} {'='*22}\n{log}")
         used_bundled_any = used_bundled_any or used_bundled
         if code != 0:
@@ -916,11 +951,14 @@ def _safe_refresh_task_outputs(output_dir):
 
 
 def _safe_start_run_all_background(
-    model_bundle, output_dir, input_text, task4_phase, current_job_id, lambda_min, lambda_max, lambda_step, task5_samples
+    model_bundle, output_dir, input_text, task4_phase, current_job_id,
+    lambda_min, lambda_max, lambda_step, task5_samples, quick_mode
 ):
     try:
         cfg = _task5_cfg(lambda_min, lambda_max, lambda_step, task5_samples)
-        status, log, job_id, task_states, flow = start_run_all_background(model_bundle, output_dir, input_text, task4_phase, cfg)
+        status, log, job_id, task_states, flow = start_run_all_background(
+            model_bundle, output_dir, input_text, task4_phase, cfg, quick_mode
+        )
         return status, log, job_id, task_states, flow
     except Exception as e:
         err = f"Background start failed: {e}"
@@ -937,11 +975,12 @@ def _safe_poll_run_all_background(job_id, output_dir):
 
 
 def _safe_run_single_task_and_refresh(
-    model_bundle, task, output_dir, input_text, task4_phase, lambda_min, lambda_max, lambda_step, task5_samples
+    model_bundle, task, output_dir, input_text, task4_phase,
+    lambda_min, lambda_max, lambda_step, task5_samples, quick_mode
 ):
     try:
         cfg = _task5_cfg(lambda_min, lambda_max, lambda_step, task5_samples)
-        return run_single_task_and_refresh(model_bundle, task, output_dir, input_text, task4_phase, cfg)
+        return run_single_task_and_refresh(model_bundle, task, output_dir, input_text, task4_phase, cfg, quick_mode)
     except Exception as e:
         err = f"Task {task} failed: {e}"
         out = _safe_refresh_task_outputs(output_dir)
@@ -1050,6 +1089,11 @@ with gr.Blocks(title="Sanskrit Diffusion Client Demo", css=CUSTOM_CSS) as demo:
                         value="analyze",
                         label="Task 4 Phase",
                     )
+                    quick_mode = gr.Checkbox(
+                        value=True,
+                        label="Quick Mode (fast live analysis)",
+                        info="Runs lightweight live task analysis instead of heavy runner.",
+                    )
                     gr.Markdown("**Task 5 Controls**")
                     task5_lambda_min = gr.Slider(0.0, 3.0, value=0.0, step=0.1, label="Task5 λ min")
                     task5_lambda_max = gr.Slider(0.0, 3.0, value=3.0, step=0.1, label="Task5 λ max")
@@ -1176,7 +1220,7 @@ with gr.Blocks(title="Sanskrit Diffusion Client Demo", css=CUSTOM_CSS) as demo:
         fn=_safe_run_single_task_and_refresh,
         inputs=[
             model_state, task_choice, analysis_output_dir, analysis_input, task4_phase,
-            task5_lambda_min, task5_lambda_max, task5_lambda_step, task5_samples
+            task5_lambda_min, task5_lambda_max, task5_lambda_step, task5_samples, quick_mode
         ],
         outputs=[
             task_run_status,
@@ -1199,7 +1243,7 @@ with gr.Blocks(title="Sanskrit Diffusion Client Demo", css=CUSTOM_CSS) as demo:
         fn=_safe_start_run_all_background,
         inputs=[
             model_state, analysis_output_dir, analysis_input, task4_phase, bg_job_state,
-            task5_lambda_min, task5_lambda_max, task5_lambda_step, task5_samples
+            task5_lambda_min, task5_lambda_max, task5_lambda_step, task5_samples, quick_mode
         ],
         outputs=[task_run_status, task_run_log, bg_job_state, task_states_view, flow_box],
     )