Upload streamlit_app.py

streamlit_app.py

@@ -1,14 +1,14 @@
-# streamlit_app.py — AI LightBox · Beauty (v0.3 UI)
+# streamlit_app.py — AI LightBox · Beauty (v0.3 UI + Minimal Demo: input.jpg & tryon.jpg)
 # Packshot → Padding (square/canvas) → Try-on (lips/eyes/face/nails/hair/brow/body/lifestyle) · SR · (optional) Video
 
 import os, io, zipfile, requests, streamlit as st
 from PIL import Image
+from pathlib import Path
 
 # ================= Env & API (robust) =================
 def _env(k): 
     return (os.getenv(k) or "").strip().strip("'\"")
 
-# Desteklenen env anahtarları (öncelik sırası)
 API_BASE = (
     _env("AI_BEAUTYBOX_API") or
     _env("AI_LIGHTBOX_API")  or
@@ -29,12 +29,19 @@ if not API_BASE:
 
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 PREVIEW_SIZE = int(os.getenv("PREVIEW_SIZE", "512"))
+APP_DIR = Path(__file__).parent
+
+# ================= Demo dosyaları (yalnızca yerel klasör) =================
+DEMO_INPUT = str(APP_DIR / "input.jpg")
+DEMO_TRYON = str(APP_DIR / "tryon.jpg")
 
 # ================= Session State (ilk yükte) =================
 defaults = {
     "results": None,
     "job_counter": 0,
     "duration": "6",
+    # input.jpg ve tryon.jpg varsa demo otomatik açık başlar
+    "demo_on": (Path(DEMO_INPUT).is_file() and Path(DEMO_TRYON).is_file()),
 }
 for k, v in defaults.items():
     if k not in st.session_state:
@@ -51,7 +58,6 @@ HAIR = {"hair_color"}
 REGIONS = ["auto","lips","eyes","face","nails","hand","hair","brow","body","lifestyle"]
 
 def tryon_label(category: str, region_override: str, sr: bool) -> str:
-    # Region override önce gelir
     r = (region_override or "auto").lower()
     if r != "auto":
         base = {
@@ -60,7 +66,6 @@ def tryon_label(category: str, region_override: str, sr: bool) -> str:
             "lifestyle":"Lifestyle"
         }.get(r, "Try-on")
         return f"{base} (SR)" if sr else base
-
     c = (category or "auto").lower()
     if c in LIPS: base = "On-Lips"
     elif c in EYES: base = "On-Eyes"
@@ -91,12 +96,8 @@ def post_chain(data: dict, files_payload):
     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",
-    }
+    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 if HF_TOKEN else None, timeout=600)
     r.raise_for_status()
@@ -104,8 +105,12 @@ def post_video(image_url: str, duration="6", resolution="768P", prompt_optimizer
 
 @st.cache_data(ttl=300, show_spinner=False)
 def fetch_bytes(url_or_path: str):
-    if not url_or_path:
-        return None
+    if not url_or_path: return None
+    p = Path(url_or_path)
+    if p.is_file():
+        try: return p.read_bytes()
+        except Exception: return None
+    # outputs/ veya URL
     try:
         url = f"{API_BASE}/{url_or_path}" if url_or_path.startswith("outputs/") else url_or_path
         headers = (HEADERS if (_needs_auth(url) and HF_TOKEN) else None)
@@ -118,8 +123,7 @@ def fetch_bytes(url_or_path: str):
 @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
+        im = Image.open(io.BytesIO(b)); return im.size
     except Exception:
         return None
 
@@ -127,15 +131,14 @@ 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")
+        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)
+    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))
@@ -144,57 +147,40 @@ def square_preview_bytes(img_bytes: bytes, size: int = PREVIEW_SIZE, bg_rgb=(255
     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
+    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
+        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
-
+        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
-
+        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")
-
+    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"
-    )
+    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()
+            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="💄")
@@ -204,37 +190,20 @@ st.caption("Packshot → Padding (canvas) → Try-on (lips/eyes/face/nails/hair/
 # ================= Üst Kontrol Şeridi =================
 ok = backend_ok()
 col1, col2, col3, col4, col5, col6, col7 = st.columns([1.8, 1.6, 1.4, 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",
-        [
-            # Lips
-            "auto","lipstick","lipgloss","lipliner",
-            # Eyes
-            "eyeshadow","eyeliner","mascara","brow","false_lashes",
-            # Face/complexion
-            "foundation","concealer","blush","bronzer","contour","highlighter","body_makeup",
-            # Nails
-            "nail_polish",
-            # Lifestyle
-            "skincare","perfume","tools","set",
-            # Hair/Body care
-            "hair_color","haircare","shampoo","conditioner","body_lotion","body_care",
-        ],
-        index=0, key="category"
-    )
+    category = st.selectbox("Category",
+        ["auto","lipstick","lipgloss","lipliner",
+         "eyeshadow","eyeliner","mascara","brow","false_lashes",
+         "foundation","concealer","blush","bronzer","contour","highlighter","body_makeup",
+         "nail_polish","skincare","perfume","tools","set",
+         "hair_color","haircare","shampoo","conditioner","body_lotion","body_care"], 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"
-    )
+    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", False, key="upscale")
 with col6:
@@ -242,7 +211,6 @@ with col6:
 with col7:
     upscale_factor = st.selectbox("SR factor", ["2","4"], index=0, key="upscale_factor")
 
-# == Tuval & Try-on kontrolleri ==
 t1, t2, t3, t4 = st.columns([1.15, 1.15, 1.2, 1.5])
 with t1:
     canvas_policy = st.selectbox("Canvas", ["fixed_1200","match_long_edge","keep_input"], index=0, key="canvas_policy")
@@ -261,61 +229,95 @@ with b2:
 with b3:
     identity_lock = st.checkbox("Identity lock", True, key="identity_lock")
 
+# ============ DEMO PREVIEW (input.jpg & tryon.jpg; backend yok) ============
+st.markdown("---")
+d1, d2 = st.columns([1.0, 1.0])
+with d1:
+    st.checkbox("Demo preview (çalıştırmadan örnek göster)", value=st.session_state["demo_on"], key="demo_on")
+with d2:
+    if st.session_state["demo_on"]:
+        st.info("Bu sadece örnek ön izleme. Run butonu demo açıkken pasif.", icon="👀")
+
+if st.session_state["demo_on"]:
+    c_in_demo, c_try_demo = st.columns(2)
+
+    # Input.jpg
+    ib = fetch_bytes(DEMO_INPUT)
+    c_in_demo.subheader("Input")
+    if ib:
+        c_in_demo.image(square_preview_bytes(ib, size=PREVIEW_SIZE, bg_rgb=(255,255,255)), use_container_width=True)
+        try:
+            wsz = Image.open(io.BytesIO(ib)).size
+            c_in_demo.caption(f"Input resolution: {wsz[0]}×{wsz[1]} px")
+        except Exception:
+            c_in_demo.caption("—")
+        c_in_demo.download_button("İndir (tam kalite)", data=ib, file_name=f"demo_input{infer_ext(ib)}",
+                                  mime="image/jpeg", key="demo_input_dl")
+    else:
+        c_in_demo.warning("input.jpg bulunamadı.")
+
+    # Tryon.jpg
+    show_tile(c_try_demo, "On-Hair (örnek)", DEMO_TRYON, filename_stub="demo_tryon", key_prefix="demo")
+
+    if st.button("Bu örneği forma aktar ve demoyu kapat", use_container_width=True, key="use_demo_btn"):
+        st.session_state["image_url"] = DEMO_INPUT  # yerel dosyayı kullan
+        if st.session_state.get("category") == "auto":
+            st.session_state["category"] = "hair_color"
+        st.session_state["apply_region"] = "hair"
+        st.session_state["demo_on"] = False
+        st.experimental_rerun()
+
+# ================= Alt kontrol şeridi =================
 cA, cB = st.columns([1.0, 1.0])
 with cA:
     st.info(f"Backend: {'Online' if ok else 'Offline'}", icon="🔌")
 with cB:
-    run = st.button("Run", type="primary", use_container_width=True, key="run_btn")
+    run = st.button("Run", type="primary", use_container_width=True, key="run_btn", disabled=st.session_state["demo_on"])
 
-# ================= Girdi Önizleme =================
-c_in, c_pack, c_pad, c_try = st.columns(4)
+# ================= Girdi Önizleme (form girdisi) =================
+c_in2, c_pack, c_pad, c_try = 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)
+    try: input_preview_bytes = file_list[0].getvalue()
+    except Exception: input_preview_bytes = None
+elif st.session_state.get("image_url"):
+    input_preview_bytes = fetch_bytes(st.session_state["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)
+    c_in2.subheader("Input")
+    c_in2.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")
+        c_in2.caption(f"Input resolution: {sz[0]}×{sz[1]} px")
     except Exception:
-        c_in.caption("—")
+        c_in2.caption("—")
 else:
-    show_tile(c_in, "Input", None)
+    show_tile(c_in2, "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):
+    elif not (file_list or st.session_state.get("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,
+                "category": st.session_state["category"],
                 "edit_prompt": "",
                 "num_images": "1",
-                "image_urls": (image_url or ""),
+                "image_urls": (st.session_state.get("image_url") or ""),
                 "mannequin_image_url": (model_ref_url or ""),
                 "padding_ratio": str(padding_ratio),
                 "identity_lock": "true" if identity_lock else "false",
-                # Canvas
-                "canvas_policy": canvas_policy,
+                "canvas_policy": st.session_state["canvas_policy"],
                 "canvas_size": str(int(canvas_size or 1200)),
-                # Shade / Region
                 "shade_hex": (shade_hex or "").strip() or None,
                 "shade_swatch_url": (shade_swatch_url or "").strip() or None,
-                "apply_region": apply_region,
-                # SR
+                "apply_region": st.session_state["apply_region"],
                 "upscale": "true" if upscale else "false",
                 "upscale_factor": st.session_state["upscale_factor"],
                 "upscale_stage": st.session_state["upscale_stage"],
@@ -332,20 +334,16 @@ if run:
             packshot_url = None
             try:
                 imgs = out["packshot_step"]["result"]["images"]
-                if imgs:
-                    packshot_url = imgs[0].get("url")
-            except Exception:
-                pass
+                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
+                    if pads: padded_url = pads[0]
+            except Exception: pass
 
             packshot_sr_list = first_present(out.get("packshot_step", {}), [
                 "upscaled_packshot_urls", "upscaled_packshot_files"
@@ -355,10 +353,8 @@ if run:
             placed_url = None
             try:
                 imgs = out["image_step"]["result"]["images"]
-                if imgs:
-                    placed_url = imgs[0].get("url")
-            except Exception:
-                pass
+                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"
@@ -381,9 +377,8 @@ if run:
                 "video_url": None,
                 "api_base": API_BASE,
                 "has_auth_header": bool(HEADERS),
-                # UI state snapshot (etiket/isimlendirme için)
-                "category": category,
-                "apply_region": apply_region,
+                "category": st.session_state["category"],
+                "apply_region": st.session_state["apply_region"],
             }
 
         except requests.HTTPError as e:
@@ -398,22 +393,15 @@ 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
-    )
+    p_url, p_bytes = show_tile(c_pack, p_title, res.get("packshot_sr_url") or res.get("packshot_url"),
+                               bg_rgb=(255,255,255), filename_stub="packshot", key_prefix=key_prefix)
+    pad_url, pad_bytes = show_tile(c_pad, "Padded", res.get("padded_url"),
+                                   bg_rgb=(255,255,255), filename_stub="padded", key_prefix=key_prefix)
 
     shown_final_url = res.get("placed_sr_url") or res.get("placed_url")
     m_title = tryon_label(res.get("category","auto"), res.get("apply_region","auto"), bool(res.get("placed_sr_url")))
-    m_url, m_bytes = show_tile(
-        c_try, m_title, shown_final_url,
-        bg_rgb=white_bg, filename_stub="tryon", key_prefix=key_prefix
-    )
+    m_url, m_bytes = show_tile(c_try, m_title, shown_final_url,
+                               bg_rgb=(255,255,255), filename_stub="tryon", key_prefix=key_prefix)
 
     # ---- Video (final görselden) ----
     st.markdown("---")
@@ -431,17 +419,11 @@ if res:
             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_select","6"),
-                    resolution="768P",
-                    prompt_optimizer=False
-                )
+                v = post_video(src_for_video, duration=st.session_state.get("duration_select","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.")
+            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:
@@ -450,13 +432,8 @@ if res:
         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"
-            )
+            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.")
 
@@ -468,15 +445,9 @@ if res:
     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
+        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")
+
     with st.expander("Debug"):
         st.json({
             "job_id": res.get("job_id"),
@@ -500,14 +471,10 @@ with st.sidebar:
     st.caption("Auth header: " + ("ON" if HF_TOKEN else "OFF"))
     if st.button("Ping /health"):
         try:
-            r = requests.get(f"{API_BASE}/health",
-                             headers=HEADERS if HF_TOKEN else None,
-                             timeout=10)
+            r = requests.get(f"{API_BASE}/health", headers=HEADERS if HF_TOKEN else None, timeout=10)
             st.write(r.status_code)
-            try:
-                st.json(r.json())
-            except Exception:
-                st.code((r.text or "")[:1000])
+            try: st.json(r.json())
+            except Exception: st.code((r.text or "")[:1000])
         except Exception as e:
             st.error(f"Health error: {e}")
     if st.button("Clear cache"):