Update app.py

app.py

@@ -11,7 +11,6 @@ from huggingface_hub import login
 SPACE = "yisol/IDM-VTON"
 API_NAME = "/tryon"
 
-# ✅ IMPORTANT: do NOT hardcode tokens in a Space repo
 HF_TOKEN = os.getenv("HF_TOKEN", "")
 
 print("HF_TOKEN set:", bool(HF_TOKEN), "len:", len(HF_TOKEN) if HF_TOKEN else 0)
@@ -34,10 +33,6 @@ def reset_client():
 
 
 def get_client():
-    """
-    Compatible with older gradio_client (no hf_token=...).
-    We rely on huggingface_hub login() above.
-    """
     global _client
     if _client is None:
         _client = Client(SPACE)
@@ -52,29 +47,11 @@ def clamp_int(x, lo, hi):
     return max(lo, min(hi, x))
 
 
-def pil_save_temp(
-    pil_img: Image.Image,
-    suffix: str = ".jpg",
-    max_side: int = 1600,
-    quality: int = 92
-) -> str:
-    """
-    Make uploads more stable (reduce write timeouts):
-    - convert to RGB
-    - downscale to max_side
-    - save optimized JPEG
-    """
-    img = pil_img.convert("RGB")
-    w, h = img.size
-    m = max(w, h)
-    if m > max_side:
-        scale = max_side / m
-        img = img.resize((int(w * scale), int(h * scale)), Image.LANCZOS)
-
+def save_pil_temp(pil_img: Image.Image, suffix: str = ".png") -> str:
     f = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
     path = f.name
     f.close()
-    img.save(path, format="JPEG", quality=int(quality), optimize=True)
+    pil_img.save(path, format="PNG")  # без ресайза и компрессии
     return path
 
 
@@ -87,14 +64,12 @@ def tryon_remote(person_pil, garment_pil, garment_desc, auto_mask, crop_center,
     denoise_steps = clamp_int(denoise_steps, 10, 40)
     seed = clamp_int(seed, 0, 999999)
 
-    # Save temp files (compressed for stability)
-    p_path = pil_save_temp(person_pil, ".jpg", max_side=1600, quality=92)
-    g_path = pil_save_temp(garment_pil, ".jpg", max_side=1600, quality=92)
+    p_path = save_pil_temp(person_pil)
+    g_path = save_pil_temp(garment_pil)
 
     try:
         last_err = None
 
-        # retries to survive ZeroGPU overload/queue/timeouts
         for attempt in range(1, 7):
             try:
                 client = get_client()
@@ -102,7 +77,7 @@ def tryon_remote(person_pil, garment_pil, garment_desc, auto_mask, crop_center,
                 result = client.predict(
                     dict={"background": handle_file(p_path), "layers": [], "composite": None},
                     garm_img=handle_file(g_path),
-                    garment_des=(garment_desc or "a t-shirt"),
+                    garment_des=garment_desc,  # ← без дефолта
                     is_checked=bool(auto_mask),
                     is_checked_crop=bool(crop_center),
                     denoise_steps=int(denoise_steps),
@@ -120,12 +95,24 @@ def tryon_remote(person_pil, garment_pil, garment_desc, auto_mask, crop_center,
                 last_err = e
                 msg = str(e).lower()
 
-                is_timeout = ("write operation timed out" in msg) or ("read operation timed out" in msg) or ("timed out" in msg)
-                is_busy = ("too many requests" in msg) or ("queue" in msg) or ("too busy" in msg) or ("overloaded" in msg) or ("capacity" in msg) or ("zerogpu" in msg)
+                is_timeout = (
+                    "write operation timed out" in msg
+                    or "read operation timed out" in msg
+                    or "timed out" in msg
+                )
+
+                is_busy = (
+                    "too many requests" in msg
+                    or "queue" in msg
+                    or "too busy" in msg
+                    or "overloaded" in msg
+                    or "capacity" in msg
+                    or "zerogpu" in msg
+                )
 
                 if is_timeout or is_busy:
                     reset_client()
-                    time.sleep(6.0 * attempt)  # 6s, 12s, 18s, ...
+                    time.sleep(6.0 * attempt)
                     continue
 
                 time.sleep(1.2 * attempt)
@@ -161,7 +148,7 @@ with gr.Blocks(title="Virtual Try-On Rendez-vous") as demo:
 
             garment_desc = gr.Textbox(
                 label="Описание одежды",
-                value="a t-shirt"
+                value=""  # ← пусто, без дефолта
             )
 
             with gr.Accordion("Настройки", open=False):