Spaces:
Running
Running
File size: 27,148 Bytes
c08826a f088758 852cbe6 26d3ab5 f088758 5fff162 26d3ab5 245d614 f088758 5fff162 c712bd1 4b66975 f088758 60a87f6 f088758 b919cc4 f088758 f19a0b1 245d614 f088758 f19a0b1 26d3ab5 b115180 26d3ab5 b115180 26d3ab5 bd3b61d b115180 852cbe6 60a87f6 3f3b1c8 bd3b61d b115180 852cbe6 b115180 b835664 26d3ab5 f19a0b1 26d3ab5 245d614 f19a0b1 b115180 26d3ab5 bbb9d35 f088758 5fff162 b835664 b115180 5fff162 b115180 f19a0b1 b115180 f19a0b1 b115180 f19a0b1 b115180 f088758 f19a0b1 b115180 f088758 2e0e104 f088758 4b66975 f19a0b1 4b66975 f19a0b1 4b66975 f088758 c08826a f088758 c08826a f088758 c08826a f088758 5fff162 4b66975 f088758 4b66975 f19a0b1 f088758 852cbe6 f19a0b1 f088758 f19a0b1 160571a f19a0b1 f088758 160571a f19a0b1 f088758 f19a0b1 f088758 f19a0b1 ceb4d59 f19a0b1 ceb4d59 f19a0b1 4cf040e f088758 26d3ab5 f088758 4fbdf25 f088758 4fbdf25 f088758 4fbdf25 f088758 4fbdf25 f088758 4fbdf25 f088758 4fbdf25 f088758 4fbdf25 f088758 4fbdf25 f088758 4fbdf25 f088758 4fbdf25 f088758 4fbdf25 f088758 631039e f19a0b1 697cb6b f088758 852cbe6 f19a0b1 852cbe6 f088758 c712bd1 f19a0b1 d855066 f088758 f19a0b1 c712bd1 f088758 f19a0b1 57f0396 f19a0b1 f088758 f19a0b1 5fff162 f088758 5fff162 f19a0b1 f088758 f19a0b1 f088758 f19a0b1 ceb4d59 f19a0b1 f088758 852cbe6 26d3ab5 f088758 f19a0b1 26d3ab5 f19a0b1 5fff162 f19a0b1 5fff162 f19a0b1 2e0e104 f19a0b1 4fbdf25 2e0e104 f19a0b1 4fd7ce6 f088758 f19a0b1 f088758 852cbe6 f19a0b1 852cbe6 2e0e104 f19a0b1 f088758 5fff162 c712bd1 f19a0b1 f088758 74dc30d 852cbe6 f19a0b1 ec7e8d3 f19a0b1 ec7e8d3 f19a0b1 f088758 852cbe6 f19a0b1 26d3ab5 11e9e20 f088758 5fff162 f19a0b1 5fff162 f088758 f19a0b1 5fff162 f088758 f19a0b1 9e2d60e ec7e8d3 f19a0b1 4fd7ce6 f088758 f19a0b1 9e2d60e ec7e8d3 f19a0b1 5fff162 f088758 f19a0b1 9e2d60e ec7e8d3 f19a0b1 5fff162 f088758 160571a 9e2d60e 5fff162 f19a0b1 5fff162 f19a0b1 9e2d60e f19a0b1 c08826a f19a0b1 9e2d60e c08826a f19a0b1 f088758 f19a0b1 f088758 f19a0b1 f088758 f19a0b1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 | # streamlit_app.py — AI LightBox · Jewelry (dark-orange UI, fixed video refresh)
# Pre-run sabit 4 kart (Input/Packshot/Padded/Video loop) + Run sonrası native odaklı kartlar
import os, io, base64, zipfile, requests, streamlit as st
from PIL import Image
# ===== Theme (programmatic dark orange) =====
try:
st._config.set_option("theme.base", "dark")
st._config.set_option("theme.primaryColor", "#ff7a1a") # dark orange
st._config.set_option("theme.backgroundColor", "#0e0f12") # app bg
st._config.set_option("theme.secondaryBackgroundColor", "#161a1f")
st._config.set_option("theme.textColor", "#e7e9ed")
except Exception:
pass
# ================= 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 {}
# UI vars
VIDEO_MAX_PX_DEFAULT = int(os.getenv("VIDEO_MAX_PX", "720"))
VIDEO_RES_OPTIONS = ["512P", "768P"]
VIDEO_DUR_OPTIONS = ["6", "10"]
ASPECT_OPTIONS = ["9:16", "3:4", "1:1"] # preview kadraj oranları
# ================= Session =================
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 "video_ui_max_px" not in st.session_state:
st.session_state["video_ui_max_px"] = VIDEO_MAX_PX_DEFAULT
# ================= HTTP/Backend helpers =================
def _needs_auth(url: str) -> bool:
return url.startswith(API_BASE) or url.startswith("outputs/") or url.startswith("/outputs/")
def _abs_backend_url(url_or_path: str) -> str:
if not url_or_path:
return ""
if url_or_path.startswith(("http://","https://")):
return url_or_path
if url_or_path.startswith("/outputs/"):
return f"{API_BASE}{url_or_path}"
if url_or_path.startswith("outputs/"):
return f"{API_BASE}/{url_or_path}"
return f"{API_BASE}/{url_or_path.lstrip('/')}"
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):
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, preview_aspect="9:16"):
payload = {
"image_url": image_url,
"duration": duration,
"resolution": resolution,
"prompt_optimizer": "true" if prompt_optimizer else "false",
"preview_aspect": preview_aspect,
}
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 = _abs_backend_url(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
def _sniff_mime_from_bytes(b: bytes) -> str:
try:
fmt = (Image.open(io.BytesIO(b)).format or "JPEG").lower()
return {"jpeg":"image/jpeg","jpg":"image/jpeg","png":"image/png","webp":"image/webp"}.get(fmt,"image/jpeg")
except Exception:
return "image/jpeg"
def image_bytes_to_data_url(b: bytes, mime: str | None = None) -> str:
if not mime:
mime = _sniff_mime_from_bytes(b)
enc = base64.b64encode(b).decode("ascii")
return f"data:{mime};base64,{enc}"
@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:
if data[:4] == b"\x00\x00\x00\x18" or data[4:8] == b"ftyp":
return ".mp4"
return ".bin"
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()
# ================= Preview helpers (uniform cards) =================
def _parse_aspect_str(s: str):
try:
a, b = s.split(":"); return max(1, int(a)), max(1, int(b))
except Exception:
return (9, 16)
def inject_base_css():
st.markdown("""
<style>
/* App-wide */
.stApp { background: #0e0f12; }
header[data-testid="stHeader"] { background: transparent; }
.block-container { padding-top: 1.25rem; }
/* Title + caption */
h1, .stMarkdown h1 { font-size: 1.85rem; line-height: 1.3; margin-bottom: 0.15rem; }
.stMarkdown p { margin-top: 0.15rem; }
/* Labels to dark-orange */
label, .stTextInput label, .stSelectbox label, .stNumberInput label, .stFileUploader label,
.stCheckbox label, .stSlider label, .stRadio label {
color: #ff8a33 !important;
font-weight: 600 !important;
}
/* Primary buttons (also Run 💎) */
.stButton > button[kind="primary"]{
background: #ff7a1a !important; border: 0 !important; color: #111 !important;
}
.stButton > button:hover { filter: brightness(1.05); }
/* Info/Warning boxes tweak */
.stAlert { border-radius: 10px; }
/* Card titles */
.lb-card-title {
font-weight: 700; font-size: 1.05rem; margin: 2px 0 8px 0; color: #eaecef;
}
/* Frames: dark background to avoid white band over title */
.lb-frame {
position: relative; width: 100%;
border-radius: 12px; overflow: hidden;
background: #111419;
display:flex; align-items:center; justify-content:center;
box-shadow: 0 0 0 1px rgba(255,122,26,0.15) inset;
}
.lb-frame img { width:100%; height:100%; display:block; }
/* Video styling */
.lb-video { display:flex; justify-content:center; }
.lb-video video { border-radius:8px; max-width: 540px; width:100%; height:auto; }
/* Orange accents for captions */
.stCaption, .st-emotion-cache-1r6slb0 { color: #ffb07a !important; }
</style>
""", unsafe_allow_html=True)
def inject_preview_css(aspect: str, fit_mode="contain"):
aw, ah = _parse_aspect_str(aspect)
st.markdown(f"""
<style>
.lb-frame {{ aspect-ratio: {aw} / {ah}; }}
.lb-frame img {{ object-fit:{fit_mode}; }}
</style>
""", unsafe_allow_html=True)
def show_tile_card_display_download(
col, title,
display_url: str | None = None,
display_bytes: bytes | None = None,
download_url: str | None = None,
download_bytes: bytes | None = None,
filename_stub: str = "image",
key_prefix: str = "",
):
"""Gösterimde HAFİF JPEG/PNG (display_bytes) tercih edilir; indirmede NATIVE."""
col.markdown(f'<div class="lb-card-title">{title}</div>', unsafe_allow_html=True)
# ---- Display ----
if not display_url and not display_bytes:
col.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True)
col.caption("—"); return None, None
if display_bytes is not None:
src = image_bytes_to_data_url(display_bytes) # mime otomatik
col.markdown(f'<div class="lb-frame"><img src="{src}"/></div>', unsafe_allow_html=True)
else:
abs_url = _abs_backend_url(display_url)
col.markdown(f'<div class="lb-frame"><img src="{abs_url}"/></div>', unsafe_allow_html=True)
# ---- Native (caption + download) ----
native = download_bytes or (fetch_bytes(download_url) if download_url else None)
if not native:
native = display_bytes # en azından çözünürlük gösterebilmek için
if native:
sz = image_size_from_bytes(native)
if sz: col.caption(f"Real Res: {sz[0]}×{sz[1]} px")
ext = infer_ext(native)
mime = "image/jpeg" if ext.lower() in [".jpg",".jpeg"] else ("image/png" if ext.lower()==".png" else "image/webp")
col.download_button("Download (Full Quality)", data=native,
file_name=f"{filename_stub}{ext if ext!='.bin' else '.jpg'}",
mime=mime, key=f"{key_prefix}_{filename_stub}_dl")
else:
col.caption("—")
return display_url or "data", native
def rgba_to_rgb_on_bg(img_rgba: Image.Image, bg_rgb=(255,255,255)) -> Image.Image:
"""RGBA görseli seçilen arka plana kompoze ederek RGB'ye çevir (siyah sorununu çözer)."""
if img_rgba.mode != "RGBA":
return img_rgba.convert("RGB")
bg = Image.new("RGBA", img_rgba.size, (*bg_rgb, 255))
bg.paste(img_rgba, (0,0), img_rgba)
return bg.convert("RGB")
@st.cache_data(ttl=300, show_spinner=False)
def make_small_jpeg_from_native(native: bytes, target_long_edge: int = 1280, bg_rgb=(255,255,255)) -> bytes:
"""Şeffaf PNG → beyaz zemine kompoze → hafif JPEG (piksel piksel yüklenmeyi azaltır)."""
im = Image.open(io.BytesIO(native))
if im.mode != "RGBA" and "transparency" not in im.info:
im = im.convert("RGB")
else:
im = im.convert("RGBA")
w, h = im.size
if w >= h:
new_w = min(target_long_edge, w)
new_h = max(1, int(h * (new_w / max(1, w))))
else:
new_h = min(target_long_edge, h)
new_w = max(1, int(w * (new_h / max(1, h))))
im2 = im.resize((new_w, new_h), Image.LANCZOS)
im2_rgb = rgba_to_rgb_on_bg(im2, bg_rgb=bg_rgb)
out = io.BytesIO(); im2_rgb.save(out, format="JPEG", quality=88, subsampling=1)
return out.getvalue()
def make_preview_frame(img_bytes: bytes, aspect: str, long_edge: int, ratio: float, bg_rgb=(255,255,255)) -> bytes:
"""UI-only: seçilen aspect’te tuval + içerik oranı (JPEG)."""
im = Image.open(io.BytesIO(img_bytes)).convert("RGBA")
aw, ah = _parse_aspect_str(aspect)
if aw >= ah:
W = max(256, min(8192, int(long_edge))); H = max(1, int(W * ah / aw))
else:
H = max(256, min(8192, int(long_edge))); W = max(1, int(H * aw / ah))
ratio = max(0.10, min(0.99, float(ratio)))
target = int(min(W, H) * ratio if aw == ah else (H * ratio if aw < ah else W * ratio))
w, h = im.size
if w >= h: new_w, new_h = target, max(1, int(h * (target / max(1, w))))
else: new_h, new_w = target, max(1, int(w * (target / max(1, h))))
im_resized = im.resize((new_w, new_h), Image.LANCZOS)
canvas = Image.new("RGBA", (W, H), (*bg_rgb, 255))
off = ((W - new_w)//2, (H - new_h)//2); canvas.paste(im_resized, off, im_resized)
out = io.BytesIO(); canvas.convert("RGB").save(out, format="JPEG", quality=90, subsampling=1)
return out.getvalue()
# ================= Yerel DEMO dosyaları =================
def load_local_demo():
demo = {}
for fname in ["input.jpg", "packshot.jpg", "model.jpg", "tryon.mp4"]:
if os.path.exists(fname):
with open(fname, "rb") as f: demo[fname] = f.read()
return demo
demo_files = load_local_demo()
# ================= Sayfa =================
st.set_page_config(page_title="AI LightBox · Jewelry", layout="wide", page_icon="💎")
inject_base_css()
st.title("💎 AI LightBox · Jewelry Studio")
st.caption("Upload · Try-on · Download")
ok = backend_ok()
# -------- Üst Kontrol Şeridi --------
col1, col2, col3, col4, col5, col6, col7 = st.columns([1.6, 1.6, 1.0, 1.2, 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 URL", key="image_url", placeholder="https://... (optional)")
with col3:
jewel_type = st.selectbox("Type", ["auto","earring","ring","necklace","bracelet","pendant","brooch","set","watch"], index=0, key="jewel_type")
with col4:
preview_ratio = st.number_input("Preview Scale (0.10–0.99)", min_value=0.10, max_value=0.99, value=0.50, step=0.01,
help="Sadece önizleme kadrajında büyüklüğü etkiler.", key="preview_ratio")
with col5:
upscale = st.checkbox("Super Res", False, key="upscale")
with col6:
upscale_stage = st.selectbox("SR Step", ["both","final","packshot"], index=0, help="Varsayılan: both", key="upscale_stage")
with col7:
upscale_factor = st.selectbox("SR Factor", ["2","4"], index=0, key="upscale_factor")
colA, colB, colC, colD, colE, colF = st.columns([1.0, 1.0, 1.1, 0.9, 0.9, 1.2])
with colA:
st.info(f"Connection: {'Online' if ok else 'Offline'}", icon="🔌")
with colB:
identity_lock = st.checkbox("Identity lock", True, key="identity_lock")
with colC:
to_video = st.checkbox("One Click Generate Video", False, key="to_video")
with colD:
duration = st.selectbox("Duration ", VIDEO_DUR_OPTIONS, index=0, key="duration_sel")
with colE:
resolution = st.selectbox("Video Res", VIDEO_RES_OPTIONS, index=1, key="resolution_sel")
with colF:
run = st.button("Run 💎", type="primary", use_container_width=True, key="run_btn")
# Tuval/kadraj + fit/fill
colT1, colT2, colT3 = st.columns([1.0, 1.0, 1.0])
with colT1:
preview_aspect = st.selectbox("Preview Frame Ratio", ASPECT_OPTIONS, index=0, key="preview_aspect")
with colT2:
preview_long_edge = st.number_input("Preview Long Edge (px)", min_value=256, max_value=8192, value=1920, step=64, key="preview_long_edge")
with colT3:
fit_mode = st.selectbox("Preview Layout", ["fit","fill"], index=0, help="Sadece önizlemeyi etkiler.", key="fit_mode")
# Inject CSS (uniform kartlar)
inject_preview_css(preview_aspect, fit_mode="contain" if fit_mode=="fit" else "cover")
# ================= PRE-RUN PREVIEWS =================
st.subheader("Previews")
col_in, col_pack, col_pad, col_vid = st.columns(4)
# Input bytes (upload/URL öncelikli; yoksa demo)
input_bytes = None
if file_list:
try: input_bytes = file_list[0].getvalue()
except Exception: input_bytes = None
elif image_url:
input_bytes = fetch_bytes(image_url)
elif demo_files.get("input.jpg"):
input_bytes = demo_files["input.jpg"]
show_tile_card_display_download(
col_in, "Sample Input preview",
display_bytes=input_bytes if input_bytes else demo_files.get("input.jpg"),
download_bytes=input_bytes if input_bytes else demo_files.get("input.jpg"),
filename_stub="input", key_prefix="pre"
)
# Packshot preview (demo dosyası varsa; yoksa input görüntü)
pack_bytes_demo = demo_files.get("packshot.jpg") or input_bytes
show_tile_card_display_download(
col_pack, "Sample Packshot (BG Removed)",
display_bytes=pack_bytes_demo,
download_bytes=pack_bytes_demo,
filename_stub="packshot", key_prefix="pre"
)
# Padded preview (packshot.jpg veya input.jpg → seçilen kadraja, UI-only)
padded_bytes = None
if pack_bytes_demo:
try:
padded_bytes = make_preview_frame(pack_bytes_demo, aspect=preview_aspect,
long_edge=int(preview_long_edge),
ratio=float(preview_ratio))
except Exception:
padded_bytes = None
show_tile_card_display_download(
col_pad, "Sample Padded Preview ",
display_bytes=padded_bytes or pack_bytes_demo,
download_bytes=padded_bytes or pack_bytes_demo,
filename_stub="padded_ui", key_prefix="pre"
)
# Video preview (loop; yerel tryon.mp4 varsa)
vid_demo = demo_files.get("tryon.mp4")
col_vid.markdown('<div class="lb-card-title">AI Generated Video Preview </div>', unsafe_allow_html=True)
if vid_demo:
data_url = "data:video/mp4;base64," + base64.b64encode(vid_demo).decode("ascii")
col_vid.markdown(
f"""
<div class="lb-frame" style="display:flex;align-items:center;justify-content:center;">
<video autoplay loop muted playsinline style="width:100%;height:100%;object-fit:cover;">
<source src="{data_url}" type="video/mp4">
</video>
</div>
""", unsafe_allow_html=True
)
col_vid.download_button("Download Video (MP4)", data=vid_demo, file_name="tryon.mp4", mime="video/mp4", key="pre_video_dl")
else:
col_vid.markdown('<div class="lb-frame"></div>', unsafe_allow_html=True)
col_vid.caption("—")
# ================= ÇALIŞTIR =================
if run:
if not ok:
st.error("Backend erişilemiyor. API_BASE / TOKEN kontrol edin.")
elif not (file_list or image_url or demo_files.get("input.jpg")):
st.error("En az bir görsel girişi yapın (dosya yükle veya URL).")
else:
try:
img_urls = image_url or ""
files_payload = []
if file_list:
for f in file_list:
files_payload.append(("files", (f.name, f.getvalue(), f.type or "image/jpeg")))
data = {
"category": st.session_state.get("jewel_type","auto"),
"edit_prompt": "",
"num_images": "1",
"image_urls": img_urls,
"mannequin_image_url": "",
"identity_lock": "true" if identity_lock else "false",
# Preview controls (backend pad/preview için)
"preview_aspect": preview_aspect,
"preview_long_edge": str(int(preview_long_edge)),
"preview_ratio": str(float(preview_ratio)),
"padding_ratio": str(float(preview_ratio)), # backward-compat
# SR
"upscale": "true" if upscale else "false",
"upscale_factor": st.session_state.get("upscale_factor","2"),
"upscale_stage": st.session_state.get("upscale_stage","both"),
# Video (opsiyonel zincir)
"to_video": "true" if to_video else "false",
"video_prompt": "",
"duration": st.session_state.get("duration_sel", VIDEO_DUR_OPTIONS[0]),
"resolution": st.session_state.get("resolution_sel", VIDEO_RES_OPTIONS[1]),
"prompt_optimizer": "false",
}
with st.spinner("AI Chain Running…"):
out = post_chain(data, files_payload if files_payload else None)
st.session_state["job_counter"] += 1
# ---- ÇIKTILAR ----
# Packshot NATIVE → DISPLAY küçük JPEG (şeffaflık beyaza kompoze)
pack_native_url = None; pack_display_jpeg = None
try:
pack_native_url = (out["packshot_step"]["result"]["images_native"] or [{}])[0].get("url")
if pack_native_url:
pn = fetch_bytes(pack_native_url)
if pn: pack_display_jpeg = make_small_jpeg_from_native(pn, 1280, bg_rgb=(255,255,255))
except Exception: pass
# Padded NATIVE → DISPLAY küçük JPEG (pad etkisi görünür)
padded_native_url = None; padded_display_jpeg = None
try:
padded_native_url = (out["packshot_step"]["result"]["padded_native"] or [{}])[0].get("url")
if padded_native_url:
nb = fetch_bytes(padded_native_url)
if nb: padded_display_jpeg = make_small_jpeg_from_native(nb, 1280, bg_rgb=(255,255,255))
except Exception: pass
# Model NATIVE → DISPLAY küçük JPEG
placed_native_url = None; placed_display_jpeg = None
try:
placed_native_url = (out["image_step"]["result"]["images_native"] or [{}])[0].get("url")
if placed_native_url:
mn = fetch_bytes(placed_native_url)
if mn: placed_display_jpeg = make_small_jpeg_from_native(mn, 1280, bg_rgb=(255,255,255))
except Exception: pass
# Video (zincirde üretildiyse)
vurl = None
try:
vurl = (out.get("video_step") or {}).get("result", {}).get("video", {}).get("url")
except Exception:
vurl = None
# UI ipucu
ui_hints = (out.get("video_step") or {}).get("ui_hints") or {}
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.session_state["results"] = {
"job_id": out.get("job_id"),
"schema_version": out.get("schema_version"),
"packshot_native_url": pack_native_url,
"packshot_display_jpeg": pack_display_jpeg,
"padded_native_url": padded_native_url,
"padded_display_jpeg": padded_display_jpeg,
"placed_native_url": placed_native_url,
"placed_display_jpeg": placed_display_jpeg,
"video_url": vurl,
"api_base": API_BASE,
"has_auth_header": bool(HEADERS),
"preview_aspect": preview_aspect,
}
except requests.HTTPError as e:
st.error(f"HTTP {e.response.status_code}")
except Exception as e:
st.error(f"Hata: {e}")
# ================= SONUÇLAR =================
res = st.session_state.get("results")
vbytes = None
if res:
st.subheader("Product Results")
c_pack, c_pad, c_model, c_vid = st.columns(4)
# Packshot (backend) — DISPLAY = küçük JPEG, DOWNLOAD = NATIVE
show_tile_card_display_download(
c_pack, "Packshot (Removed BG)",
display_bytes=res.get("packshot_display_jpeg"),
download_url=res.get("packshot_native_url"),
filename_stub="packshot_native",
key_prefix=f"job{st.session_state['job_counter']}"
)
# Padded (backend) — DISPLAY = padded küçük JPEG, DOWNLOAD = padded NATIVE
show_tile_card_display_download(
c_pad, "Padded",
display_bytes=res.get("padded_display_jpeg"),
download_url=res.get("padded_native_url"),
filename_stub="padded_backend",
key_prefix=f"job{st.session_state['job_counter']}"
)
# Model üzerinde (backend) — DISPLAY = küçük JPEG (native’den), DOWNLOAD = NATIVE
show_tile_card_display_download(
c_model, "TryON Model",
display_bytes=res.get("placed_display_jpeg"),
download_url=res.get("placed_native_url"),
filename_stub="model_native",
key_prefix=f"job{st.session_state['job_counter']}"
)
# Video
vurl = res.get("video_url")
c_vid.markdown('<div class="lb-card-title">AI Generated Video</div>', unsafe_allow_html=True)
if vurl:
abs_v = _abs_backend_url(vurl)
c_vid.markdown(
f"""
<div class="lb-frame" style="display:flex;align-items:center;justify-content:center;">
<video autoplay loop muted controls playsinline style="width:100%;height:100%;object-fit:cover;">
<source src="{abs_v}" type="video/mp4">
</video>
</div>
""", unsafe_allow_html=True
)
vb = fetch_bytes(abs_v)
if vb:
vbytes = vb
c_vid.download_button("Download Video (MP4)", data=vb, file_name="ai_lightbox_video.mp4",
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)
# ---- Generate video on demand ----
# unique key by job to avoid stale checkbox value on rerun
vid_checkbox_key = f"video_toggle_job{st.session_state['job_counter']}"
if st.checkbox("Generate Video From This Result", value=False, key=vid_checkbox_key):
src_for_video = res.get("placed_native_url") or res.get("packshot_native_url")
if not src_for_video:
st.warning("Video için kullanılacak native görsel bulunamadı.")
else:
with st.spinner("Generating Video …"):
try:
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","9:16"),
prompt_optimizer=False,
)
except requests.HTTPError as e:
st.error(f"HTTP {e.response.status_code}")
v = None
if v:
vurl2 = (v.get("result") or {}).get("video", {}).get("url")
ui_hints = v.get("ui_hints") or {}
if vurl2:
# write to session and force rerun so the video appears immediately
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))
# reset checkbox before rerun to prevent re-trigger loop
st.session_state.pop(vid_checkbox_key, None)
st.success("Video is Ready! Showing preview…")
st.rerun()
else:
st.info("Video URL gelmedi.")
# Native ZIP
named = []
pn = fetch_bytes(res.get("packshot_native_url")) if res.get("packshot_native_url") else None
mn = fetch_bytes(res.get("placed_native_url")) if res.get("placed_native_url") else None
if pn: named.append((f"packshot_native{infer_ext(pn)}", pn))
if mn: named.append((f"model_native{infer_ext(mn)}", mn))
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_outputs.zip", mime="application/zip",
key=f"job{st.session_state['job_counter']}_zip_dl")
# ================= Sidebar =================
with st.sidebar:
st.caption(f"API_BASE: {API_BASE}")
if HF_TOKEN: st.caption("Auth: Bearer (aktif)")
st.slider("Video önizleme genişliği (px)", 360, 1080, st.session_state["video_ui_max_px"],
step=10, key="video_ui_max_px")
if st.button("Önbelleği temizle"): st.cache_data.clear(); st.success("Önbellek temizlendi.")
if st.button("Son çıktıları temizle"): st.session_state["results"] = None; st.success("Çıktılar temizlendi.")
|