Upload streamlit_app.py

streamlit_app.py

@@ -8,9 +8,14 @@ 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"))  # sadece UI kare önizleme
 
+# ================= 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  # her çalıştırmada artar; download_button key'leri için
+
 # ================= Yardımcılar =================
 def _needs_auth(url: str) -> bool:
-    # Sadece kendi backend çağrılarında token gönder
     return url.startswith(API_BASE) or url.startswith("outputs/")
 
 def backend_ok() -> bool:
@@ -22,7 +27,6 @@ def backend_ok() -> bool:
         return False
 
 def post_chain(data: dict, files_payload):
-    # Videoyu backend'e bırakmıyoruz (SR final ile biz tetikleriz)
     data = {**data, "to_video": "false"}
     r = requests.post(f"{API_BASE}/v1/tryon/chain", data=data, files=files_payload or None,
                       headers=HEADERS, timeout=600)
@@ -34,8 +38,7 @@ def post_video(image_url: str, duration="6", resolution="768P", prompt_optimizer
         "image_url": image_url,
         "duration": duration,
         "resolution": resolution,
-        "prompt_optimizer": "true" if prompt_optimizer else "false",  # backend default False
-        # prompt vermiyoruz -> backend JEWELRY_VIDEO_PROMPT (zoom yok)
+        "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()
@@ -61,7 +64,7 @@ def fetch_bytes(url_or_path: str):
 def image_size_from_bytes(b: bytes):
     try:
         im = Image.open(io.BytesIO(b))
-        return im.size  # (w, h)
+        return im.size
     except Exception:
         return None
 
@@ -99,7 +102,7 @@ def first_present(d: dict, keys: list, default=None):
             return v
     return default
 
-def show_tile(col, title, url: str | None, bg_rgb=(255,255,255), filename_stub="image"):
+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))
@@ -119,9 +122,14 @@ def show_tile(col, title, url: str | None, bg_rgb=(255,255,255), filename_stub="
         col.caption(f"Gerçek çözünürlük: {sz[0]}×{sz[1]} px")
 
     ext = infer_ext(b)
-    # MIME'ı görüntünün gerçek tipinden bağımsız sabit jpeg vermeyelim:
     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)
+    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:
@@ -143,37 +151,38 @@ ok = backend_ok()
 col1, col2, col3, col4, col5, col6, col7 = st.columns([1.6, 1.6, 1.0, 1.3, 0.9, 1.1, 0.9])
 
 with col1:
-    file_list = st.file_uploader("Takı görseli (yükle)", type=["jpg","jpeg","png","webp"], accept_multiple_files=True)
+    file_list = st.file_uploader("Takı görseli (yükle)", type=["jpg","jpeg","png","webp"], accept_multiple_files=True, key="file_upl")
 with col2:
-    image_url = st.text_input("veya URL")
+    image_url = st.text_input("veya URL", key="image_url")
 with col3:
-    jewel_type = st.selectbox("Tür", ["auto","earring","ring","necklace","bracelet","pendant","brooch","set"], index=0)
+    jewel_type = st.selectbox("Tür", ["auto","earring","ring","necklace","bracelet","pendant","brooch","set"], index=0, key="jewel_type")
 with col4:
     padding_ratio_val = st.number_input(
         "İçerik ölçeği (0.30–0.95)",
         min_value=0.30, max_value=0.95, value=0.50, step=0.01,
-        help="1200×1200 kare tuvalde ürünün uzun kenarı = 1200 × bu oran. 0.50 önerilir."
+        help="1200×1200 kare tuvalde ürünün uzun kenarı = 1200 × bu oran. 0.50 önerilir.",
+        key="padding_ratio"
     )
 with col5:
-    upscale = st.checkbox("Super Resolution", True)
+    upscale = st.checkbox("Super Resolution", True, key="upscale")
 with col6:
-    upscale_stage = st.selectbox("SR aşaması", ["both","final","packshot"], index=0, help="Varsayılan: both")
+    upscale_stage = st.selectbox("SR aşaması", ["both","final","packshot"], index=0, help="Varsayılan: both", key="upscale_stage")
 with col7:
-    upscale_factor = st.selectbox("SR katsayı", ["2","4"], index=0)
+    upscale_factor = st.selectbox("SR katsayı", ["2","4"], index=0, key="upscale_factor")
 
 colA, colB, colC, colD, colE = st.columns([1.0, 1.2, 1.2, 0.9, 1.0])
 with colA:
     st.info(f"Bağlantı: {'Online' if ok else 'Offline'}", icon="🔌")
 with colB:
-    identity_lock = st.checkbox("Identity lock (önerilir)", True)
+    identity_lock = st.checkbox("Identity lock (önerilir)", True, key="identity_lock")
 with colC:
-    to_video = st.checkbox("Video üret (SR kaynaklı)", False)
+    to_video = st.checkbox("Video üret (SR kaynaklı)", False, key="to_video")
 with colD:
-    duration = st.selectbox("Süre (sn)", ["6","10"], index=0)
+    duration = st.selectbox("Süre (sn)", ["6","10"], index=0, key="duration")
 with colE:
-    run = st.button("Çalıştır", type="primary", use_container_width=True)
+    run = st.button("Çalıştır", type="primary", use_container_width=True, key="run_btn")
 
-mannequin_url = st.text_input("Manken URL (opsiyonel)")
+mannequin_url = st.text_input("Manken URL (opsiyonel)", key="mannequin_url")
 
 # ================= Girdi Önizleme =================
 c_in, c_pack, c_pad, c_model = st.columns(4)
@@ -199,7 +208,7 @@ if input_preview_bytes:
 else:
     show_tile(c_in, "Input", None)
 
-# ================ Koş & Göster ================
+# ================ Koş & Sonuçları Kaydet ================
 if run:
     if not ok:
         st.error("Backend erişilemiyor. API_BASE / TOKEN kontrol edin.")
@@ -216,10 +225,9 @@ if run:
                 "mannequin_image_url": (mannequin_url or ""),
                 "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,  # varsayılan both
+                "upscale_stage": upscale_stage,
             }
             files_payload = []
             if file_list:
@@ -229,7 +237,7 @@ if run:
             with st.spinner("Zincir çalışıyor…"):
                 out = post_chain(data, files_payload)
 
-            # ---- Çıktıları topla ----
+            # ---- Çıktı URL'lerini topla ----
             packshot_url = None
             try:
                 imgs = out["packshot_step"]["result"]["images"]
@@ -240,7 +248,6 @@ if run:
 
             padded_url = None
             try:
-                # backward compat
                 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 []
@@ -267,28 +274,8 @@ if run:
             ], default=[]) or []
             placed_sr_url = final_sr_list[0] if final_sr_list else None
 
-            # ---- Görselleri göster + indir (tam kalite) ----
-            p_url, p_bytes = show_tile(
-                c_pack,
-                "Packshot (SR)" if packshot_sr_url else "Packshot",
-                packshot_sr_url or packshot_url,
-                bg_rgb=white_bg,
-                filename_stub="packshot"
-            )
-
-            pad_url, pad_bytes = show_tile(c_pad, "Padded", padded_url, bg_rgb=white_bg, filename_stub="padded")
-
-            shown_final_url = placed_sr_url or placed_url
-            m_url, m_bytes = show_tile(
-                c_model,
-                "Model Üzerinde (SR)" if placed_sr_url else "Model Üzerinde",
-                shown_final_url,
-                bg_rgb=white_bg,
-                filename_stub="model"
-            )
-
-            # ---- Video (daima SR edilmiş final görseliyle) ----
-            vurl, vbytes = None, None
+            # ---- Video (SR edilmiş final görseliyle) ----
+            vurl = None
             if to_video:
                 src_for_video = placed_sr_url or placed_url
                 if not src_for_video:
@@ -297,56 +284,100 @@ if run:
                     with st.spinner("Video üretiliyor (SR kaynak)…"):
                         v = post_video(src_for_video, duration=duration, resolution="768P", prompt_optimizer=False)
                     vurl = (v.get("result") or {}).get("video", {}).get("url")
-                    if vurl:
-                        st.subheader("Video (SR kaynak)")
-                        st.video(vurl, format="video/mp4")
-                        vb = fetch_bytes(vurl)
-                        if vb:
-                            vbytes = vb
-                            st.download_button("Videoyu indir (MP4)", data=vb, file_name="ai_lightbox_video.mp4", mime="video/mp4")
-                    else:
-                        st.info("Video URL gelmedi.")
-
-            # ---- Hepsini ZIP indir ----
-            named = []
-            if p_bytes:
-                ext = infer_ext(p_bytes)
-                named.append((f"packshot{ext}", p_bytes))
-            if pad_bytes:
-                ext = infer_ext(pad_bytes)
-                named.append((f"padded{ext}", pad_bytes))
-            if m_bytes:
-                ext = infer_ext(m_bytes)
-                named.append((f"model{ext}", m_bytes))
-            if vbytes:
-                named.append(("video.mp4", vbytes))
-            if named:
-                zip_bytes = make_zip(named)
-                st.download_button("Hepsini ZIP indir", data=zip_bytes, file_name="ai_lightbox_outputs.zip", mime="application/zip")
-
-            # ---- Debug ----
-            with st.expander("Debug / Kaynak Alanlar"):
-                st.json({
-                    "api_base": API_BASE,
-                    "has_auth_header": bool(HEADERS),
-                    "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": vurl,
-                })
+
+            # ---- Sonuçları session_state'e yaz ----
+            st.session_state["job_counter"] += 1
+            st.session_state["results"] = {
+                "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": vurl,
+                "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}")
 
-# Küçük yardımcı: cache temizleme
+# ================ Sonuçları Göster (kalıcı) ================
+res = st.session_state.get("results")
+if res:
+    key_prefix = f"job{st.session_state['job_counter']}"
+    # Görsel sütunları: Input zaten solda
+    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 = "Model Üzerinde (SR)" if res.get("placed_sr_url") else "Model Üzerinde"
+    m_url, m_bytes = show_tile(
+        c_model, m_title, shown_final_url,
+        bg_rgb=white_bg, filename_stub="model", key_prefix=key_prefix
+    )
+
+    # Video
+    vurl = res.get("video_url")
+    vbytes = None
+    if vurl:
+        st.subheader("Video (SR kaynak)")
+        st.video(vurl, format="video/mp4")
+        vb = fetch_bytes(vurl)
+        if vb:
+            vbytes = vb
+            st.download_button(
+                "Videoyu indir (MP4)",
+                data=vb,
+                file_name="ai_lightbox_video.mp4",
+                mime="video/mp4",
+                key=f"{key_prefix}_video_dl"
+            )
+        else:
+            st.info("Video URL bulundu ama içerik indirilemedi.")
+
+    # Hepsini ZIP
+    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"model{infer_ext(m_bytes)}", m_bytes))
+    if vbytes: named.append(("video.mp4", vbytes))
+    if named:
+        zip_bytes = make_zip(named)
+        st.download_button(
+            "Hepsini ZIP indir",
+            data=zip_bytes,
+            file_name="ai_lightbox_outputs.zip",
+            mime="application/zip",
+            key=f"{key_prefix}_zip_dl"
+        )
+
+    with st.expander("Debug / Kaynak Alanlar"):
+        st.json({
+            **{k: res.get(k) for k in [
+                "api_base","has_auth_header",
+                "packshot_url","packshot_sr_url",
+                "padded_url","placed_url","placed_sr_url","video_url"
+            ]}
+        })
+
+# ================= Sidebar =================
 with st.sidebar:
     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.")
     st.caption(f"API_BASE: {API_BASE}")
     if HF_TOKEN:
-        st.caption("Auth: Bearer (aktif)")
+        st.caption("Auth: Bearer (aktif)")