Upload 3 files

Dockerfile

@@ -0,0 +1,13 @@
+FROM python:3.11-slim
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+WORKDIR /app
+
+COPY --chown=user requirements.txt .
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . .
+EXPOSE 7860
+CMD ["streamlit", "run", "streamlit_app.py", "--server.address=0.0.0.0", "--server.port=7860"]

requirements.txt

@@ -0,0 +1,12 @@
+streamlit
+httpx>=0.27
+pyjwt>=2.9
+fastapi
+uvicorn[standard]
+pillow
+requests
+fal-client
+python-multipart
+aiofiles
+websockets
+websocket-client

streamlit_app.py

@@ -0,0 +1,417 @@
+# streamlit_app.py — AI LightBox · Beauty (packshot + on-face + SR + video)
+import os, io, zipfile, requests, streamlit as st
+from PIL import Image
+
+# ================= Env & API =================
+API_BASE = (os.getenv("AI_LIGHTBOX_API", "http://127.0.0.1:8000") or "http://127.0.0.1:8000").strip().strip("'\"").rstrip("/")
+HF_TOKEN = (os.getenv("AI_LIGHTBOX_TOKEN", "") or "").strip().strip("'\"")
+HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
+PREVIEW_SIZE = int(os.getenv("PREVIEW_SIZE", "512"))
+
+# ================= Session State =================
+if "results" not in st.session_state:
+    st.session_state["results"] = None
+if "job_counter" not in st.session_state:
+    st.session_state["job_counter"] = 0
+if "duration" not in st.session_state:
+    st.session_state["duration"] = "6"
+
+# ================= Yardımcılar =================
+def _needs_auth(url: str) -> bool:
+    return url.startswith(API_BASE) or url.startswith("outputs/")
+
+def backend_ok() -> bool:
+    try:
+        r = requests.get(f"{API_BASE}/health", headers=HEADERS, timeout=10)
+        r.raise_for_status()
+        return True
+    except Exception:
+        return False
+
+def post_chain(data: dict, files_payload):
+    data = {**data, "to_video": "false"}  # videoyu ayrı tetikleriz
+    r = requests.post(f"{API_BASE}/v1/tryon/chain", data=data, files=files_payload or None,
+                      headers=HEADERS, timeout=600)
+    r.raise_for_status()
+    return r.json()
+
+def post_video(image_url: str, duration="6", resolution="768P", prompt_optimizer=False):
+    payload = {
+        "image_url": image_url,
+        "duration": duration,
+        "resolution": resolution,
+        "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()
+    return r.json()
+
+@st.cache_data(ttl=300, show_spinner=False)
+def fetch_bytes(url_or_path: str):
+    if not url_or_path:
+        return None
+    try:
+        url = f"{API_BASE}/{url_or_path}" if url_or_path.startswith("outputs/") else url_or_path
+        headers = HEADERS if _needs_auth(url) else None
+        r = requests.get(url, headers=headers, timeout=180)
+        r.raise_for_status()
+        return r.content
+    except Exception:
+        return None
+
+@st.cache_data(ttl=300, show_spinner=False)
+def image_size_from_bytes(b: bytes):
+    try:
+        im = Image.open(io.BytesIO(b))
+        return im.size
+    except Exception:
+        return None
+
+def infer_ext(data: bytes) -> str:
+    try:
+        im = Image.open(io.BytesIO(data))
+        fmt = (im.format or "JPEG").lower()
+        return "." + {"jpeg": "jpg", "jpg": "jpg", "png": "png", "webp": "webp"}.get(fmt, "jpg")
+    except Exception:
+        return ".jpg"
+
+@st.cache_data(ttl=300, show_spinner=False)
+def square_preview_bytes(img_bytes: bytes, size: int = PREVIEW_SIZE, bg_rgb=(255, 255, 255)) -> bytes:
+    im = Image.open(io.BytesIO(img_bytes)).convert("RGBA")
+    w, h = im.size
+    scale = min(size / w, size / h)
+    new_w, new_h = max(1, int(w * scale)), max(1, int(h * scale))
+    im_resized = im.resize((new_w, new_h), Image.LANCZOS)
+    canvas = Image.new("RGBA", (size, size), (*bg_rgb, 255))
+    off = ((size - new_w) // 2, (size - new_h) // 2)
+    canvas.paste(im_resized, off, im_resized)
+    buf = io.BytesIO(); canvas.save(buf, format="PNG"); return buf.getvalue()
+
+def clamp(v: float, lo: float, hi: float) -> float:
+    try:
+        v = float(v)
+    except Exception:
+        v = (lo + hi) / 2.0
+    return max(lo, min(hi, v))
+
+def first_present(d: dict, keys: list, default=None):
+    for k in keys:
+        v = d.get(k)
+        if v:
+            return v
+    return default
+
+def show_tile(col, title, url: str | None, bg_rgb=(255,255,255), filename_stub="image", key_prefix=""):
+    col.subheader(title)
+    if not url:
+        placeholder = Image.new("RGBA", (PREVIEW_SIZE, PREVIEW_SIZE), (0, 0, 0, 0))
+        col.image(placeholder, use_container_width=True)
+        col.caption("—")
+        return None, None
+
+    b = fetch_bytes(url)
+    if not b:
+        col.warning("Görsel yüklenemedi")
+        return None, None
+
+    pv = square_preview_bytes(b, size=PREVIEW_SIZE, bg_rgb=bg_rgb)
+    col.image(pv, use_container_width=True)
+    sz = image_size_from_bytes(b)
+    if sz:
+        col.caption(f"Gerçek çözünürlük: {sz[0]}×{sz[1]} px")
+
+    ext = infer_ext(b)
+    mime = "image/jpeg" if ext.lower() in [".jpg", ".jpeg"] else ("image/png" if ext.lower()==".png" else "image/webp")
+    col.download_button(
+        "İndir (tam kalite)",
+        data=b,
+        file_name=f"{filename_stub}{ext}",
+        mime=mime,
+        key=f"{key_prefix}_{filename_stub}_dl"
+    )
+    return url, b
+
+def make_zip(named_bytes: list[tuple[str, bytes]]) -> bytes:
+    buf = io.BytesIO()
+    with zipfile.ZipFile(buf, "w", compression=zipfile.ZIP_DEFLATED) as z:
+        for fname, b in named_bytes:
+            if b:
+                z.writestr(fname, b)
+    buf.seek(0)
+    return buf.read()
+
+# ================= Sayfa =================
+st.set_page_config(page_title="AI LightBox · Beauty", layout="wide", page_icon="💄")
+st.title("💄 AI LightBox · Beauty")
+st.caption("Packshot → Padding (kare) → On-Face yerleştirme · Super Resolution · (opsiyonel) Video")
+
+# ================= Üst Kontrol Şeridi =================
+ok = backend_ok()
+col1, col2, col3, col4, col5, col6, col7 = st.columns([1.7, 1.6, 1.2, 1.3, 0.9, 1.1, 0.9])
+
+with col1:
+    file_list = st.file_uploader("Product image (upload)", type=["jpg","jpeg","png","webp"], accept_multiple_files=True, key="file_upl")
+with col2:
+    image_url = st.text_input("or Product URL", key="image_url")
+with col3:
+    category = st.selectbox(
+        "Category",
+        [
+            "auto","lipstick","lipgloss","lipliner",
+            "eyeshadow","eyeliner","mascara",
+            "foundation","concealer","blush","bronzer","contour","highlighter",
+            "nail_polish","skincare","perfume","tools","set"
+        ],
+        index=0, key="category"
+    )
+with col4:
+    padding_ratio_val = st.number_input(
+        "Content scale (0.30–0.95)",
+        min_value=0.30, max_value=0.95, value=0.50, step=0.01,
+        help="Kare tuvalde ürünün uzun kenarı = tuval × bu oran.",
+        key="padding_ratio"
+    )
+with col5:
+    upscale = st.checkbox("Super Resolution", True, key="upscale")
+with col6:
+    upscale_stage = st.selectbox("SR stage", ["both","final","packshot"], index=0, help="Default: both", key="upscale_stage")
+with col7:
+    upscale_factor = st.selectbox("SR factor", ["2","4"], index=0, key="upscale_factor")
+
+colA, colB, colC, colD, colE = st.columns([1.0, 1.1, 1.2, 0.9, 1.0])
+with colA:
+    st.info(f"Backend: {'Online' if ok else 'Offline'}", icon="🔌")
+with colB:
+    identity_lock = st.checkbox("Identity lock", True, key="identity_lock")
+with colC:
+    to_video = st.checkbox("Enable video (later)", False, key="to_video")
+with colD:
+    duration = st.selectbox("Video sec", ["6","10"], index=0, key="duration")
+    st.session_state["duration"] = duration
+with colE:
+    run = st.button("Run", type="primary", use_container_width=True, key="run_btn")
+
+# == Tuval seçenekleri ==
+colT1, colT2 = st.columns([1.3, 1.0])
+with colT1:
+    canvas_policy = st.selectbox("Canvas", ["fixed_1200","match_long_edge","keep_input"], index=0, key="canvas_policy")
+with colT2:
+    canvas_size = st.number_input("Canvas size (px)", min_value=256, max_value=8192, value=1200, step=64, key="canvas_size")
+
+face_url = st.text_input("Face image URL (optional)", key="face_url")
+
+# ================= Girdi Önizleme =================
+c_in, c_pack, c_pad, c_face = st.columns(4)
+white_bg = (255, 255, 255)
+
+input_preview_bytes = None
+if file_list:
+    try:
+        input_preview_bytes = file_list[0].getvalue()
+    except Exception:
+        input_preview_bytes = None
+elif image_url:
+    input_preview_bytes = fetch_bytes(image_url)
+
+if input_preview_bytes:
+    c_in.subheader("Input")
+    c_in.image(square_preview_bytes(input_preview_bytes, size=PREVIEW_SIZE, bg_rgb=white_bg), use_container_width=True)
+    try:
+        sz = Image.open(io.BytesIO(input_preview_bytes)).size
+        c_in.caption(f"Input resolution: {sz[0]}×{sz[1]} px")
+    except Exception:
+        c_in.caption("—")
+else:
+    show_tile(c_in, "Input", None)
+
+# ================ Koş & Sonuçları Kaydet ================
+if run:
+    if not ok:
+        st.error("Backend erişilemiyor. API_BASE / TOKEN kontrol edin.")
+    elif not (file_list or image_url):
+        st.error("En az bir product görseli girin (upload veya URL).")
+    else:
+        try:
+            padding_ratio = clamp(padding_ratio_val, 0.30, 0.95)
+            data = {
+                "category": category,
+                "edit_prompt": "",
+                "num_images": "1",
+                "image_urls": (image_url or ""),
+                "mannequin_image_url": (face_url or ""),  # face ref
+                "padding_ratio": str(padding_ratio),
+                "identity_lock": "true" if identity_lock else "false",
+                # SR
+                "upscale": "true" if upscale else "false",
+                "upscale_factor": upscale_factor,
+                "upscale_stage": upscale_stage,
+                # Canvas
+                "canvas_policy": canvas_policy,
+                "canvas_size": str(int(canvas_size or 1200)),
+            }
+            files_payload = []
+            if file_list:
+                for f in file_list:
+                    files_payload.append(("files", (f.name, f.getvalue(), f.type or "image/jpeg")))
+
+            with st.spinner("Running chain…"):
+                out = post_chain(data, files_payload)
+
+            # ---- Çıktıları topla ----
+            packshot_url = None
+            try:
+                imgs = out["packshot_step"]["result"]["images"]
+                if imgs:
+                    packshot_url = imgs[0].get("url")
+            except Exception:
+                pass
+
+            padded_url = None
+            try:
+                padded_url = out.get("padding_step", {}).get("saved_file")
+                if not padded_url:
+                    pads = out.get("packshot_step", {}).get("padded_urls") or out.get("padded_urls") or []
+                    if pads:
+                        padded_url = pads[0]
+            except Exception:
+                pass
+
+            packshot_sr_list = first_present(out.get("packshot_step", {}), [
+                "upscaled_packshot_urls", "upscaled_packshot_files"
+            ], default=[]) or []
+            packshot_sr_url = packshot_sr_list[0] if packshot_sr_list else None
+
+            placed_url = None
+            try:
+                imgs = out["image_step"]["result"]["images"]
+                if imgs:
+                    placed_url = imgs[0].get("url")
+            except Exception:
+                pass
+
+            final_sr_list = first_present(out.get("image_step", {}), [
+                "upscaled_final_urls", "upscaled_final_files"
+            ], default=[]) or []
+            placed_sr_url = final_sr_list[0] if final_sr_list else None
+
+            st.session_state["job_counter"] += 1
+            st.session_state["results"] = {
+                "job_id": out.get("job_id"),
+                "schema_version": out.get("schema_version"),
+                "final_canvas": (
+                    out.get("packshot_step", {}).get("final_canvas")
+                    or out.get("padding_step", {}).get("final_canvas")
+                ),
+                "packshot_url": packshot_url,
+                "packshot_sr_url": packshot_sr_url,
+                "padded_url": padded_url,
+                "placed_url": placed_url,
+                "placed_sr_url": placed_sr_url,
+                "video_url": None,
+                "api_base": API_BASE,
+                "has_auth_header": bool(HEADERS),
+            }
+
+        except requests.HTTPError as e:
+            st.error(f"HTTP {e.response.status_code}")
+        except Exception as e:
+            st.error(f"Hata: {e}")
+
+# ================ Sonuçları Göster (kalıcı) ================
+res = st.session_state.get("results")
+vbytes = None
+if res:
+    key_prefix = f"job{st.session_state['job_counter']}"
+
+    p_title = "Packshot (SR)" if res.get("packshot_sr_url") else "Packshot"
+    p_url, p_bytes = show_tile(
+        c_pack, p_title, res.get("packshot_sr_url") or res.get("packshot_url"),
+        bg_rgb=white_bg, filename_stub="packshot", key_prefix=key_prefix
+    )
+
+    pad_url, pad_bytes = show_tile(
+        c_pad, "Padded", res.get("padded_url"),
+        bg_rgb=white_bg, filename_stub="padded", key_prefix=key_prefix
+    )
+
+    shown_final_url = res.get("placed_sr_url") or res.get("placed_url")
+    m_title = "On-Face (SR)" if res.get("placed_sr_url") else "On-Face"
+    m_url, m_bytes = show_tile(
+        c_face, m_title, shown_final_url,
+        bg_rgb=white_bg, filename_stub="onface", key_prefix=key_prefix
+    )
+
+    # ---- Video (final görselden) ----
+    if st.checkbox("Generate video from this result", value=False, key="video_toggle"):
+        src_for_video = res.get("placed_sr_url") or res.get("placed_url")
+        if not src_for_video:
+            st.warning("Video için final görsel yok.")
+        else:
+            with st.spinner("Generating video…"):
+                v = post_video(src_for_video, duration=st.session_state.get("duration", "6"), resolution="768P", prompt_optimizer=False)
+            vurl = (v.get("result") or {}).get("video", {}).get("url")
+            if vurl:
+                st.session_state["results"]["video_url"] = vurl
+            else:
+                st.info("Video URL gelmedi.")
+
+    vurl = st.session_state["results"].get("video_url")
+    if vurl:
+        st.subheader("Video")
+        st.video(vurl, format="video/mp4")
+        vb = fetch_bytes(vurl)
+        if vb:
+            vbytes = vb
+            st.download_button(
+                "Download video (MP4)",
+                data=vb,
+                file_name="ai_lightbox_video.mp4",
+                mime="video/mp4",
+                key=f"{key_prefix}_video_dl"
+            )
+        else:
+            st.info("Video URL var ama içerik indirilemedi.")
+
+    # ---- ZIP indir ----
+    named = []
+    if p_bytes: named.append((f"packshot{infer_ext(p_bytes)}", p_bytes))
+    if pad_bytes: named.append((f"padded{infer_ext(pad_bytes)}", pad_bytes))
+    if m_bytes: named.append((f"onface{infer_ext(m_bytes)}", m_bytes))
+    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="ai_lightbox_beauty_outputs.zip",
+            mime="application/zip",
+            key=f"{key_prefix}_zip_dl"
+        )
+
+    # Debug
+    with st.expander("Debug"):
+        st.json({
+            "job_id": res.get("job_id"),
+            "schema_version": res.get("schema_version"),
+            "final_canvas": res.get("final_canvas"),
+            "api_base": res.get("api_base"),
+            "has_auth_header": res.get("has_auth_header"),
+            "packshot_url": res.get("packshot_url"),
+            "packshot_sr_url": res.get("packshot_sr_url"),
+            "padded_url": res.get("padded_url"),
+            "placed_url": res.get("placed_url"),
+            "placed_sr_url": res.get("placed_sr_url"),
+            "video_url": res.get("video_url"),
+        })
+
+# ================= Sidebar =================
+with st.sidebar:
+    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("Results cleared.")
+    st.caption(f"API_BASE: {API_BASE}")
+    if HF_TOKEN:
+        st.caption("Auth: Bearer (active)")