Update app.py

app.py

@@ -1,11 +1,11 @@
 import os
-import random
-import uuid
+import gc
 import json
+import uuid
 import time
-import asyncio
+import base64
+from io import BytesIO
 from threading import Thread
-from typing import Iterable
 
 import gradio as gr
 import spaces
@@ -13,204 +13,20 @@ import torch
 import numpy as np
 from PIL import Image
 import cv2
-import requests
 
 from transformers import (
     Qwen2VLForConditionalGeneration,
     Qwen2_5_VLForConditionalGeneration,
     AutoProcessor,
     TextIteratorStreamer,
-    AutoModel,
-    AutoTokenizer,
-)
-from transformers.image_utils import load_image
-from gradio.themes import Soft
-from gradio.themes.utils import colors, fonts, sizes
-
-colors.steel_blue = colors.Color(
-    name="steel_blue",
-    c50="#EBF3F8",
-    c100="#D3E5F0",
-    c200="#A8CCE1",
-    c300="#7DB3D2",
-    c400="#529AC3",
-    c500="#4682B4",
-    c600="#3E72A0",
-    c700="#36638C",
-    c800="#2E5378",
-    c900="#264364",
-    c950="#1E3450",
 )
 
-class SteelBlueTheme(Soft):
-    def __init__(
-        self,
-        *,
-        primary_hue: colors.Color | str = colors.gray,
-        secondary_hue: colors.Color | str = colors.steel_blue,
-        neutral_hue: colors.Color | str = colors.slate,
-        text_size: sizes.Size | str = sizes.text_lg,
-        font: fonts.Font | str | Iterable[fonts.Font | str] = (
-            fonts.GoogleFont("Outfit"), "Arial", "sans-serif",
-        ),
-        font_mono: fonts.Font | str | Iterable[fonts.Font | str] = (
-            fonts.GoogleFont("IBM Plex Mono"), "ui-monospace", "monospace",
-        ),
-    ):
-        super().__init__(
-            primary_hue=primary_hue,
-            secondary_hue=secondary_hue,
-            neutral_hue=neutral_hue,
-            text_size=text_size,
-            font=font,
-            font_mono=font_mono,
-        )
-        super().set(
-            background_fill_primary="*primary_50",
-            background_fill_primary_dark="*primary_900",
-            body_background_fill="linear-gradient(135deg, *primary_200, *primary_100)",
-            body_background_fill_dark="linear-gradient(135deg, *primary_900, *primary_800)",
-            button_primary_text_color="white",
-            button_primary_text_color_hover="white",
-            button_primary_background_fill="linear-gradient(90deg, *secondary_500, *secondary_600)",
-            button_primary_background_fill_hover="linear-gradient(90deg, *secondary_600, *secondary_700)",
-            button_primary_background_fill_dark="linear-gradient(90deg, *secondary_600, *secondary_800)",
-            button_primary_background_fill_hover_dark="linear-gradient(90deg, *secondary_500, *secondary_500)",
-            slider_color="*secondary_500",
-            slider_color_dark="*secondary_600",
-            block_title_text_weight="600",
-            block_border_width="3px",
-            block_shadow="*shadow_drop_lg",
-            button_primary_shadow="*shadow_drop_lg",
-            button_large_padding="11px",
-            color_accent_soft="*primary_100",
-            block_label_background_fill="*primary_200",
-        )
-
-steel_blue_theme = SteelBlueTheme()
-
-css = """
-#main-title h1 {
-    font-size: 2.3em !important;
-}
-#output-title h2 {
-    font-size: 2.2em !important;
-}
-
-/* RadioAnimated Styles */
-.ra-wrap{ width: fit-content; }
-.ra-inner{
-  position: relative; display: inline-flex; align-items: center; gap: 0; padding: 6px;
-  background: var(--neutral-200); border-radius: 9999px; overflow: hidden;
-}
-.ra-input{ display: none; }
-.ra-label{
-  position: relative; z-index: 2; padding: 8px 16px;
-  font-family: inherit; font-size: 14px; font-weight: 600;
-  color: var(--neutral-500); cursor: pointer; transition: color 0.2s; white-space: nowrap;
-}
-.ra-highlight{
-  position: absolute; z-index: 1; top: 6px; left: 6px;
-  height: calc(100% - 12px); border-radius: 9999px;
-  background: white; box-shadow: 0 2px 4px rgba(0,0,0,0.1);
-  transition: transform 0.2s, width 0.2s;
-}
-.ra-input:checked + .ra-label{ color: black; }
-
-/* Dark mode adjustments for Radio */
-.dark .ra-inner { background: var(--neutral-800); }
-.dark .ra-label { color: var(--neutral-400); }
-.dark .ra-highlight { background: var(--neutral-600); }
-.dark .ra-input:checked + .ra-label { color: white; }
-
-#gpu-duration-container {
-    padding: 10px;
-    border-radius: 8px;
-    background: var(--background-fill-secondary);
-    border: 1px solid var(--border-color-primary);
-    margin-top: 10px;
-}
-"""
-
 MAX_MAX_NEW_TOKENS = 2048
 DEFAULT_MAX_NEW_TOKENS = 1024
 MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-
-class RadioAnimated(gr.HTML):
-    def __init__(self, choices, value=None, **kwargs):
-        if not choices or len(choices) < 2:
-            raise ValueError("RadioAnimated requires at least 2 choices.")
-        if value is None:
-            value = choices[0]
-
-        uid = uuid.uuid4().hex[:8]
-        group_name = f"ra-{uid}"
-
-        inputs_html = "\n".join(
-            f"""
-            <input class="ra-input" type="radio" name="{group_name}" id="{group_name}-{i}" value="{c}">
-            <label class="ra-label" for="{group_name}-{i}">{c}</label>
-            """
-            for i, c in enumerate(choices)
-        )
-
-        html_template = f"""
-        <div class="ra-wrap" data-ra="{uid}">
-          <div class="ra-inner">
-            <div class="ra-highlight"></div>
-            {inputs_html}
-          </div>
-        </div>
-        """
-
-        js_on_load = r"""
-        (() => {
-          const wrap = element.querySelector('.ra-wrap');
-          const inner = element.querySelector('.ra-inner');
-          const highlight = element.querySelector('.ra-highlight');
-          const inputs = Array.from(element.querySelectorAll('.ra-input'));
-
-          if (!inputs.length) return;
-
-          const choices = inputs.map(i => i.value);
-
-          function setHighlightByIndex(idx) {
-            const n = choices.length;
-            const pct = 100 / n;
-            highlight.style.width = `calc(${pct}% - 6px)`;
-            highlight.style.transform = `translateX(${idx * 100}%)`;
-          }
-
-          function setCheckedByValue(val, shouldTrigger=false) {
-            const idx = Math.max(0, choices.indexOf(val));
-            inputs.forEach((inp, i) => { inp.checked = (i === idx); });
-            setHighlightByIndex(idx);
-
-            props.value = choices[idx];
-            if (shouldTrigger) trigger('change', props.value);
-          }
-
-          setCheckedByValue(props.value ?? choices[0], false);
-
-          inputs.forEach((inp) => {
-            inp.addEventListener('change', () => {
-              setCheckedByValue(inp.value, true);
-            });
-          });
-        })();
-        """
-
-        super().__init__(
-            value=value,
-            html_template=html_template,
-            js_on_load=js_on_load,
-            **kwargs
-        )
-
-def apply_gpu_duration(val: str):
-    return int(val)
+print("Using device:", device)
 
 MODEL_ID_M = "nvidia/Cosmos-Reason1-7B"
 processor_m = AutoProcessor.from_pretrained(MODEL_ID_M, trust_remote_code=True)
@@ -248,72 +64,206 @@ model_v = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     torch_dtype=torch.float16
 ).to(device).eval()
 
+MODEL_MAP = {
+    "Cosmos-Reason1-7B": (processor_m, model_m),
+    "docscopeOCR-7B-050425-exp": (processor_x, model_x),
+    "Captioner-7B-Qwen2.5VL": (processor_z, model_z),
+    "visionOCR-3B": (processor_v, model_v),
+}
+
+MODEL_CHOICES = list(MODEL_MAP.keys())
+
+image_examples = [
+    {"query": "Perform OCR on the text in the image.", "media": "images/1.jpg", "model": "docscopeOCR-7B-050425-exp", "mode": "image"},
+    {"query": "Explain the scene in detail.", "media": "images/2.jpg", "model": "Cosmos-Reason1-7B", "mode": "image"},
+]
+
+video_examples = [
+    {"query": "Explain the Ad in Detail", "media": "videos/1.mp4", "model": "Captioner-7B-Qwen2.5VL", "mode": "video"},
+    {"query": "Identify the main actions in the video", "media": "videos/2.mp4", "model": "visionOCR-3B", "mode": "video"},
+]
+
+all_examples = image_examples + video_examples
+
+
+def pil_to_data_url(img: Image.Image, fmt="PNG"):
+    buf = BytesIO()
+    img.save(buf, format=fmt)
+    data = base64.b64encode(buf.getvalue()).decode()
+    mime = "image/png" if fmt.upper() == "PNG" else "image/jpeg"
+    return f"data:{mime};base64,{data}"
+
+
+def file_to_data_url(path):
+    if not os.path.exists(path):
+        return ""
+    ext = path.rsplit(".", 1)[-1].lower()
+    mime = {
+        "jpg": "image/jpeg",
+        "jpeg": "image/jpeg",
+        "png": "image/png",
+        "webp": "image/webp",
+        "mp4": "video/mp4",
+        "mov": "video/quicktime",
+        "webm": "video/webm",
+    }.get(ext, "application/octet-stream")
+    with open(path, "rb") as f:
+        data = base64.b64encode(f.read()).decode()
+    return f"data:{mime};base64,{data}"
+
+
+def make_thumb_b64(path, mode="image", max_dim=240):
+    try:
+        if mode == "video":
+            cap = cv2.VideoCapture(path)
+            ok, frame = cap.read()
+            cap.release()
+            if not ok:
+                return ""
+            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            img = Image.fromarray(frame).convert("RGB")
+        else:
+            img = Image.open(path).convert("RGB")
+        img.thumbnail((max_dim, max_dim))
+        return pil_to_data_url(img, "JPEG")
+    except Exception as e:
+        print("Thumbnail error:", e)
+        return ""
+
+
+def build_example_cards_html():
+    cards = ""
+    for i, ex in enumerate(all_examples):
+        thumb = make_thumb_b64(ex["media"], ex["mode"])
+        prompt_short = ex["query"][:72] + ("..." if len(ex["query"]) > 72 else "")
+        media_badge = "VIDEO" if ex["mode"] == "video" else "IMAGE"
+        cards += f"""
+        <div class="example-card" data-idx="{i}">
+            <div class="example-thumb-wrap">
+                {"<img src='" + thumb + "' alt=''>" if thumb else "<div class='example-thumb-placeholder'>Preview</div>"}
+                <div class="example-media-chip">{media_badge}</div>
+            </div>
+            <div class="example-meta-row">
+                <span class="example-badge">{ex["model"]}</span>
+            </div>
+            <div class="example-prompt-text">{prompt_short}</div>
+        </div>
+        """
+    return cards
+
+
+EXAMPLE_CARDS_HTML = build_example_cards_html()
+
+
+def load_example_data(idx_str):
+    try:
+        idx = int(float(idx_str))
+    except Exception:
+        return json.dumps({"status": "error", "message": "Invalid example index"})
+    if idx < 0 or idx >= len(all_examples):
+        return json.dumps({"status": "error", "message": "Example index out of range"})
+    ex = all_examples[idx]
+    media_b64 = file_to_data_url(ex["media"])
+    if not media_b64:
+        return json.dumps({"status": "error", "message": f"Could not load example {ex['mode']}"})
+    return json.dumps({
+        "status": "ok",
+        "query": ex["query"],
+        "media": media_b64,
+        "model": ex["model"],
+        "mode": ex["mode"],
+        "name": os.path.basename(ex["media"]),
+    })
+
+
+def b64_to_pil(b64_str):
+    if not b64_str:
+        return None
+    try:
+        if b64_str.startswith("data:"):
+            _, data = b64_str.split(",", 1)
+        else:
+            data = b64_str
+        image_data = base64.b64decode(data)
+        return Image.open(BytesIO(image_data)).convert("RGB")
+    except Exception:
+        return None
+
+
+def b64_to_temp_video(b64_str):
+    if not b64_str:
+        return None
+    try:
+        if b64_str.startswith("data:"):
+            header, data = b64_str.split(",", 1)
+            mime = header.split(";")[0].replace("data:", "")
+        else:
+            data = b64_str
+            mime = "video/mp4"
+        ext = {
+            "video/mp4": ".mp4",
+            "video/webm": ".webm",
+            "video/quicktime": ".mov",
+        }.get(mime, ".mp4")
+        raw = base64.b64decode(data)
+        temp_dir = os.path.join("/tmp", "docscope_r1_media")
+        os.makedirs(temp_dir, exist_ok=True)
+        path = os.path.join(temp_dir, f"{uuid.uuid4().hex}{ext}")
+        with open(path, "wb") as f:
+            f.write(raw)
+        return path
+    except Exception:
+        return None
+
+
 def downsample_video(video_path):
-    """
-    Downsamples the video to evenly spaced frames.
-    Each frame is returned as a PIL image along with its timestamp.
-    """
     vidcap = cv2.VideoCapture(video_path)
     total_frames = int(vidcap.get(cv2.CAP_PROP_FRAME_COUNT))
-    fps = vidcap.get(cv2.CAP_PROP_FPS)
+    fps = vidcap.get(cv2.CAP_PROP_FPS) or 1.0
     frames = []
-    frame_indices = np.linspace(0, total_frames - 1, min(total_frames, 10), dtype=int)
+    frame_count = min(total_frames, 10) if total_frames > 0 else 0
+    if frame_count == 0:
+        vidcap.release()
+        return frames
+    frame_indices = np.linspace(0, total_frames - 1, frame_count, dtype=int)
     for i in frame_indices:
-        vidcap.set(cv2.CAP_PROP_POS_FRAMES, i)
+        vidcap.set(cv2.CAP_PROP_POS_FRAMES, int(i))
         success, image = vidcap.read()
         if success:
             image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
             pil_image = Image.fromarray(image)
-            timestamp = round(i / fps, 2)
+            timestamp = round(float(i) / float(fps), 2)
             frames.append((pil_image, timestamp))
     vidcap.release()
     return frames
 
-def calc_timeout_image(model_name: str, text: str, image: Image.Image,
-                       max_new_tokens: int, temperature: float, top_p: float,
-                       top_k: int, repetition_penalty: float, gpu_timeout: int):
-    """Calculate GPU timeout duration for image inference."""
+
+def calc_timeout_image(model_name, text, image, max_new_tokens, temperature, top_p, top_k, repetition_penalty, gpu_timeout):
     try:
         return int(gpu_timeout)
-    except:
+    except Exception:
         return 60
 
-def calc_timeout_video(model_name: str, text: str, video_path: str,
-                       max_new_tokens: int, temperature: float, top_p: float,
-                       top_k: int, repetition_penalty: float, gpu_timeout: int):
-    """Calculate GPU timeout duration for video inference."""
+
+def calc_timeout_video(model_name, text, video_path, max_new_tokens, temperature, top_p, top_k, repetition_penalty, gpu_timeout):
     try:
         return int(gpu_timeout)
-    except:
+    except Exception:
         return 60
 
-@spaces.GPU(duration=calc_timeout_image)
-def generate_image(model_name: str, text: str, image: Image.Image,
-                   max_new_tokens: int = 1024,
-                   temperature: float = 0.6,
-                   top_p: float = 0.9,
-                   top_k: int = 50,
-                   repetition_penalty: float = 1.2,
-                   gpu_timeout: int = 60):
-    """
-    Generates responses using the selected model for image input.
-    Yields raw text and Markdown-formatted text.
-    """
-    if model_name == "Cosmos-Reason1-7B":
-        processor, model = processor_m, model_m
-    elif model_name == "docscopeOCR-7B-050425-exp":
-        processor, model = processor_x, model_x
-    elif model_name == "Captioner-7B-Qwen2.5VL":
-        processor, model = processor_z, model_z
-    elif model_name == "visionOCR-3B":
-        processor, model = processor_v, model_v
-    else:
-        yield "Invalid model selected.", "Invalid model selected."
-        return
 
+@spaces.GPU(duration=calc_timeout_image)
+def generate_image(model_name, text, image, max_new_tokens=1024, temperature=0.6, top_p=0.9, top_k=50, repetition_penalty=1.2, gpu_timeout=60):
+    if not model_name or model_name not in MODEL_MAP:
+        raise gr.Error("Please select a valid model.")
     if image is None:
-        yield "Please upload an image.", "Please upload an image."
-        return
+        raise gr.Error("Please upload an image.")
+    if not text or not str(text).strip():
+        raise gr.Error("Please enter your instruction.")
+    if len(str(text)) > MAX_INPUT_TOKEN_LENGTH * 8:
+        raise gr.Error("Query is too long. Please shorten your input.")
+
+    processor, model = MODEL_MAP[model_name]
 
     messages = [{
         "role": "user",
@@ -322,7 +272,13 @@ def generate_image(model_name: str, text: str, image: Image.Image,
             {"type": "text", "text": text},
         ]
     }]
-    prompt_full = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+
+    prompt_full = processor.apply_chat_template(
+        messages,
+        tokenize=False,
+        add_generation_prompt=True
+    )
+
     inputs = processor(
         text=[prompt_full],
         images=[image],
@@ -331,53 +287,58 @@ def generate_image(model_name: str, text: str, image: Image.Image,
         truncation=True,
         max_length=MAX_INPUT_TOKEN_LENGTH
     ).to(device)
+
     streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
-    generation_kwargs = {**inputs, "streamer": streamer, "max_new_tokens": max_new_tokens}
+    generation_kwargs = {
+        **inputs,
+        "streamer": streamer,
+        "max_new_tokens": int(max_new_tokens),
+        "do_sample": True,
+        "temperature": float(temperature),
+        "top_p": float(top_p),
+        "top_k": int(top_k),
+        "repetition_penalty": float(repetition_penalty),
+    }
+
     thread = Thread(target=model.generate, kwargs=generation_kwargs)
     thread.start()
+
     buffer = ""
     for new_text in streamer:
-        buffer += new_text
+        buffer += new_text.replace("<|im_end|>", "")
         time.sleep(0.01)
-        yield buffer, buffer
+        yield buffer
 
-@spaces.GPU(duration=calc_timeout_video)
-def generate_video(model_name: str, text: str, video_path: str,
-                   max_new_tokens: int = 1024,
-                   temperature: float = 0.6,
-                   top_p: float = 0.9,
-                   top_k: int = 50,
-                   repetition_penalty: float = 1.2,
-                   gpu_timeout: int = 90):
-    """
-    Generates responses using the selected model for video input.
-    Yields raw text and Markdown-formatted text.
-    """
-    if model_name == "Cosmos-Reason1-7B":
-        processor, model = processor_m, model_m
-    elif model_name == "docscopeOCR-7B-050425-exp":
-        processor, model = processor_x, model_x
-    elif model_name == "Captioner-7B-Qwen2.5VL":
-        processor, model = processor_z, model_z
-    elif model_name == "visionOCR-3B":
-        processor, model = processor_v, model_v
-    else:
-        yield "Invalid model selected.", "Invalid model selected."
-        return
+    gc.collect()
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
 
-    if video_path is None:
-        yield "Please upload a video.", "Please upload a video."
-        return
 
+@spaces.GPU(duration=calc_timeout_video)
+def generate_video(model_name, text, video_path, max_new_tokens=1024, temperature=0.6, top_p=0.9, top_k=50, repetition_penalty=1.2, gpu_timeout=90):
+    if not model_name or model_name not in MODEL_MAP:
+        raise gr.Error("Please select a valid model.")
+    if not video_path:
+        raise gr.Error("Please upload a video.")
+    if not text or not str(text).strip():
+        raise gr.Error("Please enter your instruction.")
+    if len(str(text)) > MAX_INPUT_TOKEN_LENGTH * 8:
+        raise gr.Error("Query is too long. Please shorten your input.")
+
+    processor, model = MODEL_MAP[model_name]
     frames = downsample_video(video_path)
+    if not frames:
+        raise gr.Error("Could not read the uploaded video.")
+
     messages = [
         {"role": "system", "content": [{"type": "text", "text": "You are a helpful assistant."}]},
         {"role": "user", "content": [{"type": "text", "text": text}]}
     ]
-    for frame in frames:
-        image, timestamp = frame
+
+    for image, timestamp in frames:
         messages[1]["content"].append({"type": "text", "text": f"Frame {timestamp}:"})
         messages[1]["content"].append({"type": "image", "image": image})
+
     inputs = processor.apply_chat_template(
         messages,
         tokenize=True,
@@ -387,99 +348,1096 @@ def generate_video(model_name: str, text: str, video_path: str,
         truncation=True,
         max_length=MAX_INPUT_TOKEN_LENGTH
     ).to(device)
+
     streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)
     generation_kwargs = {
         **inputs,
         "streamer": streamer,
-        "max_new_tokens": max_new_tokens,
+        "max_new_tokens": int(max_new_tokens),
         "do_sample": True,
-        "temperature": temperature,
-        "top_p": top_p,
-        "top_k": top_k,
-        "repetition_penalty": repetition_penalty,
+        "temperature": float(temperature),
+        "top_p": float(top_p),
+        "top_k": int(top_k),
+        "repetition_penalty": float(repetition_penalty),
     }
+
     thread = Thread(target=model.generate, kwargs=generation_kwargs)
     thread.start()
+
     buffer = ""
     for new_text in streamer:
-        buffer += new_text
+        buffer += new_text.replace("<|im_end|>", "")
         time.sleep(0.01)
-        yield buffer, buffer
+        yield buffer
 
-image_examples = [
-    ["Perform OCR on the text in the image.", "images/1.jpg"],
-    ["Explain the scene in detail.", "images/2.jpg"]
-]
+    gc.collect()
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
 
-video_examples = [
-    ["Explain the Ad in Detail", "videos/1.mp4"],
-    ["Identify the main actions in the video", "videos/2.mp4"]
-]
 
-with gr.Blocks() as demo:
-    gr.Markdown("# **DocScope R1**", elem_id="main-title")
-    with gr.Row():
-        with gr.Column(scale=2):
-            with gr.Tabs():
-                with gr.TabItem("Image Inference"):
-                    image_query = gr.Textbox(label="Query Input", placeholder="Enter your query here...")
-                    image_upload = gr.Image(type="pil", label="Upload Image", height=290)
-                    image_submit = gr.Button("Submit", variant="primary")
-                    gr.Examples(examples=image_examples, inputs=[image_query, image_upload])
-                with gr.TabItem("Video Inference"):
-                    video_query = gr.Textbox(label="Query Input", placeholder="Enter your query here...")
-                    video_upload = gr.Video(label="Upload Video", height=290)
-                    video_submit = gr.Button("Submit", variant="primary")
-                    gr.Examples(examples=video_examples, inputs=[video_query, video_upload])
-                    
-            with gr.Accordion("Advanced options", open=False):
-                max_new_tokens = gr.Slider(label="Max new tokens", minimum=1, maximum=MAX_MAX_NEW_TOKENS, step=1, value=DEFAULT_MAX_NEW_TOKENS)
-                temperature = gr.Slider(label="Temperature", minimum=0.1, maximum=4.0, step=0.1, value=0.6)
-                top_p = gr.Slider(label="Top-p (nucleus sampling)", minimum=0.05, maximum=1.0, step=0.05, value=0.9)
-                top_k = gr.Slider(label="Top-k", minimum=1, maximum=1000, step=1, value=50)
-                repetition_penalty = gr.Slider(label="Repetition penalty", minimum=1.0, maximum=2.0, step=0.05, value=1.2)
-                
-        with gr.Column(scale=3):
-            gr.Markdown("## Output", elem_id="output-title")
-            raw_output = gr.Textbox(label="Raw Output Stream", interactive=True, lines=11)
-            with gr.Accordion("(Result.md)", open=False):
-                markdown_output = gr.Markdown()
-                    
-            model_choice = gr.Radio(
-                choices=["Cosmos-Reason1-7B", "docscopeOCR-7B-050425-exp", "Captioner-7B-Qwen2.5VL", "visionOCR-3B"],
-                label="Select Model",
-                value="Cosmos-Reason1-7B"
+def run_inference(mode, model_name, text, image_b64, video_b64, max_new_tokens_v, temperature_v, top_p_v, top_k_v, repetition_penalty_v, gpu_timeout_v):
+    if mode == "video":
+        temp_video_path = b64_to_temp_video(video_b64)
+        if not temp_video_path:
+            raise gr.Error("Could not decode uploaded video.")
+        try:
+            yield from generate_video(
+                model_name=model_name,
+                text=text,
+                video_path=temp_video_path,
+                max_new_tokens=max_new_tokens_v,
+                temperature=temperature_v,
+                top_p=top_p_v,
+                top_k=top_k_v,
+                repetition_penalty=repetition_penalty_v,
+                gpu_timeout=gpu_timeout_v,
             )
-            
-            with gr.Row(elem_id="gpu-duration-container"):
-                with gr.Column():
-                    gr.Markdown("**GPU Duration (seconds)**")
-                    radioanimated_gpu_duration = RadioAnimated(
-                        choices=["60", "90", "120", "180", "240", "300"],
-                        value="60",
-                        elem_id="radioanimated_gpu_duration"
-                    )
-                    gpu_duration_state = gr.Number(value=60, visible=False)
-            
-            gr.Markdown("*Note: Higher GPU duration allows for longer processing but consumes more GPU quota.*")
-            
-    radioanimated_gpu_duration.change(
-        fn=apply_gpu_duration, 
-        inputs=radioanimated_gpu_duration, 
-        outputs=[gpu_duration_state], 
-        api_visibility="private"
-    )
+        finally:
+            try:
+                os.remove(temp_video_path)
+            except Exception:
+                pass
+    else:
+        image = b64_to_pil(image_b64)
+        yield from generate_image(
+            model_name=model_name,
+            text=text,
+            image=image,
+            max_new_tokens=max_new_tokens_v,
+            temperature=temperature_v,
+            top_p=top_p_v,
+            top_k=top_k_v,
+            repetition_penalty=repetition_penalty_v,
+            gpu_timeout=gpu_timeout_v,
+        )
+
+
+def noop():
+    return None
+
+
+css = r"""
+@import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700;800&family=JetBrains+Mono:wght@400;500;600&display=swap');
+*{box-sizing:border-box;margin:0;padding:0}
+html,body{height:100%;overflow-x:hidden}
+body,.gradio-container{
+    background:#0f0f13!important;
+    font-family:'Inter',system-ui,-apple-system,sans-serif!important;
+    font-size:14px!important;color:#e4e4e7!important;min-height:100vh;overflow-x:hidden;
+}
+.dark body,.dark .gradio-container{background:#0f0f13!important;color:#e4e4e7!important}
+footer{display:none!important}
+.hidden-input{display:none!important;height:0!important;overflow:hidden!important;margin:0!important;padding:0!important}
+
+#gradio-run-btn,#example-load-btn{
+    position:absolute!important;left:-9999px!important;top:-9999px!important;
+    width:1px!important;height:1px!important;opacity:0.01!important;
+    pointer-events:none!important;overflow:hidden!important;
+}
+
+.app-shell{
+    background:#18181b;border:1px solid #27272a;border-radius:16px;
+    margin:12px auto;max-width:1400px;overflow:hidden;
+    box-shadow:0 25px 50px -12px rgba(0,0,0,.6),0 0 0 1px rgba(255,255,255,.03);
+}
+.app-header{
+    background:linear-gradient(135deg,#18181b,#1e1e24);border-bottom:1px solid #27272a;
+    padding:14px 24px;display:flex;align-items:center;justify-content:space-between;flex-wrap:wrap;gap:12px;
+}
+.app-header-left{display:flex;align-items:center;gap:12px}
+.app-logo{
+    width:38px;height:38px;background:linear-gradient(135deg,#FF1493,#E1007A,#FF4DB2);
+    border-radius:10px;display:flex;align-items:center;justify-content:center;
+    box-shadow:0 4px 12px rgba(255,20,147,.35);
+}
+.app-logo svg{width:22px;height:22px;fill:#fff;flex-shrink:0}
+.app-title{
+    font-size:18px;font-weight:700;background:linear-gradient(135deg,#f5f5f5,#bdbdbd);
+    -webkit-background-clip:text;-webkit-text-fill-color:transparent;letter-spacing:-.3px;
+}
+.app-badge{
+    font-size:11px;font-weight:600;padding:3px 10px;border-radius:20px;
+    background:rgba(255,20,147,.12);color:#ff7ac7;border:1px solid rgba(255,20,147,.25);letter-spacing:.3px;
+}
+.app-badge.fast{background:rgba(225,0,122,.10);color:#ff66be;border:1px solid rgba(225,0,122,.22)}
+
+.model-tabs-bar{
+    background:#18181b;border-bottom:1px solid #27272a;padding:10px 16px;
+    display:flex;gap:8px;align-items:center;flex-wrap:wrap;
+}
+.model-tab{
+    display:inline-flex;align-items:center;justify-content:center;gap:6px;
+    min-width:32px;height:34px;background:transparent;border:1px solid #27272a;
+    border-radius:999px;cursor:pointer;font-size:12px;font-weight:600;padding:0 12px;
+    color:#ffffff!important;transition:all .15s ease;
+}
+.model-tab:hover{background:rgba(255,20,147,.12);border-color:rgba(255,20,147,.35)}
+.model-tab.active{background:rgba(255,20,147,.22);border-color:#FF1493;color:#fff!important;box-shadow:0 0 0 2px rgba(255,20,147,.10)}
+.model-tab-label{font-size:12px;color:#ffffff!important;font-weight:600}
 
-    image_submit.click(
-        fn=generate_image,
-        inputs=[model_choice, image_query, image_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty, gpu_duration_state],
-        outputs=[raw_output, markdown_output]
+.mode-tabs-bar{
+    background:#18181b;border-bottom:1px solid #27272a;padding:10px 16px 12px;
+    display:flex;gap:8px;align-items:center;flex-wrap:wrap;
+}
+.mode-tab{
+    display:inline-flex;align-items:center;justify-content:center;gap:6px;
+    min-width:110px;height:34px;background:transparent;border:1px solid #27272a;
+    border-radius:999px;cursor:pointer;font-size:12px;font-weight:700;padding:0 14px;
+    color:#ffffff!important;transition:all .15s ease;text-transform:uppercase;letter-spacing:.5px;
+}
+.mode-tab:hover{background:rgba(255,20,147,.12);border-color:rgba(255,20,147,.35)}
+.mode-tab.active{background:rgba(255,20,147,.22);border-color:#FF1493;color:#fff!important;box-shadow:0 0 0 2px rgba(255,20,147,.10)}
+
+.app-main-row{display:flex;gap:0;flex:1;overflow:hidden}
+.app-main-left{flex:1;display:flex;flex-direction:column;min-width:0;border-right:1px solid #27272a}
+.app-main-right{width:470px;display:flex;flex-direction:column;flex-shrink:0;background:#18181b}
+
+#media-drop-zone{
+    position:relative;background:#09090b;height:440px;min-height:440px;max-height:440px;
+    overflow:hidden;
+}
+#media-drop-zone.drag-over{outline:2px solid #FF1493;outline-offset:-2px;background:rgba(255,20,147,.04)}
+.upload-prompt-modern{
+    position:absolute;inset:0;display:flex;align-items:center;justify-content:center;
+    padding:20px;z-index:20;overflow:hidden;
+}
+.upload-click-area{
+    display:flex;flex-direction:column;align-items:center;justify-content:center;
+    cursor:pointer;padding:28px 36px;max-width:92%;max-height:92%;
+    border:2px dashed #3f3f46;border-radius:16px;
+    background:rgba(255,20,147,.03);transition:all .2s ease;gap:8px;text-align:center;
+    overflow:hidden;
+}
+.upload-click-area:hover{background:rgba(255,20,147,.08);border-color:#FF1493;transform:scale(1.02)}
+.upload-click-area:active{background:rgba(255,20,147,.12);transform:scale(.99)}
+.upload-click-area svg{width:86px;height:86px;max-width:100%;flex-shrink:0}
+.upload-main-text{color:#a1a1aa;font-size:14px;font-weight:600;margin-top:4px}
+.upload-sub-text{color:#71717a;font-size:12px}
+
+.single-preview-wrap{
+    width:100%;height:100%;display:none;align-items:center;justify-content:center;padding:16px;
+    overflow:hidden;
+}
+.single-preview-card{
+    width:100%;height:100%;max-width:100%;max-height:100%;border-radius:14px;
+    overflow:hidden;border:1px solid #27272a;background:#111114;
+    display:flex;align-items:center;justify-content:center;position:relative;
+}
+.single-preview-card img,.single-preview-card video{
+    width:100%;height:100%;max-width:100%;max-height:100%;
+    object-fit:contain;display:block;background:#000;
+}
+.preview-overlay-actions{
+    position:absolute;top:12px;right:12px;display:flex;gap:8px;z-index:5;
+}
+.preview-action-btn{
+    display:inline-flex;align-items:center;justify-content:center;
+    min-width:34px;height:34px;padding:0 12px;background:rgba(0,0,0,.65);
+    border:1px solid rgba(255,255,255,.14);border-radius:10px;cursor:pointer;
+    color:#fff!important;font-size:12px;font-weight:600;transition:all .15s ease;
+}
+.preview-action-btn:hover{background:#FF1493;border-color:#FF1493}
+
+.hint-bar{
+    background:rgba(255,20,147,.06);border-top:1px solid #27272a;border-bottom:1px solid #27272a;
+    padding:10px 20px;font-size:13px;color:#a1a1aa;line-height:1.7;
+}
+.hint-bar b{color:#ff7ac7;font-weight:600}
+.hint-bar kbd{
+    display:inline-block;padding:1px 6px;background:#27272a;border:1px solid #3f3f46;
+    border-radius:4px;font-family:'JetBrains Mono',monospace;font-size:11px;color:#a1a1aa;
+}
+
+.examples-section{border-top:1px solid #27272a;padding:12px 16px}
+.examples-title{
+    font-size:12px;font-weight:600;color:#71717a;text-transform:uppercase;
+    letter-spacing:.8px;margin-bottom:10px;
+}
+.examples-scroll{display:flex;gap:10px;overflow-x:auto;padding-bottom:8px}
+.examples-scroll::-webkit-scrollbar{height:6px}
+.examples-scroll::-webkit-scrollbar-track{background:#09090b;border-radius:3px}
+.examples-scroll::-webkit-scrollbar-thumb{background:#27272a;border-radius:3px}
+.examples-scroll::-webkit-scrollbar-thumb:hover{background:#3f3f46}
+.example-card{
+    position:relative;
+    flex-shrink:0;width:220px;background:#09090b;border:1px solid #27272a;
+    border-radius:10px;overflow:hidden;cursor:pointer;transition:all .2s ease;
+}
+.example-card:hover{border-color:#FF1493;transform:translateY(-2px);box-shadow:0 4px 12px rgba(255,20,147,.15)}
+.example-card.loading{opacity:.5;pointer-events:none}
+.example-thumb-wrap{height:120px;overflow:hidden;background:#18181b;position:relative}
+.example-thumb-wrap img{width:100%;height:100%;object-fit:cover}
+.example-media-chip{
+    position:absolute;top:8px;left:8px;
+    display:inline-flex;padding:3px 7px;background:rgba(0,0,0,.7);border:1px solid rgba(255,255,255,.12);
+    border-radius:999px;font-size:10px;font-weight:700;color:#fff;letter-spacing:.5px;
+}
+.example-thumb-placeholder{
+    width:100%;height:100%;display:flex;align-items:center;justify-content:center;
+    background:#18181b;color:#3f3f46;font-size:11px;
+}
+.example-meta-row{padding:6px 10px;display:flex;align-items:center;gap:6px}
+.example-badge{
+    display:inline-flex;padding:2px 7px;background:rgba(255,20,147,.12);border-radius:4px;
+    font-size:10px;font-weight:600;color:#ff7ac7;font-family:'JetBrains Mono',monospace;white-space:nowrap;
+}
+.example-prompt-text{
+    padding:0 10px 8px;font-size:11px;color:#a1a1aa;line-height:1.4;
+    display:-webkit-box;-webkit-line-clamp:2;-webkit-box-orient:vertical;overflow:hidden;
+}
+
+.panel-card{border-bottom:1px solid #27272a}
+.panel-card-title{
+    padding:12px 20px;font-size:12px;font-weight:600;color:#71717a;
+    text-transform:uppercase;letter-spacing:.8px;border-bottom:1px solid rgba(39,39,42,.6);
+}
+.panel-card-body{padding:16px 20px;display:flex;flex-direction:column;gap:8px}
+.modern-label{font-size:13px;font-weight:500;color:#a1a1aa;margin-bottom:4px;display:block}
+.modern-textarea{
+    width:100%;background:#09090b;border:1px solid #27272a;border-radius:8px;
+    padding:10px 14px;font-family:'Inter',sans-serif;font-size:14px;color:#e4e4e7;
+    resize:none;outline:none;min-height:100px;transition:border-color .2s;
+}
+.modern-textarea:focus{border-color:#FF1493;box-shadow:0 0 0 3px rgba(255,20,147,.15)}
+.modern-textarea::placeholder{color:#3f3f46}
+.modern-textarea.error-flash{
+    border-color:#ef4444!important;box-shadow:0 0 0 3px rgba(239,68,68,.2)!important;animation:shake .4s ease;
+}
+@keyframes shake{0%,100%{transform:translateX(0)}20%,60%{transform:translateX(-4px)}40%,80%{transform:translateX(4px)}}
+
+.toast-notification{
+    position:fixed;top:24px;left:50%;transform:translateX(-50%) translateY(-120%);
+    z-index:9999;padding:10px 24px;border-radius:10px;font-family:'Inter',sans-serif;
+    font-size:14px;font-weight:600;display:flex;align-items:center;gap:8px;
+    box-shadow:0 8px 24px rgba(0,0,0,.5);
+    transition:transform .35s cubic-bezier(.34,1.56,.64,1),opacity .35s ease;opacity:0;pointer-events:none;
+}
+.toast-notification.visible{transform:translateX(-50%) translateY(0);opacity:1;pointer-events:auto}
+.toast-notification.error{background:linear-gradient(135deg,#dc2626,#b91c1c);color:#fff;border:1px solid rgba(255,255,255,.15)}
+.toast-notification.warning{background:linear-gradient(135deg,#d97706,#b45309);color:#fff;border:1px solid rgba(255,255,255,.15)}
+.toast-notification.info{background:linear-gradient(135deg,#ec4899,#be185d);color:#fff;border:1px solid rgba(255,255,255,.15)}
+.toast-notification .toast-icon{font-size:16px;line-height:1}
+.toast-notification .toast-text{line-height:1.3}
+
+.btn-run{
+    display:flex;align-items:center;justify-content:center;gap:8px;width:100%;
+    background:linear-gradient(135deg,#FF1493,#D10073);border:none;border-radius:10px;
+    padding:12px 24px;cursor:pointer;font-size:15px;font-weight:600;font-family:'Inter',sans-serif;
+    color:#ffffff!important;-webkit-text-fill-color:#ffffff!important;
+    transition:all .2s ease;letter-spacing:-.2px;
+    box-shadow:0 4px 16px rgba(255,20,147,.3),inset 0 1px 0 rgba(255,255,255,.1);
+}
+.btn-run:hover{
+    background:linear-gradient(135deg,#ff4db2,#FF1493);transform:translateY(-1px);
+    box-shadow:0 6px 24px rgba(255,20,147,.45),inset 0 1px 0 rgba(255,255,255,.15);
+}
+.btn-run:active{transform:translateY(0);box-shadow:0 2px 8px rgba(255,20,147,.3)}
+#custom-run-btn,#custom-run-btn *,#run-btn-label,.btn-run,.btn-run *{
+    color:#ffffff!important;-webkit-text-fill-color:#ffffff!important;fill:#ffffff!important;
+}
+
+.output-frame{border-bottom:1px solid #27272a;display:flex;flex-direction:column;position:relative}
+.output-frame .out-title,
+.output-frame .out-title *,
+#output-title-label{
+    color:#ffffff!important;
+    -webkit-text-fill-color:#ffffff!important;
+}
+.output-frame .out-title{
+    padding:10px 20px;font-size:13px;font-weight:700;
+    text-transform:uppercase;letter-spacing:.8px;border-bottom:1px solid rgba(39,39,42,.6);
+    display:flex;align-items:center;justify-content:space-between;gap:8px;flex-wrap:wrap;
+}
+.out-title-right{display:flex;gap:8px;align-items:center}
+.out-action-btn{
+    display:inline-flex;align-items:center;justify-content:center;background:rgba(255,20,147,.1);
+    border:1px solid rgba(255,20,147,.2);border-radius:6px;cursor:pointer;padding:3px 10px;
+    font-size:11px;font-weight:500;color:#ff7ac7!important;gap:4px;height:24px;transition:all .15s;
+}
+.out-action-btn:hover{background:rgba(255,20,147,.2);border-color:rgba(255,20,147,.35);color:#ffffff!important}
+.out-action-btn svg{width:12px;height:12px;fill:#ff7ac7}
+.output-frame .out-body{
+    flex:1;background:#09090b;display:flex;align-items:stretch;justify-content:stretch;
+    overflow:hidden;min-height:320px;position:relative;
+}
+.output-scroll-wrap{
+    width:100%;height:100%;padding:0;overflow:hidden;
+}
+.output-textarea{
+    width:100%;height:320px;min-height:320px;max-height:320px;background:#09090b;color:#e4e4e7;
+    border:none;outline:none;padding:16px 18px;font-size:13px;line-height:1.6;
+    font-family:'JetBrains Mono',monospace;overflow:auto;resize:none;white-space:pre-wrap;
+}
+.output-textarea::placeholder{color:#52525b}
+.output-textarea.error-flash{
+    box-shadow:inset 0 0 0 2px rgba(239,68,68,.6);
+}
+.modern-loader{
+    display:none;position:absolute;top:0;left:0;right:0;bottom:0;background:rgba(9,9,11,.92);
+    z-index:15;flex-direction:column;align-items:center;justify-content:center;gap:16px;backdrop-filter:blur(4px);
+}
+.modern-loader.active{display:flex}
+.modern-loader .loader-spinner{
+    width:36px;height:36px;border:3px solid #27272a;border-top-color:#FF1493;
+    border-radius:50%;animation:spin .8s linear infinite;
+}
+@keyframes spin{to{transform:rotate(360deg)}}
+.modern-loader .loader-text{font-size:13px;color:#a1a1aa;font-weight:500}
+.loader-bar-track{width:200px;height:4px;background:#27272a;border-radius:2px;overflow:hidden}
+.loader-bar-fill{
+    height:100%;background:linear-gradient(90deg,#FF1493,#FF69C8,#FF1493);
+    background-size:200% 100%;animation:shimmer 1.5s ease-in-out infinite;border-radius:2px;
+}
+@keyframes shimmer{0%{background-position:200% 0}100%{background-position:-200% 0}}
+
+.settings-group{border:1px solid #27272a;border-radius:10px;margin:12px 16px;padding:0;overflow:hidden}
+.settings-group-title{
+    font-size:12px;font-weight:600;color:#71717a;text-transform:uppercase;letter-spacing:.8px;
+    padding:10px 16px;border-bottom:1px solid #27272a;background:rgba(24,24,27,.5);
+}
+.settings-group-body{padding:14px 16px;display:flex;flex-direction:column;gap:12px}
+.slider-row{display:flex;align-items:center;gap:10px;min-height:28px}
+.slider-row label{font-size:13px;font-weight:500;color:#a1a1aa;min-width:118px;flex-shrink:0}
+.slider-row input[type="range"]{
+    flex:1;-webkit-appearance:none;appearance:none;height:6px;background:#27272a;
+    border-radius:3px;outline:none;min-width:0;
+}
+.slider-row input[type="range"]::-webkit-slider-thumb{
+    -webkit-appearance:none;width:16px;height:16px;background:linear-gradient(135deg,#FF1493,#D10073);
+    border-radius:50%;cursor:pointer;box-shadow:0 2px 6px rgba(255,20,147,.4);transition:transform .15s;
+}
+.slider-row input[type="range"]::-webkit-slider-thumb:hover{transform:scale(1.2)}
+.slider-row input[type="range"]::-moz-range-thumb{
+    width:16px;height:16px;background:linear-gradient(135deg,#FF1493,#D10073);
+    border-radius:50%;cursor:pointer;border:none;box-shadow:0 2px 6px rgba(255,20,147,.4);
+}
+.slider-row .slider-val{
+    min-width:58px;text-align:right;font-family:'JetBrains Mono',monospace;font-size:12px;
+    font-weight:500;padding:3px 8px;background:#09090b;border:1px solid #27272a;
+    border-radius:6px;color:#a1a1aa;flex-shrink:0;
+}
+
+.app-statusbar{
+    background:#18181b;border-top:1px solid #27272a;padding:6px 20px;
+    display:flex;gap:12px;height:34px;align-items:center;font-size:12px;
+}
+.app-statusbar .sb-section{
+    padding:0 12px;flex:1;display:flex;align-items:center;font-family:'JetBrains Mono',monospace;
+    font-size:12px;color:#52525b;overflow:hidden;white-space:nowrap;
+}
+.app-statusbar .sb-section.sb-fixed{
+    flex:0 0 auto;min-width:110px;text-align:center;justify-content:center;
+    padding:3px 12px;background:rgba(255,20,147,.08);border-radius:6px;color:#ff7ac7;font-weight:500;
+}
+
+.exp-note{padding:10px 20px;font-size:12px;color:#52525b;border-top:1px solid #27272a;text-align:center}
+.exp-note a{color:#ff7ac7;text-decoration:none}
+.exp-note a:hover{text-decoration:underline}
+
+::-webkit-scrollbar{width:8px;height:8px}
+::-webkit-scrollbar-track{background:#09090b}
+::-webkit-scrollbar-thumb{background:#27272a;border-radius:4px}
+::-webkit-scrollbar-thumb:hover{background:#3f3f46}
+
+@media(max-width:980px){
+    .app-main-row{flex-direction:column}
+    .app-main-right{width:100%}
+    .app-main-left{border-right:none;border-bottom:1px solid #27272a}
+}
+"""
+
+gallery_js = r"""
+() => {
+function init() {
+    if (window.__docScopeInitDone) return;
+
+    const dropZone = document.getElementById('media-drop-zone');
+    const uploadPrompt = document.getElementById('upload-prompt');
+    const uploadClick = document.getElementById('upload-click-area');
+    const fileInput = document.getElementById('custom-file-input');
+    const previewWrap = document.getElementById('single-preview-wrap');
+    const previewImg = document.getElementById('single-preview-img');
+    const previewVideo = document.getElementById('single-preview-video');
+    const btnUpload = document.getElementById('preview-upload-btn');
+    const btnClear = document.getElementById('preview-clear-btn');
+    const promptInput = document.getElementById('custom-query-input');
+    const runBtnEl = document.getElementById('custom-run-btn');
+    const outputArea = document.getElementById('custom-output-textarea');
+    const mediaStatus = document.getElementById('sb-media-status');
+    const exampleResultContainer = document.getElementById('example-result-data');
+
+    if (!dropZone || !fileInput || !promptInput || !previewWrap || !previewImg || !previewVideo) {
+        setTimeout(init, 250);
+        return;
+    }
+
+    window.__docScopeInitDone = true;
+    let mediaState = null;
+    let currentMode = 'image';
+    let toastTimer = null;
+
+    function showToast(message, type) {
+        let toast = document.getElementById('app-toast');
+        if (!toast) {
+            toast = document.createElement('div');
+            toast.id = 'app-toast';
+            toast.className = 'toast-notification';
+            toast.innerHTML = '<span class="toast-icon"></span><span class="toast-text"></span>';
+            document.body.appendChild(toast);
+        }
+        const icon = toast.querySelector('.toast-icon');
+        const text = toast.querySelector('.toast-text');
+        toast.className = 'toast-notification ' + (type || 'error');
+        if (type === 'warning') icon.textContent = '\u26A0';
+        else if (type === 'info') icon.textContent = '\u2139';
+        else icon.textContent = '\u2717';
+        text.textContent = message;
+        if (toastTimer) clearTimeout(toastTimer);
+        void toast.offsetWidth;
+        toast.classList.add('visible');
+        toastTimer = setTimeout(() => toast.classList.remove('visible'), 3500);
+    }
+    window.__showToast = showToast;
+
+    function showLoader() {
+        const l = document.getElementById('output-loader');
+        if (l) l.classList.add('active');
+        const sb = document.getElementById('sb-run-state');
+        if (sb) sb.textContent = 'Processing...';
+    }
+    function hideLoader() {
+        const l = document.getElementById('output-loader');
+        if (l) l.classList.remove('active');
+        const sb = document.getElementById('sb-run-state');
+        if (sb) sb.textContent = 'Done';
+    }
+    window.__showLoader = showLoader;
+    window.__hideLoader = hideLoader;
+
+    function flashPromptError() {
+        promptInput.classList.add('error-flash');
+        promptInput.focus();
+        setTimeout(() => promptInput.classList.remove('error-flash'), 800);
+    }
+
+    function flashOutputError() {
+        if (!outputArea) return;
+        outputArea.classList.add('error-flash');
+        setTimeout(() => outputArea.classList.remove('error-flash'), 800);
+    }
+
+    function setGradioValue(containerId, value) {
+        const container = document.getElementById(containerId);
+        if (!container) return;
+        container.querySelectorAll('input, textarea').forEach(el => {
+            if (el.type === 'file' || el.type === 'range' || el.type === 'checkbox') return;
+            const proto = el.tagName === 'TEXTAREA' ? HTMLTextAreaElement.prototype : HTMLInputElement.prototype;
+            const ns = Object.getOwnPropertyDescriptor(proto, 'value');
+            if (ns && ns.set) {
+                ns.set.call(el, value);
+                el.dispatchEvent(new Event('input', {bubbles:true, composed:true}));
+                el.dispatchEvent(new Event('change', {bubbles:true, composed:true}));
+            }
+        });
+    }
+
+    function syncMediaToGradio() {
+        setGradioValue('hidden-image-b64', mediaState && mediaState.mode === 'image' ? mediaState.b64 : '');
+        setGradioValue('hidden-video-b64', mediaState && mediaState.mode === 'video' ? mediaState.b64 : '');
+        const txt = mediaState ? (`1 ${mediaState.mode} uploaded`) : `No ${currentMode} uploaded`;
+        if (mediaStatus) mediaStatus.textContent = txt;
+    }
+
+    function syncPromptToGradio() {
+        setGradioValue('prompt-gradio-input', promptInput.value);
+    }
+
+    function syncModelToGradio(name) {
+        setGradioValue('hidden-model-name', name);
+    }
+
+    function syncModeToGradio(mode) {
+        setGradioValue('hidden-mode-name', mode);
+    }
+
+    function renderPreview() {
+        if (!mediaState) {
+            previewImg.src = '';
+            previewVideo.src = '';
+            previewImg.style.display = 'none';
+            previewVideo.style.display = 'none';
+            previewWrap.style.display = 'none';
+            if (uploadPrompt) uploadPrompt.style.display = 'flex';
+            syncMediaToGradio();
+            return;
+        }
+
+        if (mediaState.mode === 'video') {
+            previewImg.src = '';
+            previewImg.style.display = 'none';
+            previewVideo.src = mediaState.b64;
+            previewVideo.style.display = 'block';
+            previewWrap.style.display = 'flex';
+        } else {
+            previewVideo.pause();
+            previewVideo.removeAttribute('src');
+            previewVideo.load();
+            previewVideo.style.display = 'none';
+            previewImg.src = mediaState.b64;
+            previewImg.style.display = 'block';
+            previewWrap.style.display = 'flex';
+        }
+        if (uploadPrompt) uploadPrompt.style.display = 'none';
+        syncMediaToGradio();
+    }
+
+    function setPreview(b64, name, mode) {
+        mediaState = {b64, name: name || 'file', mode: mode || currentMode};
+        renderPreview();
+    }
+    window.__setPreview = setPreview;
+
+    function clearPreview() {
+        mediaState = null;
+        renderPreview();
+    }
+    window.__clearPreview = clearPreview;
+
+    function processFile(file) {
+        if (!file) return;
+        if (currentMode === 'image' && !file.type.startsWith('image/')) {
+            showToast('Only image files are supported in Image mode', 'error');
+            return;
+        }
+        if (currentMode === 'video' && !file.type.startsWith('video/')) {
+            showToast('Only video files are supported in Video mode', 'error');
+            return;
+        }
+        const reader = new FileReader();
+        reader.onload = (e) => setPreview(e.target.result, file.name, currentMode);
+        reader.readAsDataURL(file);
+    }
+
+    fileInput.addEventListener('change', (e) => {
+        const file = e.target.files && e.target.files[0] ? e.target.files[0] : null;
+        if (file) processFile(file);
+        e.target.value = '';
+    });
+
+    function updateAccept() {
+        fileInput.accept = currentMode === 'video' ? 'video/*' : 'image/*';
+        const main = document.getElementById('upload-main-text');
+        const sub = document.getElementById('upload-sub-text');
+        if (main) main.textContent = currentMode === 'video' ? 'Click or drag a video here' : 'Click or drag an image here';
+        if (sub) sub.textContent = currentMode === 'video'
+            ? 'Upload one short video clip for document-aware video understanding'
+            : 'Upload one document, page, screenshot, receipt, or scene image for OCR and reasoning';
+        if (!mediaState && mediaStatus) mediaStatus.textContent = `No ${currentMode} uploaded`;
+    }
+
+    if (uploadClick) uploadClick.addEventListener('click', () => fileInput.click());
+    if (btnUpload) btnUpload.addEventListener('click', () => fileInput.click());
+    if (btnClear) btnClear.addEventListener('click', clearPreview);
+
+    dropZone.addEventListener('dragover', (e) => {
+        e.preventDefault();
+        dropZone.classList.add('drag-over');
+    });
+    dropZone.addEventListener('dragleave', (e) => {
+        e.preventDefault();
+        dropZone.classList.remove('drag-over');
+    });
+    dropZone.addEventListener('drop', (e) => {
+        e.preventDefault();
+        dropZone.classList.remove('drag-over');
+        if (e.dataTransfer.files && e.dataTransfer.files.length) processFile(e.dataTransfer.files[0]);
+    });
+
+    promptInput.addEventListener('input', syncPromptToGradio);
+
+    function activateModelTab(name) {
+        document.querySelectorAll('.model-tab[data-model]').forEach(btn => {
+            btn.classList.toggle('active', btn.getAttribute('data-model') === name);
+        });
+        syncModelToGradio(name);
+    }
+    window.__activateModelTab = activateModelTab;
+
+    function activateModeTab(mode) {
+        currentMode = mode;
+        document.querySelectorAll('.mode-tab[data-mode]').forEach(btn => {
+            btn.classList.toggle('active', btn.getAttribute('data-mode') === mode);
+        });
+        syncModeToGradio(mode);
+        updateAccept();
+        if (mediaState && mediaState.mode !== mode) clearPreview();
+    }
+    window.__activateModeTab = activateModeTab;
+
+    document.querySelectorAll('.model-tab[data-model]').forEach(btn => {
+        btn.addEventListener('click', () => activateModelTab(btn.getAttribute('data-model')));
+    });
+    document.querySelectorAll('.mode-tab[data-mode]').forEach(btn => {
+        btn.addEventListener('click', () => activateModeTab(btn.getAttribute('data-mode')));
+    });
+
+    activateModelTab('Cosmos-Reason1-7B');
+    activateModeTab('image');
+
+    function syncSlider(customId, gradioId) {
+        const slider = document.getElementById(customId);
+        const valSpan = document.getElementById(customId + '-val');
+        if (!slider) return;
+        slider.addEventListener('input', () => {
+            if (valSpan) valSpan.textContent = slider.value;
+            const container = document.getElementById(gradioId);
+            if (!container) return;
+            container.querySelectorAll('input[type="range"],input[type="number"]').forEach(el => {
+                const ns = Object.getOwnPropertyDescriptor(HTMLInputElement.prototype, 'value');
+                if (ns && ns.set) {
+                    ns.set.call(el, slider.value);
+                    el.dispatchEvent(new Event('input', {bubbles:true, composed:true}));
+                    el.dispatchEvent(new Event('change', {bubbles:true, composed:true}));
+                }
+            });
+        });
+    }
+
+    syncSlider('custom-max-new-tokens', 'gradio-max-new-tokens');
+    syncSlider('custom-temperature', 'gradio-temperature');
+    syncSlider('custom-top-p', 'gradio-top-p');
+    syncSlider('custom-top-k', 'gradio-top-k');
+    syncSlider('custom-repetition-penalty', 'gradio-repetition-penalty');
+    syncSlider('custom-gpu-duration', 'gradio-gpu-duration');
+
+    function validateBeforeRun() {
+        const promptVal = promptInput.value.trim();
+        if (!mediaState && !promptVal) {
+            showToast(`Please upload a ${currentMode} and enter your instruction`, 'error');
+            flashPromptError();
+            return false;
+        }
+        if (!mediaState) {
+            showToast(`Please upload a ${currentMode}`, 'error');
+            return false;
+        }
+        if (mediaState.mode !== currentMode) {
+            showToast(`Uploaded media does not match ${currentMode} mode`, 'error');
+            return false;
+        }
+        if (!promptVal) {
+            showToast('Please enter your instruction', 'warning');
+            flashPromptError();
+            return false;
+        }
+        const currentModel = (document.querySelector('.model-tab.active') || {}).dataset?.model;
+        if (!currentModel) {
+            showToast('Please select a model', 'error');
+            return false;
+        }
+        return true;
+    }
+
+    window.__clickGradioRunBtn = function() {
+        if (!validateBeforeRun()) return;
+        syncPromptToGradio();
+        syncMediaToGradio();
+        const activeModel = document.querySelector('.model-tab.active');
+        if (activeModel) syncModelToGradio(activeModel.getAttribute('data-model'));
+        const activeMode = document.querySelector('.mode-tab.active');
+        if (activeMode) syncModeToGradio(activeMode.getAttribute('data-mode'));
+        if (outputArea) outputArea.value = '';
+        showLoader();
+        setTimeout(() => {
+            const gradioBtn = document.getElementById('gradio-run-btn');
+            if (!gradioBtn) return;
+            const btn = gradioBtn.querySelector('button');
+            if (btn) btn.click(); else gradioBtn.click();
+        }, 180);
+    };
+
+    if (runBtnEl) runBtnEl.addEventListener('click', () => window.__clickGradioRunBtn());
+
+    const copyBtn = document.getElementById('copy-output-btn');
+    if (copyBtn) {
+        copyBtn.addEventListener('click', async () => {
+            try {
+                const text = outputArea ? outputArea.value : '';
+                if (!text.trim()) {
+                    showToast('No output to copy', 'warning');
+                    flashOutputError();
+                    return;
+                }
+                await navigator.clipboard.writeText(text);
+                showToast('Output copied to clipboard', 'info');
+            } catch(e) {
+                showToast('Copy failed', 'error');
+            }
+        });
+    }
+
+    const saveBtn = document.getElementById('save-output-btn');
+    if (saveBtn) {
+        saveBtn.addEventListener('click', () => {
+            const text = outputArea ? outputArea.value : '';
+            if (!text.trim()) {
+                showToast('No output to save', 'warning');
+                flashOutputError();
+                return;
+            }
+            const blob = new Blob([text], {type: 'text/plain;charset=utf-8'});
+            const a = document.createElement('a');
+            a.href = URL.createObjectURL(blob);
+            a.download = 'docscope_r1_output.txt';
+            document.body.appendChild(a);
+            a.click();
+            setTimeout(() => {
+                URL.revokeObjectURL(a.href);
+                document.body.removeChild(a);
+            }, 200);
+            showToast('Output saved', 'info');
+        });
+    }
+
+    document.querySelectorAll('.example-card[data-idx]').forEach(card => {
+        card.addEventListener('click', () => {
+            const idx = card.getAttribute('data-idx');
+            document.querySelectorAll('.example-card.loading').forEach(c => c.classList.remove('loading'));
+            card.classList.add('loading');
+            showToast('Loading example...', 'info');
+            setGradioValue('example-result-data', '');
+            setGradioValue('example-idx-input', idx);
+            setTimeout(() => {
+                const btn = document.getElementById('example-load-btn');
+                if (btn) {
+                    const b = btn.querySelector('button');
+                    if (b) b.click(); else btn.click();
+                }
+            }, 150);
+            setTimeout(() => card.classList.remove('loading'), 12000);
+        });
+    });
+
+    function checkExampleResult() {
+        if (!exampleResultContainer) return;
+        const el = exampleResultContainer.querySelector('textarea') || exampleResultContainer.querySelector('input');
+        if (!el || !el.value) return;
+        if (window.__lastExampleVal === el.value) return;
+        try {
+            const data = JSON.parse(el.value);
+            if (data.status === 'ok') {
+                window.__lastExampleVal = el.value;
+                if (data.mode) activateModeTab(data.mode);
+                if (data.media) setPreview(data.media, data.name || 'example', data.mode || 'image');
+                if (data.query) {
+                    promptInput.value = data.query;
+                    syncPromptToGradio();
+                }
+                if (data.model) activateModelTab(data.model);
+                document.querySelectorAll('.example-card.loading').forEach(c => c.classList.remove('loading'));
+                showToast('Example loaded', 'info');
+            } else if (data.status === 'error') {
+                document.querySelectorAll('.example-card.loading').forEach(c => c.classList.remove('loading'));
+                showToast(data.message || 'Failed to load example', 'error');
+            }
+        } catch(e) {}
+    }
+
+    const obsExample = new MutationObserver(checkExampleResult);
+    if (exampleResultContainer) {
+        obsExample.observe(exampleResultContainer, {childList:true, subtree:true, characterData:true, attributes:true});
+    }
+    setInterval(checkExampleResult, 500);
+
+    if (outputArea) outputArea.value = '';
+    const sb = document.getElementById('sb-run-state');
+    if (sb) sb.textContent = 'Ready';
+    if (mediaStatus) mediaStatus.textContent = 'No image uploaded';
+}
+init();
+}
+"""
+
+wire_outputs_js = r"""
+() => {
+function watchOutputs() {
+    const resultContainer = document.getElementById('gradio-result');
+    const outArea = document.getElementById('custom-output-textarea');
+    if (!resultContainer || !outArea) { setTimeout(watchOutputs, 500); return; }
+
+    let lastText = '';
+
+    function syncOutput() {
+        const el = resultContainer.querySelector('textarea') || resultContainer.querySelector('input');
+        if (!el) return;
+        const val = el.value || '';
+        if (val !== lastText) {
+            lastText = val;
+            outArea.value = val;
+            outArea.scrollTop = outArea.scrollHeight;
+            if (window.__hideLoader && val.trim()) window.__hideLoader();
+        }
+    }
+
+    const observer = new MutationObserver(syncOutput);
+    observer.observe(resultContainer, {childList:true, subtree:true, characterData:true, attributes:true});
+    setInterval(syncOutput, 500);
+}
+watchOutputs();
+}
+"""
+
+DOC_LOGO_SVG = """
+<svg viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg">
+  <path d="M7 3h7l5 5v11a2 2 0 0 1-2 2H7a2 2 0 0 1-2-2V5a2 2 0 0 1 2-2Zm7 1.5V9h4.5" fill="none" stroke="white" stroke-width="1.8" stroke-linecap="round" stroke-linejoin="round"/>
+  <path d="M9 12h6M9 15h6M9 18h4" fill="none" stroke="white" stroke-width="1.8" stroke-linecap="round"/>
+</svg>
+"""
+
+UPLOAD_PREVIEW_SVG = """
+<svg viewBox="0 0 80 80" fill="none" xmlns="http://www.w3.org/2000/svg">
+    <rect x="8" y="14" width="64" height="52" rx="6" fill="none" stroke="#FF1493" stroke-width="2" stroke-dasharray="4 3"/>
+    <polygon points="12,62 30,40 42,50 54,34 68,62" fill="rgba(255,20,147,0.15)" stroke="#FF1493" stroke-width="1.5"/>
+    <circle cx="28" cy="30" r="6" fill="rgba(255,20,147,0.2)" stroke="#FF1493" stroke-width="1.5"/>
+</svg>
+"""
+
+COPY_SVG = """<svg viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg"><path d="M16 1H4C2.9 1 2 1.9 2 3v12h2V3h12V1zm3 4H8C6.9 5 6 5.9 6 7v14c0 1.1.9 2 2 2h11c1.1 0 2-.9 2-2V7c0-1.1-.9-2-2-2zm0 16H8V7h11v14z"/></svg>"""
+SAVE_SVG = """<svg viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg"><path d="M17 3H5a2 2 0 0 0-2 2v14a2 2 0 0 0 2 2h14a2 2 0 0 0 2-2V7l-4-4zM7 5h8v4H7V5zm12 14H5v-6h14v6z"/></svg>"""
+
+MODEL_TABS_HTML = "".join([
+    f'<button class="model-tab{" active" if m == "Cosmos-Reason1-7B" else ""}" data-model="{m}"><span class="model-tab-label">{m}</span></button>'
+    for m in MODEL_CHOICES
+])
+
+MODE_TABS_HTML = """
+<button class="mode-tab active" data-mode="image">Image Inference</button>
+<button class="mode-tab" data-mode="video">Video Inference</button>
+"""
+
+with gr.Blocks() as demo:
+    hidden_mode_name = gr.Textbox(value="image", elem_id="hidden-mode-name", elem_classes="hidden-input", container=False)
+    hidden_image_b64 = gr.Textbox(value="", elem_id="hidden-image-b64", elem_classes="hidden-input", container=False)
+    hidden_video_b64 = gr.Textbox(value="", elem_id="hidden-video-b64", elem_classes="hidden-input", container=False)
+    prompt = gr.Textbox(value="", elem_id="prompt-gradio-input", elem_classes="hidden-input", container=False)
+    hidden_model_name = gr.Textbox(value="Cosmos-Reason1-7B", elem_id="hidden-model-name", elem_classes="hidden-input", container=False)
+
+    max_new_tokens = gr.Slider(minimum=1, maximum=MAX_MAX_NEW_TOKENS, step=1, value=DEFAULT_MAX_NEW_TOKENS, elem_id="gradio-max-new-tokens", elem_classes="hidden-input", container=False)
+    temperature = gr.Slider(minimum=0.1, maximum=4.0, step=0.1, value=0.6, elem_id="gradio-temperature", elem_classes="hidden-input", container=False)
+    top_p = gr.Slider(minimum=0.05, maximum=1.0, step=0.05, value=0.9, elem_id="gradio-top-p", elem_classes="hidden-input", container=False)
+    top_k = gr.Slider(minimum=1, maximum=1000, step=1, value=50, elem_id="gradio-top-k", elem_classes="hidden-input", container=False)
+    repetition_penalty = gr.Slider(minimum=1.0, maximum=2.0, step=0.05, value=1.2, elem_id="gradio-repetition-penalty", elem_classes="hidden-input", container=False)
+    gpu_duration_state = gr.Number(value=60, elem_id="gradio-gpu-duration", elem_classes="hidden-input", container=False)
+
+    result = gr.Textbox(value="", elem_id="gradio-result", elem_classes="hidden-input", container=False)
+
+    example_idx = gr.Textbox(value="", elem_id="example-idx-input", elem_classes="hidden-input", container=False)
+    example_result = gr.Textbox(value="", elem_id="example-result-data", elem_classes="hidden-input", container=False)
+    example_load_btn = gr.Button("Load Example", elem_id="example-load-btn")
+
+    gr.HTML(f"""
+    <div class="app-shell">
+        <div class="app-header">
+            <div class="app-header-left">
+                <div class="app-logo">{DOC_LOGO_SVG}</div>
+                <span class="app-title">DocScope R1</span>
+                <span class="app-badge">vision enabled</span>
+                <span class="app-badge fast">OCR + Reasoning</span>
+            </div>
+        </div>
+
+        <div class="model-tabs-bar">
+            {MODEL_TABS_HTML}
+        </div>
+
+        <div class="mode-tabs-bar">
+            {MODE_TABS_HTML}
+        </div>
+
+        <div class="app-main-row">
+            <div class="app-main-left">
+                <div id="media-drop-zone">
+                    <div id="upload-prompt" class="upload-prompt-modern">
+                        <div id="upload-click-area" class="upload-click-area">
+                            {UPLOAD_PREVIEW_SVG}
+                            <span id="upload-main-text" class="upload-main-text">Click or drag an image here</span>
+                            <span id="upload-sub-text" class="upload-sub-text">Upload one document, page, screenshot, receipt, or scene image for OCR and reasoning</span>
+                        </div>
+                    </div>
+
+                    <input id="custom-file-input" type="file" accept="image/*" style="display:none;" />
+
+                    <div id="single-preview-wrap" class="single-preview-wrap">
+                        <div class="single-preview-card">
+                            <img id="single-preview-img" src="" alt="Preview" style="display:none;">
+                            <video id="single-preview-video" controls playsinline style="display:none;"></video>
+                            <div class="preview-overlay-actions">
+                                <button id="preview-upload-btn" class="preview-action-btn" title="Replace">Upload</button>
+                                <button id="preview-clear-btn" class="preview-action-btn" title="Clear">Clear</button>
+                            </div>
+                        </div>
+                    </div>
+                </div>
+
+                <div class="hint-bar">
+                    <b>Upload:</b> Click or drag media into the panel &nbsp;&middot;&nbsp;
+                    <b>Mode:</b> Switch between image and video inference &nbsp;&middot;&nbsp;
+                    <b>Model:</b> Change models from the header &nbsp;&middot;&nbsp;
+                    <kbd>Clear</kbd> removes the current media
+                </div>
+
+                <div class="examples-section">
+                    <div class="examples-title">Quick Examples</div>
+                    <div class="examples-scroll">
+                        {EXAMPLE_CARDS_HTML}
+                    </div>
+                </div>
+            </div>
+
+            <div class="app-main-right">
+                <div class="panel-card">
+                    <div class="panel-card-title">Vision / OCR Instruction</div>
+                    <div class="panel-card-body">
+                        <label class="modern-label" for="custom-query-input">Query Input</label>
+                        <textarea id="custom-query-input" class="modern-textarea" rows="4" placeholder="e.g., perform OCR on this image, describe the document, explain the ad, summarize the video, identify visible text, analyze the scene..."></textarea>
+                    </div>
+                </div>
+
+                <div style="padding:12px 20px;">
+                    <button id="custom-run-btn" class="btn-run">
+                        <span id="run-btn-label">Run Inference</span>
+                    </button>
+                </div>
+
+                <div class="output-frame">
+                    <div class="out-title">
+                        <span id="output-title-label">Raw Output Stream</span>
+                        <div class="out-title-right">
+                            <button id="copy-output-btn" class="out-action-btn" title="Copy">{COPY_SVG} Copy</button>
+                            <button id="save-output-btn" class="out-action-btn" title="Save">{SAVE_SVG} Save File</button>
+                        </div>
+                    </div>
+                    <div class="out-body">
+                        <div class="modern-loader" id="output-loader">
+                            <div class="loader-spinner"></div>
+                            <div class="loader-text">Running inference...</div>
+                            <div class="loader-bar-track"><div class="loader-bar-fill"></div></div>
+                        </div>
+                        <div class="output-scroll-wrap">
+                            <textarea id="custom-output-textarea" class="output-textarea" placeholder="Raw output will appear here..." readonly></textarea>
+                        </div>
+                    </div>
+                </div>
+
+                <div class="settings-group">
+                    <div class="settings-group-title">Advanced Settings</div>
+                    <div class="settings-group-body">
+                        <div class="slider-row">
+                            <label>Max new tokens</label>
+                            <input type="range" id="custom-max-new-tokens" min="1" max="{MAX_MAX_NEW_TOKENS}" step="1" value="{DEFAULT_MAX_NEW_TOKENS}">
+                            <span class="slider-val" id="custom-max-new-tokens-val">{DEFAULT_MAX_NEW_TOKENS}</span>
+                        </div>
+                        <div class="slider-row">
+                            <label>Temperature</label>
+                            <input type="range" id="custom-temperature" min="0.1" max="4.0" step="0.1" value="0.6">
+                            <span class="slider-val" id="custom-temperature-val">0.6</span>
+                        </div>
+                        <div class="slider-row">
+                            <label>Top-p</label>
+                            <input type="range" id="custom-top-p" min="0.05" max="1.0" step="0.05" value="0.9">
+                            <span class="slider-val" id="custom-top-p-val">0.9</span>
+                        </div>
+                        <div class="slider-row">
+                            <label>Top-k</label>
+                            <input type="range" id="custom-top-k" min="1" max="1000" step="1" value="50">
+                            <span class="slider-val" id="custom-top-k-val">50</span>
+                        </div>
+                        <div class="slider-row">
+                            <label>Repetition penalty</label>
+                            <input type="range" id="custom-repetition-penalty" min="1.0" max="2.0" step="0.05" value="1.2">
+                            <span class="slider-val" id="custom-repetition-penalty-val">1.2</span>
+                        </div>
+                        <div class="slider-row">
+                            <label>GPU Duration (seconds)</label>
+                            <input type="range" id="custom-gpu-duration" min="60" max="300" step="30" value="60">
+                            <span class="slider-val" id="custom-gpu-duration-val">60</span>
+                        </div>
+                    </div>
+                </div>
+            </div>
+        </div>
+
+        <div class="exp-note">
+            Experimental document vision suite &middot; Open on <a href="https://github.com/PRITHIVSAKTHIUR/DocScope-R1" target="_blank">GitHub</a>
+        </div>
+
+        <div class="app-statusbar">
+            <div class="sb-section" id="sb-media-status">No image uploaded</div>
+            <div class="sb-section sb-fixed" id="sb-run-state">Ready</div>
+        </div>
+    </div>
+    """)
+
+    run_btn = gr.Button("Run", elem_id="gradio-run-btn")
+
+    demo.load(fn=noop, inputs=None, outputs=None, js=gallery_js)
+    demo.load(fn=noop, inputs=None, outputs=None, js=wire_outputs_js)
+
+    run_btn.click(
+        fn=run_inference,
+        inputs=[
+            hidden_mode_name,
+            hidden_model_name,
+            prompt,
+            hidden_image_b64,
+            hidden_video_b64,
+            max_new_tokens,
+            temperature,
+            top_p,
+            top_k,
+            repetition_penalty,
+            gpu_duration_state,
+        ],
+        outputs=[result],
+        js=r"""(mode, model, p, img, vid, mnt, t, tp, tk, rp, gd) => {
+            const modelEl = document.querySelector('.model-tab.active');
+            const modeEl = document.querySelector('.mode-tab.active');
+            const modelVal = modelEl ? modelEl.getAttribute('data-model') : model;
+            const modeVal = modeEl ? modeEl.getAttribute('data-mode') : mode;
+            const promptEl = document.getElementById('custom-query-input');
+            const promptVal = promptEl ? promptEl.value : p;
+
+            let imgVal = img;
+            let vidVal = vid;
+
+            const imgContainer = document.getElementById('hidden-image-b64');
+            const vidContainer = document.getElementById('hidden-video-b64');
+
+            if (imgContainer) {
+                const inner = imgContainer.querySelector('textarea, input');
+                if (inner) imgVal = inner.value;
+            }
+            if (vidContainer) {
+                const inner = vidContainer.querySelector('textarea, input');
+                if (inner) vidVal = inner.value;
+            }
+
+            return [modeVal, modelVal, promptVal, imgVal, vidVal, mnt, t, tp, tk, rp, gd];
+        }""",
     )
-    video_submit.click(
-        fn=generate_video,
-        inputs=[model_choice, video_query, video_upload, max_new_tokens, temperature, top_p, top_k, repetition_penalty, gpu_duration_state],
-        outputs=[raw_output, markdown_output]
+
+    example_load_btn.click(
+        fn=load_example_data,
+        inputs=[example_idx],
+        outputs=[example_result],
+        queue=False,
     )
 
 if __name__ == "__main__":
-    demo.queue(max_size=30).launch(css=css, theme=steel_blue_theme, mcp_server=True, ssr_mode=False, show_error=True)
+    demo.queue(max_size=30).launch(
+        css=css,
+        mcp_server=True,
+        ssr_mode=False,
+        show_error=True,
+        allowed_paths=["images", "videos"],
+    )