update app

app.py

@@ -1,143 +1,25 @@
 import os
-import sys
-import random
-import uuid
+import gc
 import json
+import base64
 import time
+from io import BytesIO
 from threading import Thread
-from typing import Iterable
-from huggingface_hub import snapshot_download
 
 import gradio as gr
 import spaces
 import torch
-import numpy as np
 from PIL import Image
-import cv2
 
 from transformers import (
     Qwen2_5_VLForConditionalGeneration,
+    Qwen3_5ForConditionalGeneration,
     Qwen3VLForConditionalGeneration,
-    AutoModelForImageTextToText,
     AutoModelForCausalLM,
     AutoProcessor,
     TextIteratorStreamer,
 )
 
-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)",
-            button_secondary_text_color="black",
-            button_secondary_text_color_hover="white",
-            button_secondary_background_fill="linear-gradient(90deg, *primary_300, *primary_300)",
-            button_secondary_background_fill_hover="linear-gradient(90deg, *primary_400, *primary_400)",
-            button_secondary_background_fill_dark="linear-gradient(90deg, *primary_500, *primary_600)",
-            button_secondary_background_fill_hover_dark="linear-gradient(90deg, *primary_500, *primary_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 = 4096
 DEFAULT_MAX_NEW_TOKENS = 2048
@@ -153,82 +35,8 @@ print("cuda device count:", torch.cuda.device_count())
 if torch.cuda.is_available():
     print("current device:", torch.cuda.current_device())
     print("device name:", torch.cuda.get_device_name(torch.cuda.current_device()))
-
 print("Using device:", device)
 
-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)
 
 MODEL_ID_V = "datalab-to/chandra"
 processor_v = AutoProcessor.from_pretrained(MODEL_ID_V, trust_remote_code=True)
@@ -248,7 +56,7 @@ model_x = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     torch_dtype=torch.bfloat16,
 ).to(device).eval()
 
-MODEL_PATH_D = "prithivMLmods/Dots.OCR-Latest-BF16" # -> alt of [rednote-hilab/dots.ocr]
+MODEL_PATH_D = "prithivMLmods/Dots.OCR-Latest-BF16"
 processor_d = AutoProcessor.from_pretrained(MODEL_PATH_D, trust_remote_code=True)
 model_d = AutoModelForCausalLM.from_pretrained(
     MODEL_PATH_D,
@@ -267,42 +75,127 @@ model_m = Qwen2_5_VLForConditionalGeneration.from_pretrained(
     torch_dtype=torch.float16
 ).to(device).eval()
 
-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."""
+MODEL_ID_C = "datalab-to/chandra-ocr-2"
+processor_c = AutoProcessor.from_pretrained(MODEL_ID_C, trust_remote_code=True)
+model_c = Qwen3_5ForConditionalGeneration.from_pretrained(
+    MODEL_ID_C,
+    attn_implementation="kernels-community/flash-attn2",
+    trust_remote_code=True,
+    torch_dtype=torch.float16
+).to(device).eval()
+
+MODEL_MAP = {
+    "Chandra-OCR-2": (processor_c, model_c),
+    "Nanonets-OCR2-3B": (processor_x, model_x),
+    "Chandra-OCR": (processor_v, model_v),
+    "Dots.OCR": (processor_d, model_d),
+    "olmOCR-2-7B-1025": (processor_m, model_m),
+}
+
+MODEL_CHOICES = list(MODEL_MAP.keys())
+
+image_examples = [
+    {"query": "Convert to Markdown.", "image": "examples/3.jpg", "model": "Chandra-OCR-2"},
+    {"query": "Perform OCR on the image. [Markdown]", "image": "examples/1.jpg", "model": "Nanonets-OCR2-3B"},
+    {"query": "Extract the contents. [Markdown].", "image": "examples/2.jpg", "model": "olmOCR-2-7B-1025"},
+    {"query": "OCR the Image", "image": "examples/4.jpg", "model": "Chandra-OCR"},
+]
+
+
+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",
+    }.get(ext, "image/jpeg")
+    with open(path, "rb") as f:
+        data = base64.b64encode(f.read()).decode()
+    return f"data:{mime};base64,{data}"
+
+
+def make_thumb_b64(path, max_dim=240):
+    try:
+        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(image_examples):
+        thumb = make_thumb_b64(ex["image"])
+        prompt_short = ex["query"][:72] + ("..." if len(ex["query"]) > 72 else "")
+        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>
+            <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(image_examples):
+        return json.dumps({"status": "error", "message": "Example index out of range"})
+    ex = image_examples[idx]
+    img_b64 = file_to_data_url(ex["image"])
+    if not img_b64:
+        return json.dumps({"status": "error", "message": "Could not load example image"})
+    return json.dumps({
+        "status": "ok",
+        "query": ex["query"],
+        "image": img_b64,
+        "model": ex["model"],
+        "name": os.path.basename(ex["image"]),
+    })
+
+
+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
 
-@spaces.GPU(duration=calc_timeout_image)
-def generate_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 = 60):
-    """
-    Generates responses using the selected model for image input.
-    Yields raw text and Markdown-formatted text.
-    """
-    if model_name == "olmOCR-2-7B-1025":
-        processor = processor_m
-        model = model_m
-    elif model_name == "Nanonets-OCR2-3B":
-        processor = processor_x
-        model = model_x
-    elif model_name == "Chandra-OCR":
-        processor = processor_v
-        model = model_v
-    elif model_name == "Dots.OCR":
-        processor = processor_d
-        model = model_d
-    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, temperature, top_p, top_k, repetition_penalty, gpu_timeout=60):
+    if 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 OCR/query 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",
@@ -317,90 +210,986 @@ def generate_image(model_name: str, text: str, image: Image.Image,
         text=[prompt_full],
         images=[image],
         return_tensors="pt",
-        padding=True).to(device)
+        padding=True
+    ).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 = buffer.replace("<|im_end|>", "")
         time.sleep(0.01)
-        yield buffer, buffer
+        yield buffer
 
-image_examples = [
-    ["Convert to Markdown.", "examples/3.jpg"],
-    ["Perform OCR on the image. [Markdown]", "examples/1.jpg"],
-    ["Extract the contents. [Markdown].", "examples/2.jpg"],
-]
+    gc.collect()
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
+
+
+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,#00FFFF,#4DFFFF,#99FFFF);
+    border-radius:10px;display:flex;align-items:center;justify-content:center;
+    box-shadow:0 4px 12px rgba(0,255,255,.30);
+}
+.app-logo svg{width:22px;height:22px;fill:#0b0f12;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(0,255,255,.10);color:#9EFFFF;border:1px solid rgba(0,255,255,.24);letter-spacing:.3px;
+}
+.app-badge.fast{background:rgba(0,255,255,.08);color:#6AFFFF;border:1px solid rgba(0,255,255,.20)}
+
+.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(0,255,255,.10);border-color:rgba(0,255,255,.35)}
+.model-tab.active{background:rgba(0,255,255,.16);border-color:#00FFFF;color:#fff!important;box-shadow:0 0 0 2px rgba(0,255,255,.08)}
+.model-tab-label{font-size:12px;color:#ffffff!important;font-weight:600}
+
+.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}
+
+#image-drop-zone{
+    position:relative;background:#09090b;height:440px;min-height:440px;max-height:440px;
+    overflow:hidden;
+}
+#image-drop-zone.drag-over{outline:2px solid #00FFFF;outline-offset:-2px;background:rgba(0,255,255,.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(0,255,255,.03);transition:all .2s ease;gap:8px;text-align:center;
+    overflow:hidden;
+}
+.upload-click-area:hover{background:rgba(0,255,255,.08);border-color:#00FFFF;transform:scale(1.02)}
+.upload-click-area:active{background:rgba(0,255,255,.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{
+    width:100%;height:100%;max-width:100%;max-height:100%;
+    object-fit:contain;display:block;
+}
+.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:#00FFFF;border-color:#00FFFF;color:#0b0f12!important}
+
+.hint-bar{
+    background:rgba(0,255,255,.05);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:#8EFFFF;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{
+    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:#00FFFF;transform:translateY(-2px);box-shadow:0 4px 12px rgba(0,255,255,.14)}
+.example-card.loading{opacity:.5;pointer-events:none}
+.example-thumb-wrap{height:120px;overflow:hidden;background:#18181b}
+.example-thumb-wrap img{width:100%;height:100%;object-fit:cover}
+.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(0,255,255,.12);border-radius:4px;
+    font-size:10px;font-weight:600;color:#8EFFFF;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:#00FFFF;box-shadow:0 0 0 3px rgba(0,255,255,.14)}
+.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,#0891b2,#0e7490);color:#fff;border:1px solid rgba(255,255,255,.15)}
+.toast-notification.info{background:linear-gradient(135deg,#06b6d4,#0891b2);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,#00FFFF,#00D9D9);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(0,255,255,.25),inset 0 1px 0 rgba(255,255,255,.18);
+}
+.btn-run:hover{
+    background:linear-gradient(135deg,#5EFFFF,#00FFFF);transform:translateY(-1px);
+    box-shadow:0 6px 24px rgba(0,255,255,.35),inset 0 1px 0 rgba(255,255,255,.22);
+}
+.btn-run:active{transform:translateY(0);box-shadow:0 2px 8px rgba(0,255,255,.25)}
+#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;
+}
+body:not(.dark) .btn-run,body:not(.dark) .btn-run *,
+.dark .btn-run,.dark .btn-run *,
+.gradio-container .btn-run,.gradio-container .btn-run *,
+.gradio-container #custom-run-btn,.gradio-container #custom-run-btn *{
+    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(0,255,255,.10);
+    border:1px solid rgba(0,255,255,.2);border-radius:6px;cursor:pointer;padding:3px 10px;
+    font-size:11px;font-weight:500;color:#8EFFFF!important;gap:4px;height:24px;transition:all .15s;
+}
+.out-action-btn:hover{background:rgba(0,255,255,.2);border-color:rgba(0,255,255,.35);color:#ffffff!important}
+.out-action-btn svg{width:12px;height:12px;fill:#8EFFFF}
+.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:#00FFFF;
+    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,#00FFFF,#6AFFFF,#00FFFF);
+    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,#00FFFF,#00D9D9);
+    border-radius:50%;cursor:pointer;box-shadow:0 2px 6px rgba(0,255,255,.35);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,#00FFFF,#00D9D9);
+    border-radius:50%;cursor:pointer;border:none;box-shadow:0 2px 6px rgba(0,255,255,.35);
+}
+.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(0,255,255,.08);border-radius:6px;color:#8EFFFF;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:#8EFFFF;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.__ocr3InitDone) return;
+
+    const dropZone = document.getElementById('image-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 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 imgStatus = document.getElementById('sb-image-status');
+    const exampleResultContainer = document.getElementById('example-result-data');
+
+    if (!dropZone || !fileInput || !promptInput || !previewWrap || !previewImg) {
+        setTimeout(init, 250);
+        return;
+    }
+
+    window.__ocr3InitDone = true;
+    let imageState = null;
+    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 syncImageToGradio() {
+        setGradioValue('hidden-image-b64', imageState ? imageState.b64 : '');
+        const txt = imageState ? '1 image uploaded' : 'No image uploaded';
+        if (imgStatus) imgStatus.textContent = txt;
+    }
+
+    function syncPromptToGradio() {
+        setGradioValue('prompt-gradio-input', promptInput.value);
+    }
+
+    function syncModelToGradio(name) {
+        setGradioValue('hidden-model-name', name);
+    }
+
+    function setPreview(b64, name) {
+        imageState = {b64, name: name || 'image'};
+        previewImg.src = b64;
+        previewWrap.style.display = 'flex';
+        if (uploadPrompt) uploadPrompt.style.display = 'none';
+        syncImageToGradio();
+    }
+    window.__setPreview = setPreview;
+
+    function clearPreview() {
+        imageState = null;
+        previewImg.src = '';
+        previewWrap.style.display = 'none';
+        if (uploadPrompt) uploadPrompt.style.display = 'flex';
+        syncImageToGradio();
+    }
+    window.__clearPreview = clearPreview;
+
+    function processFile(file) {
+        if (!file) return;
+        if (!file.type.startsWith('image/')) {
+            showToast('Only image files are supported', 'error');
+            return;
+        }
+        const reader = new FileReader();
+        reader.onload = (e) => setPreview(e.target.result, file.name);
+        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 = '';
+    });
+
+    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;
+
+    document.querySelectorAll('.model-tab[data-model]').forEach(btn => {
+        btn.addEventListener('click', () => {
+            const model = btn.getAttribute('data-model');
+            activateModelTab(model);
+        });
+    });
+
+    activateModelTab('Chandra-OCR-2');
+
+    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 (!imageState && !promptVal) {
+            showToast('Please upload an image and enter your OCR instruction', 'error');
+            flashPromptError();
+            return false;
+        }
+        if (!imageState) {
+            showToast('Please upload an image', 'error');
+            return false;
+        }
+        if (!promptVal) {
+            showToast('Please enter your OCR/query 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();
+        syncImageToGradio();
+        const active = document.querySelector('.model-tab.active');
+        if (active) syncModelToGradio(active.getAttribute('data-model'));
+        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 = 'multimodal_ocr3_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.__lastExampleVal3 === el.value) return;
+        try {
+            const data = JSON.parse(el.value);
+            if (data.status === 'ok') {
+                window.__lastExampleVal3 = el.value;
+                if (data.image) setPreview(data.image, data.name || 'example.jpg');
+                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 (imgStatus) imgStatus.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();
+}
+"""
+
+OCR_LOGO_SVG = """
+<svg viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg">
+  <path d="M4 5.5A2.5 2.5 0 0 1 6.5 3H11v2H6.5a.5.5 0 0 0-.5.5V10H4V5.5Z"/>
+  <path d="M20 10h-2V5.5a.5.5 0 0 0-.5-.5H13V3h4.5A2.5 2.5 0 0 1 20 5.5V10Z"/>
+  <path d="M4 14h2v4.5a.5.5 0 0 0 .5.5H11v2H6.5A2.5 2.5 0 0 1 4 18.5V14Z"/>
+  <path d="M20 14v4.5A2.5 2.5 0 0 1 17.5 21H13v-2h4.5a.5.5 0 0 0 .5-.5V14h2Z"/>
+  <path d="M8 8h8v2H8V8Zm0 3h8v2H8v-2Zm0 3h5v2H8v-2Z"/>
+</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="#00FFFF" stroke-width="2" stroke-dasharray="4 3"/>
+    <polygon points="12,62 30,40 42,50 54,34 68,62" fill="rgba(0,255,255,0.14)" stroke="#00FFFF" stroke-width="1.5"/>
+    <circle cx="28" cy="30" r="6" fill="rgba(0,255,255,0.2)" stroke="#00FFFF" 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 == "Chandra-OCR-2" else ""}" data-model="{m}"><span class="model-tab-label">{m}</span></button>'
+    for m in MODEL_CHOICES
+])
 
 with gr.Blocks() as demo:
-    gr.Markdown("# **Multimodal OCR3**", elem_id="main-title")
-    with gr.Row():
-        with gr.Column(scale=2):
-            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.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.7)
-                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.1)
-                
-        with gr.Column(scale=3):
-            gr.Markdown("## Output", elem_id="output-title")
-            output = gr.Textbox(label="Raw Output Stream", interactive=True, lines=15)
-            with gr.Accordion("(Result.md)", open=False):
-                markdown_output = gr.Markdown(label="(Result.Md)")
-
-            model_choice = gr.Radio(
-                choices=["Nanonets-OCR2-3B", "Chandra-OCR", "Dots.OCR", "olmOCR-2-7B-1025"],
-                label="Select Model",
-                value="Nanonets-OCR2-3B"
-            )
-            
-            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"
+    hidden_image_b64 = gr.Textbox(value="", elem_id="hidden-image-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="Chandra-OCR-2", 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.7, 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.1, 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">{OCR_LOGO_SVG}</div>
+                <span class="app-title">Multimodal OCR3</span>
+                <span class="app-badge">vision enabled</span>
+                <span class="app-badge fast">OCR Suite</span>
+            </div>
+        </div>
+
+        <div class="model-tabs-bar">
+            {MODEL_TABS_HTML}
+        </div>
+
+        <div class="app-main-row">
+            <div class="app-main-left">
+                <div id="image-drop-zone">
+                    <div id="upload-prompt" class="upload-prompt-modern">
+                        <div id="upload-click-area" class="upload-click-area">
+                            {UPLOAD_PREVIEW_SVG}
+                            <span class="upload-main-text">Click or drag an image here</span>
+                            <span class="upload-sub-text">Upload one document, page, receipt, screenshot, or scene image for OCR and multimodal understanding</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">
+                            <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 to add an image &nbsp;&middot;&nbsp;
+                    <b>Model:</b> Switch model tabs from the header &nbsp;&middot;&nbsp;
+                    <kbd>Clear</kbd> removes the current image
+                </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">OCR / Vision 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., convert to markdown, extract the contents, OCR the image, read all visible text, preserve layout..."></textarea>
+                    </div>
+                </div>
+
+                <div style="padding:12px 20px;">
+                    <button id="custom-run-btn" class="btn-run">
+                        <span id="run-btn-label">Run OCR</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 OCR...</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.7">
+                            <span class="slider-val" id="custom-temperature-val">0.7</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.1">
+                            <span class="slider-val" id="custom-repetition-penalty-val">1.1</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 OCR Suite &middot; Open on <a href="https://github.com/PRITHIVSAKTHIUR/Multimodal-OCR3" target="_blank">GitHub</a>
+        </div>
+
+        <div class="app-statusbar">
+            <div class="sb-section" id="sb-image-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")
+
+    def b64_to_pil(b64_str):
+        if not b64_str:
+            return None
+        try:
+            if b64_str.startswith("data:image"):
+                _, 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 run_ocr(model_name, text, image_b64, max_new_tokens_v, temperature_v, top_p_v, top_k_v, repetition_penalty_v, gpu_timeout_v):
+        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,
+        )
+
+    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_ocr,
+        inputs=[
+            hidden_model_name,
+            prompt,
+            hidden_image_b64,
+            max_new_tokens,
+            temperature,
+            top_p,
+            top_k,
+            repetition_penalty,
+            gpu_duration_state,
+        ],
+        outputs=[result],
+        js=r"""(m, p, img, mnt, t, tp, tk, rp, gd) => {
+            const modelEl = document.querySelector('.model-tab.active');
+            const model = modelEl ? modelEl.getAttribute('data-model') : m;
+            const promptEl = document.getElementById('custom-query-input');
+            const promptVal = promptEl ? promptEl.value : p;
+            const imgContainer = document.getElementById('hidden-image-b64');
+            let imgVal = img;
+            if (imgContainer) {
+                const inner = imgContainer.querySelector('textarea, input');
+                if (inner) imgVal = inner.value;
+            }
+            return [model, promptVal, imgVal, mnt, t, tp, tk, rp, gd];
+        }""",
     )
 
-    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=[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=50).launch(css=css, theme=steel_blue_theme, mcp_server=True, ssr_mode=False, show_error=True)
+    demo.queue(max_size=50).launch(
+        css=css,
+        mcp_server=True,
+        ssr_mode=False,
+        show_error=True,
+        allowed_paths=["examples"],
+    )