Upload streamlit_app.py

streamlit_app.py

@@ -1,8 +1,6 @@
 # streamlit_app.py — LamboVision · AI Car Studio (Automotive, 3-step flow, no padded)
-# Step 1: Gallery (clean packshot, native)
-# Step 2: AI Composition (config prompt, native)
-# Step 3: AI Generated Video (from composition native)
-# Uniform tiles by CSS aspect; images/videos fit via object-fit: contain. No padding images are generated.
+# 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).
 
 import os, io, base64, zipfile, requests, streamlit as st
 from PIL import Image
@@ -25,7 +23,7 @@ HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 VIDEO_MAX_PX_DEFAULT = int(os.getenv("VIDEO_MAX_PX", "720"))
 VIDEO_RES_OPTIONS = ["512P", "768P"]
 VIDEO_DUR_OPTIONS = ["6", "10"]
-ASPECT_OPTIONS = ["3:4", "9:16", "1:1", "16:9"]  # visual frame only
+ASPECT_OPTIONS = ["3:4", "9:16", "1:1", "16:9"]  # default 3:4
 
 # ================= Session =================
 if "results" not in st.session_state:
@@ -35,26 +33,21 @@ if "job_counter" not in st.session_state:
 if "video_ui_max_px" not in st.session_state:
     st.session_state["video_ui_max_px"] = VIDEO_MAX_PX_DEFAULT
 
-# ================= HTTP helpers =================
+# ================= Helpers =================
 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
 
@@ -64,14 +57,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", aspect="3:4", prompt_text="", prompt_optimizer=False):
+def post_video(image_url: str, duration="6", resolution="768P", preview_aspect="3:4", prompt_text=""):
     payload = {
         "image_url": image_url,
         "duration": duration,
         "resolution": resolution,
-        "aspect": aspect,
+        "preview_aspect": preview_aspect,
         "prompt": prompt_text or "",
-        "prompt_optimizer": "true" if prompt_optimizer else "false",
+        "prompt_optimizer": "false",
     }
     r = requests.post(f"{API_BASE}/v1/video/from-image", data=payload, headers=HEADERS, timeout=600)
     r.raise_for_status()
@@ -97,8 +90,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)
+    if not mime: mime = _sniff_mime_from_bytes(b)
+    import base64
     enc = base64.b64encode(b).decode("ascii")
     return f"data:{mime};base64,{enc}"
 
@@ -115,8 +108,7 @@ 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:
@@ -126,13 +118,6 @@ def make_zip(named_bytes: list[tuple[str, bytes]]) -> bytes:
             if b: z.writestr(fname, b)
     buf.seek(0); return buf.read()
 
-# ================= UI CSS =================
-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>
@@ -154,6 +139,7 @@ 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};
@@ -162,15 +148,24 @@ def inject_base_css(card_fit_mode="contain"):
     </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)
+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)
+
+# ================= 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
 
-# ================= Prompts (builder) =================
+demo_files = load_local_demo()
+
+# ================= 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. "
@@ -187,8 +182,7 @@ 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 = {
@@ -202,6 +196,26 @@ 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):
+    return {
+        "category": "auto",
+        "prompt": "",  # server will default to AUTO_CLEAN_PROMPT
+        "num_images": "1",
+        "preview_aspect": tile_aspect,  # only hint; backend does not pad
+        "upscale": "false",
+        "upscale_factor": "2",
+    }
+
+def make_composition_payload(edit_prompt, tile_aspect, upscale, upscale_factor):
+    return {
+        "category": "auto",
+        "prompt": edit_prompt,
+        "num_images": "1",
+        "preview_aspect": tile_aspect,  # only hint; backend does not pad
+        "upscale": "true" if upscale else "false",
+        "upscale_factor": upscale_factor,
+    }
+
 # ================= Page =================
 st.set_page_config(page_title="LamboVision · AI Car Studio", layout="wide", page_icon="🏎️")
 inject_base_css(card_fit_mode="contain")
@@ -212,13 +226,13 @@ st.caption("Step 1: Gallery (clean packshot) · Step 2: AI Composition (config)
 ok = backend_ok()
 
 # -------- Top control bar --------
-col1, col2, col3, col4, col5, col6, col7 = st.columns([1.6, 1.6, 1.1, 1.1, 1.0, 1.1, 1.1])
+col1, col2, col3, col4, col5, col6, col7 = st.columns([1.6, 1.6, 1.0, 1.2, 0.9, 1.1, 0.9])
 with col1:
     file_list = st.file_uploader("Car Image (Upload)", type=["jpg","jpeg","png","webp"], accept_multiple_files=True, key="file_upl")
 with col2:
     image_url = st.text_input("Or URL", key="image_url", placeholder="https://... (optional)")
 with col3:
-    preview_aspect = st.selectbox("Tile Aspect", ASPECT_OPTIONS, index=0, key="preview_aspect")  # only frame
+    tile_aspect = st.selectbox("Tile Aspect", ASPECT_OPTIONS, index=0, key="tile_aspect")  # default 3:4
 with col4:
     to_video = st.checkbox("Generate Video", False, key="to_video")
 with col5:
@@ -228,8 +242,7 @@ with col6:
 with col7:
     run = st.button("Run 🏁", type="primary", use_container_width=True, key="run_btn")
 
-inject_preview_css(preview_aspect)
-st.caption("Tiles share the same aspect ratio (CSS). Images and videos fit; no overflow. No padded images are generated.")
+set_tile_aspect_css(tile_aspect)
 
 # -------- Configurator (dropdowns + custom) --------
 st.markdown("### Configurator")
@@ -273,15 +286,17 @@ with c5:
 with c6:
     extra_prompt = st.text_input("Extra (optional)", value="", placeholder="stripe delete, studio floor reflection, …", key="cfg_extra")
 
-# -------- Sample Previews (no padded) --------
-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
-demo_files = load_local_demo()
+# -------- Scene --------
+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")
+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 =================
 st.subheader("Sample Previews")
 col_in, col_gallery, col_comp, col_vid = st.columns(4)
 
@@ -294,32 +309,41 @@ elif image_url:
 elif demo_files.get("input.jpg"):
     input_bytes = demo_files["input.jpg"]
 
-def show_card(col, title, display_bytes=None, download_bytes=None, url=None, fname="image"):
+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 = "data:" + (_sniff_mime_from_bytes(display_bytes)) + ";base64," + base64.b64encode(display_bytes).decode("ascii")
-        col.markdown(f'<div class="lb-frame"><img src="{src}"/></div>', unsafe_allow_html=True)
-    elif url:
-        abs_url = _abs_backend_url(url)
-        col.markdown(f'<div class="lb-frame"><img src="{abs_url}"/></div>', unsafe_allow_html=True)
+        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:
-        col.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True)
-        col.caption("—"); return
-    native = download_bytes or (fetch_bytes(url) if url else display_bytes)
+        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)
+        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 in [".jpg",".jpeg"] else ("image/png" if ext==".png" else "image/webp")
-        col.download_button("Download (Full Quality)", data=native, file_name=f"{fname}{ext if ext!='.bin' else '.jpg'}", mime=mime)
+        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")
+
+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_card(col_in, "Input", display_bytes=input_bytes or demo_files.get("input.jpg"), download_bytes=input_bytes or demo_files.get("input.jpg"), fname="input")
-show_card(col_gallery, "Gallery", display_bytes=demo_files.get("packshot.jpg") or input_bytes, download_bytes=demo_files.get("packshot.jpg") or input_bytes, fname="gallery")
-show_card(col_comp, "AI Composition", display_bytes=demo_files.get("packshot.jpg") or input_bytes, download_bytes=demo_files.get("packshot.jpg") or input_bytes, fname="ai_composition")
+show_tile(col_comp, "AI Composition", display_bytes=gallery_bytes, download_bytes=gallery_bytes,
+          fname="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)
-if demo_files.get("tryon.mp4"):
-    data_url = "data:video/mp4;base64," + base64.b64encode(demo_files["tryon.mp4"]).decode("ascii")
+if vid_demo:
+    data_url = "data:video/mp4;base64," + base64.b64encode(vid_demo).decode("ascii")
     col_vid.markdown(
         f"""
         <div class="lb-frame">
@@ -329,19 +353,11 @@ if demo_files.get("tryon.mp4"):
         </div>
         """, unsafe_allow_html=True
     )
-    col_vid.download_button("Download Video (MP4)", data=demo_files["tryon.mp4"], file_name="tryon.mp4", mime="video/mp4")
+    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("—")
-
-# ================= RUN (3-step chain, native only) =================
-def make_gallery_payload():
-    return {"category": "auto", "prompt": "", "num_images": "1", "upscale": "false", "upscale_factor": "2"}
-
-def make_composition_payload(edit_prompt, upscale=False, upscale_factor="2"):
-    return {"category": "auto", "prompt": edit_prompt, "num_images": "1",
-            "upscale": "true" if upscale else "false", "upscale_factor": upscale_factor}
+    col_vid.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True); col_vid.caption("—")
 
+# ================= RUN (3-step chain, no padded) =================
 if run:
     if not ok:
         st.error("Backend not reachable. Check API_BASE / TOKEN.")
@@ -354,51 +370,50 @@ if run:
                 for f in file_list:
                     files_payload.append(("files", (f.name, f.getvalue(), f.type or "image/jpeg")))
 
-            # Step 1: Gallery (clean)
-            g_payload = make_gallery_payload()
-            if image_url: g_payload["image_urls"] = image_url
-            out_gallery = post_image_edit(g_payload, files_payload if files_payload else None)
+            # ---- Step 1: Gallery (clean packshot) ----
+            gallery_payload = make_gallery_payload(tile_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)
+            # ---- Step 2: AI Composition (config prompt) ----
             edit_prompt = build_car_prompt(
-                model_name=model_name or "Lamborghini",
-                body_color=body_color or "Verde Ithaca",
+                model_name=model_name if model_name else "Lamborghini",
+                body_color=body_color if body_color else "Verde Ithaca",
                 finish=finish,
-                wheel_style=wheel_style or "lightweight forged",
-                caliper_color=caliper_color or "yellow",
-                extra=extra_prompt or ""
+                wheel_style=wheel_style if wheel_style else "lightweight forged",
+                caliper_color=caliper_color if caliper_color else "yellow",
+                extra=extra_prompt
             )
-            c_payload = make_composition_payload(edit_prompt, upscale=False, upscale_factor="2")
-            c_payload["image_urls"] = gallery_native_url or ""  # pass /outputs/* back; backend re-uploads to fal
-            out_comp = post_image_edit(c_payload, files_payload=None)
+            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
+            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
 
-            # Step 3: Video (optional)
+            # ---- Step 3: Video (optional) ----
             vurl = None; ui_hints = {}
             if st.session_state.get("to_video", False) and (composition_native_url or gallery_native_url):
-                video_prompt = build_video_prompt(st.session_state.get("scene_name","Studio Cinematic"),
-                                                  st.session_state.get("scene_extra",""))
+                video_prompt = build_video_prompt(scene_name, scene_extra)
                 v = post_video(
                     composition_native_url or gallery_native_url,
-                    duration=st.session_state.get("duration_sel","6"),
-                    resolution=st.session_state.get("resolution_sel","768P"),
-                    aspect=st.session_state.get("preview_aspect","3:4"),
+                    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"),
                     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))
+            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 ----
             st.session_state["job_counter"] += 1
             st.session_state["results"] = {
                 "gallery_native_url": gallery_native_url,
                 "composition_native_url": composition_native_url,
                 "video_url": vurl,
                 "api_base": API_BASE,
-                "preview_aspect": st.session_state.get("preview_aspect","3:4"),
+                "tile_aspect": tile_aspect,
             }
 
         except requests.HTTPError as e:
@@ -411,29 +426,43 @@ res = st.session_state.get("results")
 vbytes = None
 if res:
     st.subheader("Results")
-    c_gallery, c_comp, c_vid = st.columns(3)
-
-    # Gallery (native only)
-    def show_card_url(col, title, url, fname):
-        col.markdown(f'<div class="lb-card-title">{title}</div>', unsafe_allow_html=True)
-        if url:
-            abs_url = _abs_backend_url(url)
-            col.markdown(f'<div class="lb-frame"><img src="{abs_url}"/></div>', unsafe_allow_html=True)
-            nb = fetch_bytes(abs_url)
-            if nb:
-                sz = image_size_from_bytes(nb)
-                if sz: col.caption(f"Real Res: {sz[0]}×{sz[1]} px")
-                ext = infer_ext(nb)
-                mime = "image/jpeg" if ext in [".jpg",".jpeg"] else ("image/png" if ext==".png" else "image/webp")
-                col.download_button("Download (Full Quality)", data=nb, file_name=f"{fname}{ext if ext!='.bin' else '.jpg'}", mime=mime)
-        else:
-            col.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True)
-            col.caption("—")
-
-    show_card_url(c_gallery, "Gallery", res.get("gallery_native_url"), "gallery_native")
-    show_card_url(c_comp, "AI Composition", res.get("composition_native_url"), "ai_composition_native")
-
-    # Video
+    c_gallery, c_comp, c_spare, c_vid = st.columns(4)
+
+    # 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("—")
+
+    # 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 Generated Video
     c_vid.markdown('<div class="lb-card-title">AI Generated Video</div>', unsafe_allow_html=True)
     vurl = res.get("video_url")
     if vurl:
@@ -453,8 +482,7 @@ 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 = []
@@ -469,18 +497,11 @@ if res:
                            file_name="lambovision_outputs.zip", mime="application/zip",
                            key=f"job{st.session_state['job_counter']}_zip_dl")
 
-# ================= Scene controls (moved under results to avoid clutter) =================
-st.markdown("### Scene")
-s1, s2 = st.columns([1.2, 1.8])
-with s1:
-    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")
-
 # ================= Sidebar =================
 with st.sidebar:
     st.caption(f"API_BASE: {API_BASE}")
     if HF_TOKEN: st.caption("Auth: Bearer (active)")
-    st.slider("Video preview width (px)", 360, 1080, st.session_state["video_ui_max_px"], step=10, key="video_ui_max_px")
+    st.slider("Video preview width (px)", 360, 1080, st.session_state["video_ui_max_px"],
+              step=10, key="video_ui_max_px")
     if st.button("Clear cache"): st.cache_data.clear(); st.success("Cache cleared.")
     if st.button("Clear results"): st.session_state["results"] = None; st.success("Cleared.")