Upload streamlit_app.py

streamlit_app.py

@@ -1,4 +1,4 @@
-# streamlit_app.py — AI LightBox · Beauty (packshot + on-face + SR + video)
+# streamlit_app.py — AI LightBox · Beauty (packshot + try-on + SR + video)
 import os, io, zipfile, requests, streamlit as st
 from PIL import Image
 
@@ -16,7 +16,24 @@ if "job_counter" not in st.session_state:
 if "duration" not in st.session_state:
     st.session_state["duration"] = "6"  # widget oluşturulmadan önce default
 
-# ================= Yardımcılar ================= 
+# ====== Try-on başlığı için kategori eşleme ======
+LIPS = {"lipstick","lipgloss","lipliner"}
+EYES = {"eyeshadow","eyeliner","mascara"}
+FACE = {"foundation","concealer","blush","bronzer","contour","highlighter"}
+HANDS = {"nail_polish"}
+LIFESTYLE = {"skincare","perfume","tools","set"}
+
+def tryon_label(cat: str, sr: bool) -> str:
+    c = (cat or "auto").lower()
+    if c in LIPS: base = "On-Lips"
+    elif c in EYES: base = "On-Eyes"
+    elif c in HANDS: base = "On-Nails"
+    elif c in FACE: base = "On-Face"
+    elif c in LIFESTYLE: base = "Lifestyle"
+    else: base = "Try-on"
+    return f"{base} (SR)" if sr else base
+
+# ================= Yardımcılar =================
 def _needs_auth(url: str) -> bool:
     return url.startswith(API_BASE) or url.startswith("outputs/")
 
@@ -143,7 +160,7 @@ def make_zip(named_bytes: list[tuple[str, bytes]]) -> bytes:
 # ================= Sayfa =================
 st.set_page_config(page_title="AI LightBox · Beauty", layout="wide", page_icon="💄")
 st.title("💄 AI LightBox · Beauty")
-st.caption("Packshot → Padding (square) → On-Face placement · Super Resolution · (optional) Video")
+st.caption("Packshot → Padding (square) → Try-on (lips/eyes/face/hand/lifestyle) · Super Resolution · (optional) Video")
 
 # ================= Üst Kontrol Şeridi =================
 ok = backend_ok()
@@ -197,10 +214,10 @@ with colT1:
 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")
+face_url = st.text_input("Model image URL (face/hand/eyes — optional)", key="face_url")
 
 # ================= Girdi Önizleme =================
-c_in, c_pack, c_pad, c_face = st.columns(4)
+c_in, c_pack, c_pad, c_try = st.columns(4)
 white_bg = (255, 255, 255)
 
 input_preview_bytes = None
@@ -237,7 +254,7 @@ if run:
                 "edit_prompt": "",
                 "num_images": "1",
                 "image_urls": (image_url or ""),
-                "mannequin_image_url": (face_url or ""),  # face ref
+                "mannequin_image_url": (face_url or ""),  # face/hand/eyes ref
                 "padding_ratio": str(padding_ratio),
                 "identity_lock": "true" if identity_lock else "false",
                 # SR
@@ -334,10 +351,10 @@ if res:
     )
 
     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_title = tryon_label(st.session_state.get("category","auto"), bool(res.get("placed_sr_url")))
     m_url, m_bytes = show_tile(
-        c_face, m_title, shown_final_url,
-        bg_rgb=white_bg, filename_stub="onface", key_prefix=key_prefix
+        c_try, m_title, shown_final_url,
+        bg_rgb=white_bg, filename_stub="tryon", key_prefix=key_prefix
     )
 
     # ---- Video (final görselden) ----
@@ -380,7 +397,7 @@ if res:
     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 m_bytes: named.append((f"tryon{infer_ext(m_bytes)}", m_bytes))
     if vbytes: named.append(("video.mp4", vbytes))
     if named:
         zip_bytes = make_zip(named)
@@ -418,4 +435,4 @@ with st.sidebar:
         st.success("Results cleared.")
     st.caption(f"API_BASE: {API_BASE}")
     if HF_TOKEN:
-        st.caption("Auth: Bearer (active)")
+        st.caption("Auth: Bearer (active)")