Upload 5 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+tryon.jpg filter=lfs diff=lfs merge=lfs -text

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,428 @@
+# streamlit_app.py — Fit Studio AI (Fashion & Apparel) v0.4 UI
+# Still Image (VTO) → (optional) Runway Video
+# Demo preview: same folder "input.jpg" (girdi) ve "tryon.jpg" (örnek çıktı)
+
+import os, io, zipfile, requests, streamlit as st
+from pathlib import Path
+from PIL import Image
+
+# ================= Env & API (robust) =================
+def _env(k):
+    return (os.getenv(k) or "").strip().strip("'\"")
+
+API_BASE = (
+    _env("FITSTUDIO_API")
+    or _env("AI_LIGHTBOX_API")
+    or _env("LUXFIT_API")
+    or _env("AI_BEAUTYBOX_API")
+    or _env("AI_BEAUTY_API")
+).rstrip("/") if (
+    _env("FITSTUDIO_API")
+    or _env("AI_LIGHTBOX_API")
+    or _env("LUXFIT_API")
+    or _env("AI_BEAUTYBOX_API")
+    or _env("AI_BEAUTY_API")
+) else ""
+
+HF_TOKEN = (
+    _env("FITSTUDIO_TOKEN")
+    or _env("AI_LIGHTBOX_TOKEN")
+    or _env("LUXFIT_TOKEN")
+    or _env("AI_BEAUTYBOX_TOKEN")
+    or _env("AI_BEAUTY_TOKEN")
+)
+
+if not API_BASE:
+    st.error("API_BASE bulunamadı. Settings > Variables: FITSTUDIO_API (veya AI_LIGHTBOX_API / LUXFIT_API) tanımlayın.")
+    st.stop()
+
+HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
+PREVIEW_SIZE = int(os.getenv("PREVIEW_SIZE", "512"))
+
+# ================= Session State (ilk yük) =================
+defaults = {
+    "results": None,           # son çalıştırmanın sonuçları
+    "job_counter": 0,          # indirme butonları için unique key
+    "duration": "6",           # video süresi varsayılan
+}
+for k, v in defaults.items():
+    if k not in st.session_state:
+        st.session_state[k] = v
+
+# ================= Yardımcılar =================
+def _needs_auth(url: str) -> bool:
+    return (url.startswith(API_BASE) or url.startswith("outputs/"))
+
+@st.cache_data(ttl=300, show_spinner=False)
+def fetch_bytes(url_or_path: str | None):
+    """
+    URL, outputs/ yolu veya mevcut klasörden dosya okur.
+    """
+    if not url_or_path:
+        return None
+    try:
+        p = Path(url_or_path)
+        if p.is_file():
+            return p.read_bytes()
+        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)
+        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 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 in [".jpg",".jpeg"] else ("image/png" if ext==".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()
+
+def backend_ok() -> bool:
+    try:
+        r = requests.get(f"{API_BASE}/health", headers=HEADERS if HF_TOKEN else None, timeout=10)
+        r.raise_for_status()
+        return True
+    except Exception:
+        return False
+
+def post_image_edit(data: dict, files_payload):
+    r = requests.post(f"{API_BASE}/v1/image/edit", data=data, files=files_payload or None,
+                      headers=HEADERS if HF_TOKEN else None, timeout=600)
+    r.raise_for_status()
+    return r.json()
+
+def post_chain(data: dict, files_payload):
+    payload = {**data, "to_video": "false"}  # video UI'da ayrı akış
+    r = requests.post(f"{API_BASE}/v1/tryon/chain", data=payload, files=files_payload or None,
+                      headers=HEADERS if HF_TOKEN else None, timeout=600)
+    r.raise_for_status()
+    return r.json()
+
+def post_video(image_url: str, duration="6", resolution="768P", prompt_optimizer=True):
+    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()
+    return r.json()
+
+# ================= Sayfa =================
+st.set_page_config(page_title="Fit Studio AI", layout="wide", page_icon="🧵")
+st.title("🧵 Fit Studio AI")
+st.caption("Fashion & Apparel — Still Image (VTO) → (optional) Runway Video")
+
+# ================= Üst Kontrol Şeridi =================
+ok = backend_ok()
+c0, c1, c2, c3, c4 = st.columns([1.0, 1.8, 1.8, 1.4, 1.6])
+
+with c0:
+    category = st.selectbox("Kategori", ["general","underwear"], index=0, key="category")
+
+with c1:
+    file_list = st.file_uploader("Garment image (upload)", type=["jpg","jpeg","png","webp"], accept_multiple_files=True, key="file_upl")
+
+with c2:
+    image_url = st.text_input("veya Garment URL", key="image_url")
+
+with c3:
+    num_images = st.selectbox("Çıktı adedi", [1,2,3,4], index=0, key="num_images")
+
+with c4:
+    custom_prompt = st.text_area("Özel prompt (boş bırak → varsayılan)", value="", height=80, key="custom_prompt")
+
+bA, bB = st.columns([1.0, 1.0])
+with bA:
+    st.info(f"Backend: {'Online' if ok else 'Offline'}", icon="🔌")
+with bB:
+    # İki farklı çalıştırma seçeneği: sadece foto veya zincir
+    run_image = st.button("Çalıştır (Sadece Foto)", type="primary", use_container_width=True, key="run_img_btn")
+    run_chain = st.button("Zincir (Foto → Video hazırlık)", use_container_width=True, key="run_chain_btn")
+
+# ================= Demo Preview (mevcut klasörden) =================
+st.markdown("---")
+st.subheader("Demo preview (çalıştırmadan önce örnek)")
+st.caption("Bu bölüm sadece örnek; aynı klasörden **input.jpg** ve **tryon.jpg** okunur.")
+
+dp_in, dp_sp, dp_tp = st.columns(3)
+in_bytes = fetch_bytes("input.jpg")
+if in_bytes:
+    dp_in.subheader("Input")
+    dp_in.image(square_preview_bytes(in_bytes, size=PREVIEW_SIZE, bg_rgb=(255,255,255)), use_container_width=True)
+    sz = image_size_from_bytes(in_bytes)
+    if sz: dp_in.caption(f"Input: {sz[0]}×{sz[1]} px")
+    dp_in.download_button("İndir (tam kalite)", data=in_bytes, file_name=f"input{infer_ext(in_bytes)}",
+                          mime="image/jpeg", key="demo_input_dl")
+else:
+    show_tile(dp_in, "Input", None)
+
+try_bytes = fetch_bytes("tryon.jpg")
+if try_bytes:
+    dp_sp.subheader("Sample Output")
+    dp_sp.image(square_preview_bytes(try_bytes, size=PREVIEW_SIZE, bg_rgb=(255,255,255)), use_container_width=True)
+    sz2 = image_size_from_bytes(try_bytes)
+    if sz2: dp_sp.caption(f"Output: {sz2[0]}×{sz2[1]} px")
+    dp_sp.download_button("İndir (tam kalite)", data=try_bytes, file_name=f"tryon{infer_ext(try_bytes)}",
+                          mime="image/jpeg", key="demo_try_dl")
+else:
+    show_tile(dp_sp, "Sample Output", None)
+
+show_tile(dp_tp, "Video (örnek yok)", None)
+
+# ================= Girdi Önizleme =================
+st.markdown("---")
+g1, g2, g3, g4 = st.columns(4)
+white_bg = (255,255,255)
+
+input_preview = None
+if file_list:
+    try:
+        input_preview = file_list[0].getvalue()
+    except Exception:
+        input_preview = None
+elif image_url:
+    input_preview = fetch_bytes(image_url)
+
+if input_preview:
+    g1.subheader("Input")
+    g1.image(square_preview_bytes(input_preview, size=PREVIEW_SIZE, bg_rgb=white_bg), use_container_width=True)
+    try:
+        sz = Image.open(io.BytesIO(input_preview)).size
+        g1.caption(f"{sz[0]}×{sz[1]} px")
+    except Exception:
+        g1.caption("—")
+else:
+    show_tile(g1, "Input", None)
+
+# ================ Çalıştır ================
+def _build_files_payload():
+    files_payload = []
+    if file_list:
+        for f in file_list:
+            files_payload.append(("files", (f.name, f.getvalue(), f.type or "image/jpeg")))
+    return files_payload
+
+def _common_form_data():
+    data = {
+        "category": st.session_state["category"],
+        "num_images": str(st.session_state["num_images"]),
+    }
+    if st.session_state.get("custom_prompt","").strip():
+        data["prompt"] = st.session_state["custom_prompt"].strip()
+    if image_url:
+        data["image_urls"] = image_url.strip()
+    return data
+
+if run_image or run_chain:
+    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 garment görseli girin (upload veya URL).")
+    else:
+        try:
+            files_payload = _build_files_payload()
+            form = _common_form_data()
+
+            with st.spinner("Çalışıyor…"):
+                if run_image:
+                    out = post_image_edit(form, files_payload)
+                    image_json = out
+                else:
+                    out = post_chain(form, files_payload)   # to_video=false
+                    image_json = out.get("image_step") or out  # Fallback
+
+            # ---- Çıktıları çıkar ----
+            imgs = (image_json.get("result") or {}).get("images") or []
+            urls = [i.get("url") for i in imgs if isinstance(i, dict) and i.get("url")]
+            # UI state
+            st.session_state["job_counter"] += 1
+            st.session_state["results"] = {
+                "job_id": (image_json.get("job_id") or out.get("job_id")),
+                "schema_version": (image_json.get("schema_version") or out.get("schema_version")),
+                "image_urls": urls,
+                "video_url": None,
+            }
+
+        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 ================
+res = st.session_state.get("results") or {}
+vbytes = None
+if res:
+    key_prefix = f"job{st.session_state['job_counter']}"
+    img_urls = res.get("image_urls") or []
+
+    # İlk 4 görseli göster
+    cols = st.columns(4)
+    named = []
+    if img_urls:
+        for idx in range(min(4, len(img_urls))):
+            u = img_urls[idx]
+            c = cols[idx] if idx < len(cols) else st
+            title = "Look" if idx == 0 else f"Look #{idx+1}"
+            _, b = show_tile(c, title, u, bg_rgb=white_bg, filename_stub=f"look_{idx+1}", key_prefix=f"{key_prefix}_{idx}")
+            if b:
+                named.append((f"look_{idx+1}{infer_ext(b)}", b))
+    else:
+        show_tile(cols[0], "Look", None)
+
+    # ---- Video (ilk görselden) ----
+    st.markdown("---")
+    colv1, colv2, colv3 = st.columns([1.1, 1.0, 1.2])
+    with colv1:
+        gen_video = st.checkbox("Bu sonuçtan video üret", value=False, key="video_toggle")
+    with colv2:
+        st.selectbox("Video süresi", ["6","10"], index=(0 if st.session_state["duration"]=="6" else 1), key="duration_select")
+    with colv3:
+        go_video = st.button("Video Oluştur", use_container_width=True, key="video_btn")
+
+    if gen_video and go_video:
+        src_for_video = img_urls[0] if img_urls else None
+        if not src_for_video:
+            st.warning("Video için kaynak görsel yok.")
+        else:
+            with st.spinner("Video üretiliyor…"):
+                v = post_video(
+                    src_for_video,
+                    duration=st.session_state.get("duration_select","6"),
+                    resolution="768P",
+                    prompt_optimizer=True
+                )
+            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(
+                "Video indir (MP4)",
+                data=vb,
+                file_name="fitstudio_video.mp4",
+                mime="video/mp4",
+                key=f"{key_prefix}_video_dl"
+            )
+        else:
+            st.info("Video URL var ama içerik indirilemedi.")
+
+    # ---- ZIP indir (isteğe bağlı) ----
+    if named or vbytes:
+        if vbytes:
+            named.append(("video.mp4", vbytes))
+        zip_bytes = make_zip(named)
+        st.download_button(
+            "Tümünü indir (ZIP)",
+            data=zip_bytes,
+            file_name="fitstudio_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"),
+            "api_base": API_BASE,
+            "has_auth_header": bool(HF_TOKEN),
+            "image_urls": img_urls,
+            "video_url": res.get("video_url"),
+        })
+
+# ================= Sidebar (teşhis) =================
+with st.sidebar:
+    st.caption(f"Resolved API_BASE: {API_BASE}")
+    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)
+            st.write(r.status_code)
+            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"):
+        st.cache_data.clear(); st.success("Cache cleared.")
+    if st.button("Clear results"):
+        st.session_state["results"] = None; st.success("Results cleared.")