update app

app.py

@@ -104,8 +104,8 @@ except Exception as e:
 
 ADAPTER_SPECS = {
     "Object-Remover": {
-        "repo": "prithivMLmods/QIE-2511-Object-Remover-v2",
-        "weights": "Qwen-Image-Edit-2511-Object-Remover-v2-9200.safetensors",
+        "repo": "prithivMLmods/QIE-2509-Object-Remover-Bbox",
+        "weights": "QIE-2509-Object-Remover-Bbox-5000.safetensors",
         "adapter_name": "object-remover",
     },
 }
@@ -115,6 +115,7 @@ DEFAULT_PROMPT = "Remove the red highlighted object from the scene"
 
 
 def burn_boxes_onto_image(pil_image: Image.Image, boxes_json_str: str) -> Image.Image:
+    """Burn red outline-only rectangles onto the image (no fill)."""
     import json
     if not pil_image:
         return pil_image
@@ -125,10 +126,9 @@ def burn_boxes_onto_image(pil_image: Image.Image, boxes_json_str: str) -> Image.
     if not boxes:
         return pil_image
 
-    img = pil_image.copy().convert("RGBA")
+    img = pil_image.copy().convert("RGB")
     w, h = img.size
-    overlay = Image.new("RGBA", img.size, (0, 0, 0, 0))
-    draw = ImageDraw.Draw(overlay)
+    draw = ImageDraw.Draw(img)
     bw = max(3, w // 250)
 
     for b in boxes:
@@ -138,10 +138,10 @@ def burn_boxes_onto_image(pil_image: Image.Image, boxes_json_str: str) -> Image.
         y2 = int(b["y2"] * h)
         lx, rx = min(x1, x2), max(x1, x2)
         ty, by_ = min(y1, y2), max(y1, y2)
-        draw.rectangle([lx, ty, rx, by_], fill=(255, 0, 0, 90))
-        draw.rectangle([lx, ty, rx, by_], outline=(255, 0, 0, 255), width=bw)
+        # Red outline only — no fill
+        draw.rectangle([lx, ty, rx, by_], outline=(255, 0, 0), width=bw)
 
-    return Image.alpha_composite(img, overlay).convert("RGB")
+    return img
 
 
 @spaces.GPU
@@ -341,7 +341,6 @@ label{font-weight:600!important;color:#333!important}
 }
 .dark #bbox-debug-count{color:#C084FC;background:rgba(168,85,247,.12)}
 
-/* ===== FIX: hide the sync textbox visually but keep it in the DOM ===== */
 #boxes-json-input{
     max-height:0!important;
     overflow:hidden!important;
@@ -372,7 +371,7 @@ bbox_drawer_js = r"""
         const btnClear  = document.getElementById('tb-clear');
 
         if (!canvas || !wrap || !debugCount || !btnDraw) {
-            console.log('[BBox] waiting for DOM …');
+            console.log('[BBox] waiting for DOM...');
             setTimeout(initCanvasBbox, 250);
             return;
         }
@@ -394,8 +393,10 @@ bbox_drawer_js = r"""
         let dragStart = {x:0, y:0};
         let dragOrig  = null;
         const HANDLE  = 7;
+        const RED_STROKE = 'rgba(255,0,0,0.95)';
+        const RED_STROKE_WIDTH = 3;
+        const SEL_STROKE = 'rgba(0,120,255,0.95)';
 
-        /* ── helpers ── */
         function n2px(b) {
             return {x1:b.x1*dispW, y1:b.y1*dispH, x2:b.x2*dispW, y2:b.y2*dispH};
         }
@@ -422,12 +423,10 @@ bbox_drawer_js = r"""
                     y: Math.max(0,Math.min(dispH, cy-r.top))};
         }
 
-        /* ── FIX: always update debug + badge, then try textbox ── */
         function syncToGradio() {
             window.__bboxBoxes = boxes;
             const jsonStr = JSON.stringify(boxes);
 
-            /* 1) update on-screen debug no matter what */
             if (debugCount) {
                 debugCount.textContent = boxes.length > 0
                     ? '\u2705 ' + boxes.length + ' box' + (boxes.length > 1 ? 'es' : '') +
@@ -436,10 +435,9 @@ bbox_drawer_js = r"""
                     : '\u2B1C No boxes drawn yet';
             }
 
-            /* 2) try to push into the Gradio Textbox (best-effort) */
             const container = document.getElementById('boxes-json-input');
             if (!container) {
-                console.warn('[BBox] #boxes-json-input not in DOM (hidden?)');
+                console.warn('[BBox] #boxes-json-input not in DOM');
                 return;
             }
             const targets = [
@@ -461,7 +459,6 @@ bbox_drawer_js = r"""
             });
         }
 
-        /* ── draw ── */
         function placeholder() {
             ctx.fillStyle='#2a2a2a'; ctx.fillRect(0,0,dispW,dispH);
             ctx.strokeStyle='#444'; ctx.lineWidth=2; ctx.setLineDash([8,4]);
@@ -482,19 +479,20 @@ bbox_drawer_js = r"""
                 const p = n2px(b);
                 const lx=p.x1, ty=p.y1, w=p.x2-p.x1, h=p.y2-p.y1;
 
-                ctx.fillStyle = 'rgba(255,0,0,0.25)';
-                ctx.fillRect(lx,ty,w,h);
-
+                /* RED OUTLINE ONLY — no fill */
                 if (i === selectedIdx) {
-                    ctx.strokeStyle = 'rgba(0,120,255,0.95)';
-                    ctx.lineWidth = 2.5; ctx.setLineDash([6,3]);
+                    ctx.strokeStyle = SEL_STROKE;
+                    ctx.lineWidth = RED_STROKE_WIDTH + 1;
+                    ctx.setLineDash([6,3]);
                 } else {
-                    ctx.strokeStyle = 'rgba(255,0,0,0.9)';
-                    ctx.lineWidth = 2.5; ctx.setLineDash([]);
+                    ctx.strokeStyle = RED_STROKE;
+                    ctx.lineWidth = RED_STROKE_WIDTH;
+                    ctx.setLineDash([]);
                 }
-                ctx.strokeRect(lx,ty,w,h);
+                ctx.strokeRect(lx, ty, w, h);
                 ctx.setLineDash([]);
 
+                /* label tag */
                 ctx.fillStyle = i===selectedIdx ? 'rgba(0,120,255,0.85)' : 'rgba(255,0,0,0.85)';
                 ctx.font = 'bold 11px IBM Plex Mono,monospace';
                 ctx.textAlign = 'left'; ctx.textBaseline = 'top';
@@ -507,16 +505,17 @@ bbox_drawer_js = r"""
                 if (i === selectedIdx) drawHandles(p);
             });
 
+            /* temp drawing rect — outline only */
             if (tempRect) {
                 const rx = Math.min(tempRect.x1,tempRect.x2);
                 const ry = Math.min(tempRect.y1,tempRect.y2);
                 const rw = Math.abs(tempRect.x2-tempRect.x1);
                 const rh = Math.abs(tempRect.y2-tempRect.y1);
-                ctx.fillStyle='rgba(255,0,0,0.12)';
-                ctx.fillRect(rx,ry,rw,rh);
-                ctx.strokeStyle='rgba(255,0,0,0.7)';
-                ctx.lineWidth=2; ctx.setLineDash([6,3]);
-                ctx.strokeRect(rx,ry,rw,rh); ctx.setLineDash([]);
+                ctx.strokeStyle = RED_STROKE;
+                ctx.lineWidth = RED_STROKE_WIDTH;
+                ctx.setLineDash([6,3]);
+                ctx.strokeRect(rx, ry, rw, rh);
+                ctx.setLineDash([]);
             }
             updateBadge();
         }
@@ -582,7 +581,6 @@ bbox_drawer_js = r"""
         }
         function hideStatus() { status.style.display = 'none'; }
 
-        /* ── pointer events ── */
         function onDown(e) {
             if (!baseImg) return;
             e.preventDefault();
@@ -711,7 +709,6 @@ bbox_drawer_js = r"""
         canvas.addEventListener('touchend',   onUp,   {passive:false});
         canvas.addEventListener('touchcancel',(e)=>{e.preventDefault();dragging=false;redraw();},{passive:false});
 
-        /* ── toolbar ── */
         btnDraw.addEventListener('click',   ()=>setMode('draw'));
         btnSelect.addEventListener('click', ()=>setMode('select'));
 
@@ -739,16 +736,14 @@ bbox_drawer_js = r"""
         });
 
         btnClear.addEventListener('click', () => {
-            boxes.length = 0;            /* clear in-place */
+            boxes.length = 0;
             window.__bboxBoxes = boxes;
             selectedIdx = -1;
             syncToGradio(); redraw(); hideStatus();
         });
 
-        /* ── FIX: robust image polling with multiple selectors ── */
         let lastSrc = null;
         setInterval(() => {
-            /* try several selectors Gradio might use */
             const imgs = document.querySelectorAll('#source-image-component img');
             let el = null;
             for (const img of imgs) {
@@ -794,7 +789,6 @@ bbox_drawer_js = r"""
             }
         }, 300);
 
-        /* periodic re-sync every 500 ms */
         setInterval(() => { syncToGradio(); }, 500);
 
         new ResizeObserver(() => {
@@ -811,10 +805,11 @@ bbox_drawer_js = r"""
 """
 
 
-with gr.Blocks() as demo:
+with gr.Blocks(css=css, theme=purple_theme) as demo:
     gr.Markdown("# **QIE-Object-Remover-Bbox**", elem_id="main-title")
     gr.Markdown(
-        "Perform diverse image edits using a specialized [LoRA](https://huggingface.co/prithivMLmods/QIE-2511-Object-Remover-v2). Upload an image, draw red bounding boxes over the objects you want to remove, and click Remove Object."
+        "Perform diverse image edits using a specialized [LoRA](https://huggingface.co/prithivMLmods/QIE-2511-Object-Remover-v2). "
+        "Upload an image, draw red bounding boxes over the objects you want to remove, and click Remove Object. "
         "Multiple boxes supported. Select, move, resize or delete individual boxes.",
         elem_id="subtitle",
     )
@@ -832,7 +827,7 @@ with gr.Blocks() as demo:
             gr.HTML(
                 '<div class="bbox-hint">'
                 "<b>Draw mode:</b> Click & drag to create red rectangles. "
-                "<b>Select mode:</b> Click a box to select it → drag to <b>move</b>, "
+                "<b>Select mode:</b> Click a box to select it \u2192 drag to <b>move</b>, "
                 "drag handles to <b>resize</b>. Use <b>Delete Selected</b> to remove one box."
                 "</div>"
             )
@@ -862,7 +857,7 @@ with gr.Blocks() as demo:
                 """
             )
 
-            gr.HTML('<div id="bbox-debug-count">⬜ No boxes drawn yet</div>')
+            gr.HTML('<div id="bbox-debug-count">\u2B1C No boxes drawn yet</div>')
 
             boxes_json = gr.Textbox(
                 value="[]",
@@ -879,7 +874,7 @@ with gr.Blocks() as demo:
                 info="Edit the prompt if needed",
             )
 
-            run_btn = gr.Button("🗑️ Remove Object", variant="primary", size="lg")
+            run_btn = gr.Button("\U0001F5D1\uFE0F Remove Object", variant="primary", size="lg")
 
         with gr.Column(scale=1):
             result = gr.Image(label="Output Image", height=449)
@@ -901,7 +896,7 @@ with gr.Blocks() as demo:
                     "[prithivMLmods](https://huggingface.co/prithivMLmods). "
                     "Adapter: [QIE-2511-Object-Remover-v2]"
                     "(https://huggingface.co/prithivMLmods/QIE-2511-Object-Remover-v2). "
-                    "More adapters → [Qwen-Image-Edit-LoRAs]"
+                    "More adapters \u2192 [Qwen-Image-Edit-LoRAs]"
                     "(https://huggingface.co/models?other=base_model:adapter:Qwen/Qwen-Image-Edit-2509)."
                 )