Upload streamlit_app.py

streamlit_app.py

@@ -1,6 +1,6 @@
 # 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).
+# Flow: Step 1 Gallery (Background Cleaned) → Step 2 AI Configurated (config) → Step 3 AI Generated Video
+# Tiles use a uniform aspect. Images/videos fit inside (no overflow). “Generate Video From These Results” button included.
 
 import os, io, base64, zipfile, requests, streamlit as st
 from PIL import Image
@@ -16,8 +16,8 @@ except Exception:
     pass
 
 # ================= Env & API =================
-API_BASE = (os.getenv("AI_LamboVision_API", "http://127.0.0.1:7860") or "http://127.0.0.1:7860").strip().strip("'\"").rstrip("/")
-HF_TOKEN = (os.getenv("AI_LamboVision_TOKEN", "") or "").strip().strip("'\"")
+API_BASE = (os.getenv("AI_LIGHTBOX_API", "http://127.0.0.1:7860") or "http://127.0.0.1:7860").strip().strip("'\"").rstrip("/")
+HF_TOKEN = (os.getenv("AI_LIGHTBOX_TOKEN", "") or "").strip().strip("'\"")
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 
 VIDEO_MAX_PX_DEFAULT = int(os.getenv("VIDEO_MAX_PX", "720"))
@@ -298,7 +298,7 @@ st.set_page_config(page_title="LamboVision · AI Car Studio", layout="wide", pag
 inject_base_css(card_fit_mode="contain")
 
 st.title("🏎️ LamboVision · AI Car Studio")
-st.caption("Step 1: Gallery (clean packshot) · Step 2: AI Composition (config) · Step 3: AI Generated Video")
+st.caption("Step 1: Gallery (Background Cleaned) · Step 2: AI Configurated (config) · Step 3: AI Generated Video")
 
 ok = backend_ok()
 
@@ -311,7 +311,7 @@ with col2:
 with col3:
     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")
+    to_video = st.checkbox("Auto-generate Video after Run", False, key="to_video")
 with col5:
     duration = st.selectbox("Video Duration", VIDEO_DUR_OPTIONS, index=0, key="duration_sel")
 with col6:
@@ -372,7 +372,7 @@ with s1:
 with s2:
     scene_extra = st.text_input("Scene extra (optional)", value="", key="scene_extra")
 
-# ================= SAMPLE PREVIEWS (no padding; just native demos) =================
+# ================= SAMPLE PREVIEWS =================
 st.subheader("Sample Previews")
 col_in, col_gallery, col_comp, col_vid = st.columns(4)
 
@@ -394,7 +394,7 @@ show_tile_card_display_download(
 
 gallery_bytes = demo_files.get("packshot.jpg") or input_bytes
 show_tile_card_display_download(
-    col_gallery, "Gallery",
+    col_gallery, "Gallery (Background Cleaned)",
     display_bytes=gallery_bytes,
     download_bytes=gallery_bytes,
     filename_stub="gallery", key_prefix="pre"
@@ -402,10 +402,10 @@ show_tile_card_display_download(
 
 comp_bytes = gallery_bytes
 show_tile_card_display_download(
-    col_comp, "AI Composition",
+    col_comp, "AI Configurated",
     display_bytes=comp_bytes,
     download_bytes=comp_bytes,
-    filename_stub="ai_composition", key_prefix="pre"
+    filename_stub="ai_configurated", key_prefix="pre"
 )
 
 vid_demo = demo_files.get("tryon.mp4")
@@ -439,15 +439,14 @@ 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; native only) ----
+            # ---- Step 1: Gallery (Background Cleaned; 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; native only) ----
+            # ---- Step 2: AI Configurated (config prompt; native only) ----
             edit_prompt = build_car_prompt(
                 model_name=(model_name or "Lamborghini"),
                 body_color=(body_color or "Verde Ithaca"),
@@ -459,10 +458,9 @@ if run:
             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)
-
             comp_native_url = (out_comp["result"]["images_native"] or [{}])[0].get("url") if out_comp else None
 
-            # ---- Step 3: Video (optional) ----
+            # ---- Optional auto video right after Run ----
             vurl = None; ui_hints = {}
             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)
@@ -500,23 +498,23 @@ res = st.session_state.get("results")
 vbytes = None
 if res:
     st.subheader("Results")
-    c_gallery, c_comp, c_vid, c_zip = st.columns([1.2, 1.2, 1.2, 0.6])
+    c_gallery, c_comp, c_vid = st.columns([1.2, 1.2, 1.2])
 
-    # Gallery (clean packshot, native)
+    # Gallery (Background Cleaned, native)
     show_tile_card_display_download(
-        c_gallery, "Gallery",
+        c_gallery, "Gallery (Background Cleaned)",
         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']}"
     )
 
-    # AI Composition (configured, native)
+    # AI Configurated (native)
     show_tile_card_display_download(
-        c_comp, "AI Composition",
+        c_comp, "AI Configurated",
         display_url=res.get("composition_native_url"),
         download_url=res.get("composition_native_url"),
-        filename_stub="ai_composition_native",
+        filename_stub="ai_configurated_native",
         key_prefix=f"job{st.session_state['job_counter']}"
     )
 
@@ -541,17 +539,51 @@ if res:
                                   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("—")
-
-    # ZIP bundle
-    named = []
-    gnat = fetch_bytes(res.get("gallery_native_url")) if res.get("gallery_native_url") else None
-    cnat = fetch_bytes(res.get("composition_native_url")) if res.get("composition_native_url") else None
-    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))
-    with c_zip:
-        st.markdown('<div class="lb-card-title">Export</div>', unsafe_allow_html=True)
+        # ---- Explicit manual video trigger button ----
+        gen_now = c_vid.button("Generate Video From These Results", key=f"gen_video_job{st.session_state['job_counter']}")
+        if gen_now:
+            src_for_video = res.get("composition_native_url") or res.get("gallery_native_url")
+            if not src_for_video:
+                st.warning("No image available to generate video.")
+            else:
+                with st.spinner("Generating Video …"):
+                    try:
+                        video_prompt = build_video_prompt(st.session_state.get("scene_name", "Studio Cinematic"),
+                                                          st.session_state.get("scene_extra",""))
+                        v = post_video(
+                            src_for_video,
+                            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("preview_aspect","3:4"),
+                            prompt_text=video_prompt,
+                            prompt_optimizer=False,
+                        )
+                        vurl2 = (v.get("result") or {}).get("video", {}).get("url")
+                        ui_hints = v.get("ui_hints") or {}
+                        if vurl2:
+                            st.session_state["results"]["video_url"] = vurl2
+                            if isinstance(ui_hints, dict):
+                                st.session_state["video_ui_max_px"] = int(ui_hints.get("suggested_video_max_px", VIDEO_MAX_PX_DEFAULT))
+                            st.success("Video is ready. Showing preview…")
+                            st.rerun()
+                        else:
+                            st.info("Video URL not returned.")
+                    except requests.HTTPError as e:
+                        st.error(f"HTTP {e.response.status_code}")
+                    except Exception as e:
+                        st.error(f"Error: {e}")
+
+    # ZIP bundle (bottom row)
+    st.markdown("---")
+    exp_col = st.container()
+    with exp_col:
+        st.markdown("#### Export")
+        named = []
+        gnat = fetch_bytes(res.get("gallery_native_url")) if res.get("gallery_native_url") else None
+        cnat = fetch_bytes(res.get("composition_native_url")) if res.get("composition_native_url") else None
+        if gnat: named.append((f"gallery_native{infer_ext(gnat)}", gnat))
+        if cnat: named.append((f"ai_configurated_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,