Upload streamlit_app.py

streamlit_app.py

@@ -2,28 +2,9 @@
 import os, io, zipfile, requests, streamlit as st
 from PIL import Image
 
-# ================= Env & API (robust) =================
-def _env(k): 
-    return (os.getenv(k) or "").strip().strip("'\"")
-
-# Desteklenen env isimleri (öncelik sırası):
-# AI_BEAUTYBOX_API / AI_LIGHTBOX_API / BeautyBoxAI_API
-API_BASE = (
-    _env("AI_BEAUTYBOX_API") or
-    _env("AI_LIGHTBOX_API")  or
-    _env("BeautyBoxAI_API")
-).rstrip("/")
-
-# Fallback'ı kaldırıyoruz; yanlışlıkla 127.0.0.1'e gitmesin.
-if not API_BASE:
-    st.error("API_BASE yok. Settings > Variables: AI_BEAUTYBOX_API veya AI_LIGHTBOX_API tanımla ve Space'i yeniden başlat.")
-    st.stop()
-
-HF_TOKEN = (
-    _env("AI_BEAUTYBOX_TOKEN") or 
-    _env("AI_LIGHTBOX_TOKEN")  or 
-    _env("BeautyBoxAI_TOKEN")
-)
+# ================= 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 {}
 PREVIEW_SIZE = int(os.getenv("PREVIEW_SIZE", "512"))
 
@@ -35,14 +16,13 @@ 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 =================
+# ================= Yardımcılar ================= 
 def _needs_auth(url: str) -> bool:
     return url.startswith(API_BASE) or url.startswith("outputs/")
 
 def backend_ok() -> bool:
     try:
-        # Health check'te header göndermiyoruz (bazı proxy'ler 401/403 verebiliyor)
-        r = requests.get(f"{API_BASE}/health", timeout=10)
+        r = requests.get(f"{API_BASE}/health", headers=HEADERS, timeout=10)
         r.raise_for_status()
         return True
     except Exception:
@@ -51,7 +31,7 @@ def backend_ok() -> bool:
 def post_chain(data: dict, files_payload):
     data = {**data, "to_video": "false"}  # videoyu ayrı tetikleriz
     r = requests.post(f"{API_BASE}/v1/tryon/chain", data=data, files=files_payload or None,
-                      headers=HEADERS if HF_TOKEN else None, timeout=600)
+                      headers=HEADERS, timeout=600)
     r.raise_for_status()
     return r.json()
 
@@ -62,8 +42,7 @@ def post_video(image_url: str, duration="6", resolution="768P", prompt_optimizer
         "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 = requests.post(f"{API_BASE}/v1/video/from-image", data=payload, headers=HEADERS, timeout=600)
     r.raise_for_status()
     return r.json()
 
@@ -73,7 +52,7 @@ def fetch_bytes(url_or_path: str):
         return None
     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
+        headers = HEADERS if _needs_auth(url) else None
         r = requests.get(url, headers=headers, timeout=180)
         r.raise_for_status()
         return r.content
@@ -429,18 +408,14 @@ if res:
             "video_url": res.get("video_url"),
         })
 
-# ================= Sidebar (teşhis) =================
+# ================= Sidebar =================
 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", timeout=10)  # headers yok
-            st.write(r.status_code)
-            st.json(r.json())
-        except Exception as e:
-            st.error(f"Health error: {e}")
     if st.button("Clear cache"):
-        st.cache_data.clear(); st.success("Cache cleared.")
+        st.cache_data.clear()
+        st.success("Cache cleared.")
     if st.button("Clear results"):
-        st.session_state["results"] = None; st.success("Results cleared.")
+        st.session_state["results"] = None
+        st.success("Results cleared.")
+    st.caption(f"API_BASE: {API_BASE}")
+    if HF_TOKEN:
+        st.caption("Auth: Bearer (active)")