Update app.py

app.py

@@ -4,6 +4,7 @@ import time
 import tempfile
 
 import gradio as gr
+import spaces  # ✅ required for ZeroGPU (needs at least one @spaces.GPU function)
 from PIL import Image
 from gradio_client import Client, handle_file
 from huggingface_hub import login
@@ -35,7 +36,11 @@ def reset_client():
 def get_client():
     global _client
     if _client is None:
-        _client = Client(SPACE)
+        # HF_TOKEN может помочь при приватных спейсах / ограничениях
+        if HF_TOKEN:
+            _client = Client(SPACE, hf_token=HF_TOKEN)
+        else:
+            _client = Client(SPACE)
     return _client
 
 
@@ -55,6 +60,10 @@ def save_pil_temp(pil_img: Image.Image, suffix: str = ".png") -> str:
     return path
 
 
+# ✅ ZeroGPU: должна быть хотя бы одна функция, помеченная @spaces.GPU
+# ВАЖНО: в твоём текущем решении GPU локально не используется (ты вызываешь удалённый Space),
+# но декоратор нужен, чтобы ZeroGPU runtime не падал на старте.
+@spaces.GPU(duration=180)
 def tryon_remote(person_pil, garment_pil, garment_desc, auto_mask, crop_center, denoise_steps, seed):
     if person_pil is None:
         return None, "❌ Загрузите фото человека"
@@ -64,6 +73,11 @@ 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)
 
+    # если описание пустое — дадим нейтральный дефолт (повышает стабильность)
+    garment_desc = (garment_desc or "").strip()
+    if not garment_desc:
+        garment_desc = "a photo of a garment"
+
     p_path = save_pil_temp(person_pil)
     g_path = save_pil_temp(garment_pil)
 
@@ -77,7 +91,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,  # ← без дефолта
+                    garment_des=garment_desc,
                     is_checked=bool(auto_mask),
                     is_checked_crop=bool(crop_center),
                     denoise_steps=int(denoise_steps),
@@ -88,6 +102,8 @@ def tryon_remote(person_pil, garment_pil, garment_desc, auto_mask, crop_center,
                 if isinstance(result, (list, tuple)):
                     result = result[0]
 
+                # result обычно путь к файлу (temp) или URL — Image.open умеет путь,
+                # но если придёт нечто странное, поймаем исключение ниже.
                 out = Image.open(result).convert("RGB")
                 return out, f"✅ Готово (steps={denoise_steps}, seed={seed}, crop={crop_center})"
 
@@ -110,9 +126,12 @@ def tryon_remote(person_pil, garment_pil, garment_desc, auto_mask, crop_center,
                     or "zerogpu" in msg
                 )
 
-                if is_timeout or is_busy:
+                # ✅ ключевая правка под ZeroGPU-прокси
+                is_expired = "expired zerogpu proxy token" in msg or "zerogpu proxy token" in msg
+
+                if is_timeout or is_busy or is_expired:
                     reset_client()
-                    time.sleep(6.0 * attempt)
+                    time.sleep(4.0 * attempt)
                     continue
 
                 time.sleep(1.2 * attempt)
@@ -128,7 +147,7 @@ def tryon_remote(person_pil, garment_pil, garment_desc, auto_mask, crop_center,
 
 
 def reset_ui():
-    return None, None, None, "Ожидание..."
+    return None, None, "", "Ожидание...", None
 
 
 CUSTOM_CSS = """
@@ -138,7 +157,8 @@ div[class*="footer"] {display:none !important;}
 button[aria-label="Settings"] {display:none !important;}
 """
 
-with gr.Blocks(title="Virtual Try-On Rendez-vous") as demo:
+# ✅ CSS правильно задаём на уровне Blocks
+with gr.Blocks(title="Virtual Try-On Rendez-vous", css=CUSTOM_CSS) as demo:
     gr.Markdown("# Virtual Try-On Rendez-vous")
 
     with gr.Row():
@@ -148,7 +168,7 @@ with gr.Blocks(title="Virtual Try-On Rendez-vous") as demo:
 
             garment_desc = gr.Textbox(
                 label="Описание одежды",
-                value=""  # ← пусто, без дефолта
+                value=""
             )
 
             with gr.Accordion("Настройки", open=False):
@@ -175,7 +195,7 @@ with gr.Blocks(title="Virtual Try-On Rendez-vous") as demo:
     reset.click(
         fn=reset_ui,
         inputs=[],
-        outputs=[person, garment, out, status]
+        outputs=[person, garment, garment_desc, status, out]
     )
 
 if __name__ == "__main__":
@@ -184,6 +204,5 @@ if __name__ == "__main__":
         server_port=7860,
         share=False,
         debug=False,
-        css=CUSTOM_CSS,
         ssr_mode=False
     )