Upload streamlit_app.py

streamlit_app.py

@@ -1,6 +1,6 @@
-# streamlit_app.py — LamboVision · AI Car Studio (Automotive, 3-step flow, no padded)
-# Step 1: Gallery (clean packshot)  Step 2: AI Composition (config)  Step 3: AI Generated Video
-# Uniform tiles via CSS aspect-ratio; images/videos displayed native (object-fit: contain).
+# streamlit_app.py — LamboVision · AI Car Studio (native-only UI, no padding)
+# Flow: Step 1 Gallery (clean packshot) → Step 2 AI Composition (config) → Step 3 AI Generated Video
+# All tiles are uniform aspect (configurable). Images/videos fit inside (no overflow).
 
 import os, io, base64, zipfile, requests, streamlit as st
 from PIL import Image
@@ -38,16 +38,21 @@ def _needs_auth(url: str) -> bool:
     return url.startswith(API_BASE) or url.startswith("outputs/") or url.startswith("/outputs/")
 
 def _abs_backend_url(url_or_path: str) -> str:
-    if not url_or_path: return ""
-    if url_or_path.startswith(("http://","https://")): return url_or_path
-    if url_or_path.startswith("/outputs/"): return f"{API_BASE}{url_or_path}"
-    if url_or_path.startswith("outputs/"):  return f"{API_BASE}/{url_or_path}"
+    if not url_or_path:
+        return ""
+    if url_or_path.startswith(("http://","https://")):
+        return url_or_path
+    if url_or_path.startswith("/outputs/"):
+        return f"{API_BASE}{url_or_path}"
+    if url_or_path.startswith("outputs/"):
+        return f"{API_BASE}/{url_or_path}"
     return f"{API_BASE}/{url_or_path.lstrip('/')}"
 
 def backend_ok() -> bool:
     try:
         r = requests.get(f"{API_BASE}/health", headers=HEADERS, timeout=10)
-        r.raise_for_status(); return True
+        r.raise_for_status()
+        return True
     except Exception:
         return False
 
@@ -57,14 +62,14 @@ def post_image_edit(data: dict, files_payload):
     r.raise_for_status()
     return r.json()
 
-def post_video(image_url: str, duration="6", resolution="768P", preview_aspect="3:4", prompt_text=""):
+def post_video(image_url: str, duration="6", resolution="768P", preview_aspect="3:4", prompt_text="", prompt_optimizer=False):
     payload = {
         "image_url": image_url,
         "duration": duration,
         "resolution": resolution,
         "preview_aspect": preview_aspect,
         "prompt": prompt_text or "",
-        "prompt_optimizer": "false",
+        "prompt_optimizer": "true" if prompt_optimizer else "false",
     }
     r = requests.post(f"{API_BASE}/v1/video/from-image", data=payload, headers=HEADERS, timeout=600)
     r.raise_for_status()
@@ -90,8 +95,8 @@ def _sniff_mime_from_bytes(b: bytes) -> str:
         return "image/jpeg"
 
 def image_bytes_to_data_url(b: bytes, mime: str | None = None) -> str:
-    if not mime: mime = _sniff_mime_from_bytes(b)
-    import base64
+    if not mime:
+        mime = _sniff_mime_from_bytes(b)
     enc = base64.b64encode(b).decode("ascii")
     return f"data:{mime};base64,{enc}"
 
@@ -108,7 +113,8 @@ def infer_ext(data: bytes) -> str:
         fmt = (im.format or "JPEG").lower()
         return "." + {"jpeg":"jpg","jpg":"jpg","png":"png","webp":"webp"}.get(fmt,"jpg")
     except Exception:
-        if data[:4] == b"\x00\x00\x00\x18" or data[4:8] == b"ftyp": return ".mp4"
+        if data[:4] == b"\x00\x00\x00\x18" or data[4:8] == b"ftyp":
+            return ".mp4"
         return ".bin"
 
 def make_zip(named_bytes: list[tuple[str, bytes]]) -> bytes:
@@ -118,6 +124,12 @@ def make_zip(named_bytes: list[tuple[str, bytes]]) -> bytes:
             if b: z.writestr(fname, b)
     buf.seek(0); return buf.read()
 
+def _parse_aspect_str(s: str):
+    try:
+        a, b = s.split(":"); return max(1, int(a)), max(1, int(b))
+    except Exception:
+        return (3, 4)
+
 def inject_base_css(card_fit_mode="contain"):
     st.markdown(f"""
     <style>
@@ -139,7 +151,6 @@ def inject_base_css(card_fit_mode="contain"):
         background: #111419;
         display:flex; align-items:center; justify-content:center;
         box-shadow: 0 0 0 1px rgba(255,122,26,0.15) inset;
-        aspect-ratio: var(--tile-ar, 3/4);
       }}
       .lb-frame img, .lb-frame video {{
         width:100%; height:100%; display:block; object-fit:{card_fit_mode};
@@ -148,24 +159,75 @@ def inject_base_css(card_fit_mode="contain"):
     </style>
     """, unsafe_allow_html=True)
 
-def set_tile_aspect_css(ar: str):
-    try:
-        a, b = [int(x) for x in ar.split(":")]
-        st.markdown(f"<style>:root {{ --tile-ar: {a}/{b}; }}</style>", unsafe_allow_html=True)
-    except Exception:
-        st.markdown("<style>:root { --tile-ar: 3/4; }</style>", unsafe_allow_html=True)
+def inject_preview_css(aspect: str):
+    aw, ah = _parse_aspect_str(aspect)
+    st.markdown(f"""
+    <style>
+      .lb-frame {{ aspect-ratio: {aw} / {ah}; }}
+    </style>
+    """, unsafe_allow_html=True)
 
-# ================= Local demo assets (optional) =================
-def load_local_demo():
-    demo = {}
-    for fname in ["input.jpg", "packshot.jpg", "tryon.mp4"]:
-        if os.path.exists(fname):
-            with open(fname, "rb") as f: demo[fname] = f.read()
-    return demo
+def show_tile_card_display_download(
+    col, title,
+    display_url: str | None = None,
+    display_bytes: bytes | None = None,
+    download_url: str | None = None,
+    download_bytes: bytes | None = None,
+    filename_stub: str = "image",
+    key_prefix: str = "",
+):
+    col.markdown(f'<div class="lb-card-title">{title}</div>', unsafe_allow_html=True)
+    if not display_url and not display_bytes:
+        col.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True)
+        col.caption("—"); return None, None
+    if display_bytes is not None:
+        src = image_bytes_to_data_url(display_bytes)
+        col.markdown(f'<div class="lb-frame"><img src="{src}"/></div>', unsafe_allow_html=True)
+    else:
+        abs_url = _abs_backend_url(display_url)
+        col.markdown(f'<div class="lb-frame"><img src="{abs_url}"/></div>', unsafe_allow_html=True)
 
-demo_files = load_local_demo()
+    native = download_bytes or (fetch_bytes(download_url) if download_url else None)
+    if not native:
+        native = display_bytes
+    if native:
+        sz = image_size_from_bytes(native)
+        if sz: col.caption(f"Real Res: {sz[0]}×{sz[1]} px")
+        ext = infer_ext(native)
+        mime = "image/jpeg" if ext.lower() in [".jpg",".jpeg"] else ("image/png" if ext.lower()==".png" else "image/webp")
+        col.download_button("Download (Full Quality)", data=native,
+                            file_name=f"{filename_stub}{ext if ext!='.bin' else '.jpg'}",
+                            mime=mime, key=f"{key_prefix}_{filename_stub}_dl")
+    else:
+        col.caption("—")
+    return display_url or "data", native
+
+def rgba_to_rgb_on_bg(img_rgba: Image.Image, bg_rgb=(255,255,255)) -> Image.Image:
+    if img_rgba.mode != "RGBA":
+        return img_rgba.convert("RGB")
+    bg = Image.new("RGBA", img_rgba.size, (*bg_rgb, 255))
+    bg.paste(img_rgba, (0,0), img_rgba)
+    return bg.convert("RGB")
+
+@st.cache_data(ttl=300, show_spinner=False)
+def make_small_jpeg_from_native(native: bytes, target_long_edge: int = 1280, bg_rgb=(255,255,255)) -> bytes:
+    im = Image.open(io.BytesIO(native))
+    if im.mode != "RGBA" and "transparency" not in im.info:
+        im = im.convert("RGB")
+    else:
+        im = im.convert("RGBA")
+    w, h = im.size
+    if w >= h:
+        new_w = min(target_long_edge, w)
+        new_h = max(1, int(h * (new_w / max(1, w))))
+    else:
+        new_h = min(target_long_edge, h)
+        new_w = max(1, int(w * (new_h / max(1, h))))
+    im2 = im.resize((new_w, new_h), Image.LANCZOS)
+    im2_rgb = rgba_to_rgb_on_bg(im2, bg_rgb=bg_rgb)
+    out = io.BytesIO(); im2_rgb.save(out, format="JPEG", quality=88, subsampling=1)
+    return out.getvalue()
 
-# ================= Prompt builders =================
 def build_car_prompt(model_name: str, body_color: str, finish: str, wheel_style: str, caliper_color: str, extra: str = "") -> str:
     base = (
         "Catalog-ready studio photo of a Lamborghini from a three-quarter front view. "
@@ -182,7 +244,8 @@ def build_car_prompt(model_name: str, body_color: str, finish: str, wheel_style:
         f"Wheels: {wheel_style}; brake calipers {caliper_color}. "
         "Maintain original stance; no suspension changes; no aero additions."
     )
-    if extra.strip(): cfg += " " + extra.strip()
+    if extra.strip():
+        cfg += " " + extra.strip()
     return base + " " + cfg
 
 SCENE_PRESETS = {
@@ -196,26 +259,40 @@ def build_video_prompt(scene_name: str, extra: str = "") -> str:
     if extra.strip(): return base + " " + lock + " " + extra.strip()
     return base + " " + lock
 
-def make_gallery_payload(tile_aspect):
+def make_gallery_payload(preview_aspect):
     return {
         "category": "auto",
-        "prompt": "",  # server will default to AUTO_CLEAN_PROMPT
+        "prompt": "",             # server defaults to AUTO_CLEAN_PROMPT
         "num_images": "1",
-        "preview_aspect": tile_aspect,  # only hint; backend does not pad
+        "preview_aspect": preview_aspect,
+        "preview_long_edge": "1536",
+        "preview_ratio": "0.9",
         "upscale": "false",
         "upscale_factor": "2",
     }
 
-def make_composition_payload(edit_prompt, tile_aspect, upscale, upscale_factor):
+def make_composition_payload(edit_prompt, preview_aspect, upscale, upscale_factor):
     return {
         "category": "auto",
         "prompt": edit_prompt,
         "num_images": "1",
-        "preview_aspect": tile_aspect,  # only hint; backend does not pad
+        "preview_aspect": preview_aspect,
+        "preview_long_edge": "1536",
+        "preview_ratio": "0.9",
         "upscale": "true" if upscale else "false",
         "upscale_factor": upscale_factor,
     }
 
+# ================= Local demo assets (optional) =================
+def load_local_demo():
+    demo = {}
+    for fname in ["input.jpg", "packshot.jpg", "model.jpg", "tryon.mp4"]:
+        if os.path.exists(fname):
+            with open(fname, "rb") as f: demo[fname] = f.read()
+    return demo
+
+demo_files = load_local_demo()
+
 # ================= Page =================
 st.set_page_config(page_title="LamboVision · AI Car Studio", layout="wide", page_icon="🏎️")
 inject_base_css(card_fit_mode="contain")
@@ -232,7 +309,7 @@ with col1:
 with col2:
     image_url = st.text_input("Or URL", key="image_url", placeholder="https://... (optional)")
 with col3:
-    tile_aspect = st.selectbox("Tile Aspect", ASPECT_OPTIONS, index=0, key="tile_aspect")  # default 3:4
+    preview_aspect = st.selectbox("Tile Aspect", ASPECT_OPTIONS, index=0, key="preview_aspect")  # default 3:4
 with col4:
     to_video = st.checkbox("Generate Video", False, key="to_video")
 with col5:
@@ -242,7 +319,8 @@ with col6:
 with col7:
     run = st.button("Run 🏁", type="primary", use_container_width=True, key="run_btn")
 
-set_tile_aspect_css(tile_aspect)
+inject_preview_css(preview_aspect)
+st.caption("All tiles use the same aspect. Images and videos fit inside; no overflow.")
 
 # -------- Configurator (dropdowns + custom) --------
 st.markdown("### Configurator")
@@ -286,17 +364,15 @@ with c5:
 with c6:
     extra_prompt = st.text_input("Extra (optional)", value="", placeholder="stripe delete, studio floor reflection, …", key="cfg_extra")
 
-# -------- Scene --------
+# -------- Scene (for video) --------
 st.markdown("### Scene")
 s1, s2 = st.columns([1.2, 1.8])
 with s1:
-    scene_name = st.selectbox("Scene", list(SCENE_PRESETS.keys()), index=0, key="scene_name")
+    scene_name = st.selectbox("Scene preset", list(SCENE_PRESETS.keys()), index=0, key="scene_name")
 with s2:
     scene_extra = st.text_input("Scene extra (optional)", value="", key="scene_extra")
 
-st.caption("Tiles use a fixed aspect in UI; images and videos are displayed native with contain.")
-
-# ================= SAMPLE PREVIEWS =================
+# ================= SAMPLE PREVIEWS (no padding; just native demos) =================
 st.subheader("Sample Previews")
 col_in, col_gallery, col_comp, col_vid = st.columns(4)
 
@@ -309,36 +385,28 @@ elif image_url:
 elif demo_files.get("input.jpg"):
     input_bytes = demo_files["input.jpg"]
 
-def show_tile(col, title, display_bytes=None, download_bytes=None, url=None, dl_url=None, fname="image", key_prefix="pre"):
-    col.markdown(f'<div class="lb-card-title">{title}</div>', unsafe_allow_html=True)
-    if display_bytes is None and not url:
-        col.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True); col.caption("—"); return
-    if display_bytes is not None:
-        src = image_bytes_to_data_url(display_bytes); col.markdown(f'<div class="lb-frame"><img src="{src}"/></div>', unsafe_allow_html=True)
-        native = download_bytes or display_bytes
-    else:
-        abs_url = _abs_backend_url(url); col.markdown(f'<div class="lb-frame"><img src="{abs_url}"/></div>', unsafe_allow_html=True)
-        native = fetch_bytes(dl_url or abs_url)
-    if native:
-        sz = image_size_from_bytes(native); 
-        if sz: col.caption(f"Real Res: {sz[0]}×{sz[1]} px")
-        ext = ".mp4" if native[:4] == b"\x00\x00\x00\x18" or native[4:8] == b"ftyp" else ".jpg"
-        mime = "video/mp4" if ext == ".mp4" else "image/jpeg"
-        col.download_button("Download (Full Quality)", data=native, file_name=f"{fname}{ext if ext!='.bin' else '.jpg'}",
-                            mime=mime, key=f"{key_prefix}_{fname}_dl")
-    else:
-        col.caption("—")
-
-show_tile(col_in, "Input", display_bytes=input_bytes if input_bytes else demo_files.get("input.jpg"),
-          download_bytes=input_bytes if input_bytes else demo_files.get("input.jpg"),
-          fname="input", key_prefix="pre")
+show_tile_card_display_download(
+    col_in, "Input",
+    display_bytes=input_bytes if input_bytes else demo_files.get("input.jpg"),
+    download_bytes=input_bytes if input_bytes else demo_files.get("input.jpg"),
+    filename_stub="input", key_prefix="pre"
+)
 
 gallery_bytes = demo_files.get("packshot.jpg") or input_bytes
-show_tile(col_gallery, "Gallery", display_bytes=gallery_bytes, download_bytes=gallery_bytes,
-          fname="gallery", key_prefix="pre")
-
-show_tile(col_comp, "AI Composition", display_bytes=gallery_bytes, download_bytes=gallery_bytes,
-          fname="ai_composition", key_prefix="pre")
+show_tile_card_display_download(
+    col_gallery, "Gallery",
+    display_bytes=gallery_bytes,
+    download_bytes=gallery_bytes,
+    filename_stub="gallery", key_prefix="pre"
+)
+
+comp_bytes = gallery_bytes
+show_tile_card_display_download(
+    col_comp, "AI Composition",
+    display_bytes=comp_bytes,
+    download_bytes=comp_bytes,
+    filename_stub="ai_composition", key_prefix="pre"
+)
 
 vid_demo = demo_files.get("tryon.mp4")
 col_vid.markdown('<div class="lb-card-title">AI Generated Video</div>', unsafe_allow_html=True)
@@ -355,9 +423,10 @@ if vid_demo:
     )
     col_vid.download_button("Download Video (MP4)", data=vid_demo, file_name="tryon.mp4", mime="video/mp4", key="pre_video_dl")
 else:
-    col_vid.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True); col_vid.caption("—")
+    col_vid.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True)
+    col_vid.caption("—")
 
-# ================= RUN (3-step chain, no padded) =================
+# ================= RUN (3-step chain) =================
 if run:
     if not ok:
         st.error("Backend not reachable. Check API_BASE / TOKEN.")
@@ -370,50 +439,55 @@ if run:
                 for f in file_list:
                     files_payload.append(("files", (f.name, f.getvalue(), f.type or "image/jpeg")))
 
-            # ---- Step 1: Gallery (clean packshot) ----
-            gallery_payload = make_gallery_payload(tile_aspect)
-            if image_url: gallery_payload["image_urls"] = image_url
+            # ---- Step 1: Gallery (clean packshot; native only) ----
+            gallery_payload = make_gallery_payload(preview_aspect)
+            if image_url:
+                gallery_payload["image_urls"] = image_url
             out_gallery = post_image_edit(gallery_payload, files_payload if files_payload else None)
+
             gallery_native_url = (out_gallery["result"]["images_native"] or [{}])[0].get("url") if out_gallery else None
 
-            # ---- Step 2: AI Composition (config prompt) ----
+            # ---- Step 2: AI Composition (config prompt; native only) ----
             edit_prompt = build_car_prompt(
-                model_name=model_name if model_name else "Lamborghini",
-                body_color=body_color if body_color else "Verde Ithaca",
+                model_name=(model_name or "Lamborghini"),
+                body_color=(body_color or "Verde Ithaca"),
                 finish=finish,
-                wheel_style=wheel_style if wheel_style else "lightweight forged",
-                caliper_color=caliper_color if caliper_color else "yellow",
+                wheel_style=(wheel_style or "lightweight forged"),
+                caliper_color=(caliper_color or "yellow"),
                 extra=extra_prompt
             )
-            comp_payload = make_composition_payload(edit_prompt, tile_aspect, upscale=False, upscale_factor="2")
-            comp_payload["image_urls"] = gallery_native_url or ""  # backend re-uploads /outputs/* to vendor
+            comp_payload = make_composition_payload(edit_prompt, preview_aspect, upscale=False, upscale_factor="2")
+            comp_payload["image_urls"] = gallery_native_url or ""
             out_comp = post_image_edit(comp_payload, files_payload=None)
-            composition_native_url = (out_comp["result"]["images_native"] or [{}])[0].get("url") if out_comp else None
+
+            comp_native_url = (out_comp["result"]["images_native"] or [{}])[0].get("url") if out_comp else None
 
             # ---- Step 3: Video (optional) ----
             vurl = None; ui_hints = {}
-            if st.session_state.get("to_video", False) and (composition_native_url or gallery_native_url):
+            if st.session_state.get("to_video", False) and (comp_native_url or gallery_native_url):
                 video_prompt = build_video_prompt(scene_name, scene_extra)
                 v = post_video(
-                    composition_native_url or gallery_native_url,
+                    comp_native_url or gallery_native_url,
                     duration=st.session_state.get("duration_sel", VIDEO_DUR_OPTIONS[0]),
                     resolution=st.session_state.get("resolution_sel", VIDEO_RES_OPTIONS[1]),
-                    preview_aspect=st.session_state.get("tile_aspect","3:4"),
+                    preview_aspect=st.session_state.get("preview_aspect","3:4"),
                     prompt_text=video_prompt,
+                    prompt_optimizer=False,
                 )
                 vurl = (v.get("result") or {}).get("video", {}).get("url")
                 ui_hints = v.get("ui_hints") or {}
+
             if isinstance(ui_hints, dict):
                 st.session_state["video_ui_max_px"] = int(ui_hints.get("suggested_video_max_px", VIDEO_MAX_PX_DEFAULT))
 
-            # ---- Save session ----
+            # ---- Save results to session ----
             st.session_state["job_counter"] += 1
             st.session_state["results"] = {
                 "gallery_native_url": gallery_native_url,
-                "composition_native_url": composition_native_url,
+                "composition_native_url": comp_native_url,
                 "video_url": vurl,
                 "api_base": API_BASE,
-                "tile_aspect": tile_aspect,
+                "preview_aspect": preview_aspect,
             }
 
         except requests.HTTPError as e:
@@ -426,41 +500,25 @@ res = st.session_state.get("results")
 vbytes = None
 if res:
     st.subheader("Results")
-    c_gallery, c_comp, c_spare, c_vid = st.columns(4)
+    c_gallery, c_comp, c_vid, c_zip = st.columns([1.2, 1.2, 1.2, 0.6])
 
     # Gallery (clean packshot, native)
-    c_gallery.markdown('<div class="lb-card-title">Gallery</div>', unsafe_allow_html=True)
-    g_url = res.get("gallery_native_url")
-    if g_url:
-        abs_g = _abs_backend_url(g_url)
-        c_gallery.markdown(f'<div class="lb-frame"><img src="{abs_g}"/></div>', unsafe_allow_html=True)
-        gb = fetch_bytes(abs_g)
-        if gb:
-            sz = image_size_from_bytes(gb)
-            if sz: c_gallery.caption(f"Real Res: {sz[0]}×{sz[1]} px")
-            c_gallery.download_button("Download (Full Quality)", data=gb, file_name=f"gallery_native{infer_ext(gb)}",
-                                      mime="image/jpeg", key=f"job{st.session_state['job_counter']}_g_dl")
-    else:
-        c_gallery.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True); c_gallery.caption("—")
-
-    # AI Composition (native)
-    c_comp.markdown('<div class="lb-card-title">AI Composition</div>', unsafe_allow_html=True)
-    c_url = res.get("composition_native_url")
-    if c_url:
-        abs_c = _abs_backend_url(c_url)
-        c_comp.markdown(f'<div class="lb-frame"><img src="{abs_c}"/></div>', unsafe_allow_html=True)
-        cb = fetch_bytes(abs_c)
-        if cb:
-            sz = image_size_from_bytes(cb)
-            if sz: c_comp.caption(f"Real Res: {sz[0]}×{sz[1]} px")
-            c_comp.download_button("Download (Full Quality)", data=cb, file_name=f"ai_composition_native{infer_ext(cb)}",
-                                   mime="image/jpeg", key=f"job{st.session_state['job_counter']}_c_dl")
-    else:
-        c_comp.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True); c_comp.caption("—")
+    show_tile_card_display_download(
+        c_gallery, "Gallery",
+        display_url=res.get("gallery_native_url"),
+        download_url=res.get("gallery_native_url"),
+        filename_stub="gallery_native",
+        key_prefix=f"job{st.session_state['job_counter']}"
+    )
 
-    # Spare placeholder (kept same size for grid balance)
-    c_spare.markdown('<div class="lb-card-title">—</div>', unsafe_allow_html=True)
-    c_spare.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True); c_spare.caption("—")
+    # AI Composition (configured, native)
+    show_tile_card_display_download(
+        c_comp, "AI Composition",
+        display_url=res.get("composition_native_url"),
+        download_url=res.get("composition_native_url"),
+        filename_stub="ai_composition_native",
+        key_prefix=f"job{st.session_state['job_counter']}"
+    )
 
     # AI Generated Video
     c_vid.markdown('<div class="lb-card-title">AI Generated Video</div>', unsafe_allow_html=True)
@@ -482,7 +540,8 @@ if res:
             c_vid.download_button("Download Video (MP4)", data=vb, file_name="lambovision_video.mp4",
                                   mime="video/mp4", key=f"job{st.session_state['job_counter']}_video_dl")
     else:
-        c_vid.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True); c_vid.caption("—")
+        c_vid.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True)
+        c_vid.caption("—")
 
     # ZIP bundle
     named = []
@@ -491,11 +550,15 @@ if res:
     if gnat: named.append((f"gallery_native{infer_ext(gnat)}", gnat))
     if cnat: named.append((f"ai_composition_native{infer_ext(cnat)}", cnat))
     if vbytes: named.append(("video.mp4", vbytes))
-    if named:
-        zip_bytes = make_zip(named)
-        st.download_button("Download All (ZIP)", data=zip_bytes,
-                           file_name="lambovision_outputs.zip", mime="application/zip",
-                           key=f"job{st.session_state['job_counter']}_zip_dl")
+    with c_zip:
+        st.markdown('<div class="lb-card-title">Export</div>', unsafe_allow_html=True)
+        if named:
+            zip_bytes = make_zip(named)
+            st.download_button("Download All (ZIP)", data=zip_bytes,
+                               file_name="lambovision_outputs.zip", mime="application/zip",
+                               key=f"job{st.session_state['job_counter']}_zip_dl")
+        else:
+            st.caption("—")
 
 # ================= Sidebar =================
 with st.sidebar: