Upload app.py

app.py

@@ -0,0 +1,252 @@
+# -*- coding: utf-8 -*-
+import os
+import time
+import tempfile
+import gradio as gr
+from PIL import Image, ImageFilter, ImageEnhance
+from gradio_client import Client, handle_file
+from huggingface_hub import login
+
+SPACE = "yisol/IDM-VTON"
+API_NAME = "/tryon"
+
+HF_TOKEN = "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"
+
+print("HF_TOKEN set:", bool(HF_TOKEN), "len:", len(HF_TOKEN) if HF_TOKEN else 0)
+
+if HF_TOKEN and HF_TOKEN != "PASTE_YOUR_HF_TOKEN_HERE":
+    try:
+        login(token=HF_TOKEN, add_to_git_credential=False)
+        print("HF login: OK")
+    except Exception as e:
+        print("HF login: FAILED:", str(e)[:200])
+else:
+    print("HF login: skipped (token placeholder or empty)")
+
+_client = None
+
+
+def get_client():
+    global _client
+    if _client is None:
+        _client = Client(SPACE)
+    return _client
+
+
+def pil_save_temp(pil_img: Image.Image, suffix: str) -> str:
+    f = tempfile.NamedTemporaryFile(delete=False, suffix=suffix)
+    path = f.name
+    f.close()
+    pil_img.save(path)
+    return path
+
+
+def clamp_int(x, lo, hi):
+    try:
+        x = int(x)
+    except Exception:
+        x = lo
+    return max(lo, min(hi, x))
+
+
+def postprocess_text_boost(img: Image.Image, boost: int) -> Image.Image:
+    boost = clamp_int(boost, 0, 100)
+    if boost <= 0:
+        return img
+
+    t = boost / 100.0
+    out = img.convert("RGB")
+
+    out = out.filter(
+        ImageFilter.UnsharpMask(
+            radius=1.2 + 1.0 * t,
+            percent=int(120 + 80 * t),
+            threshold=2
+        )
+    )
+    out = ImageEnhance.Contrast(out).enhance(1.0 + 0.10 * t)
+    out = ImageEnhance.Sharpness(out).enhance(1.0 + 0.30 * t)
+
+    w, h = out.size
+    box = (int(w * 0.25), int(h * 0.22), int(w * 0.75), int(h * 0.72))
+
+    region = out.crop(box)
+    region = region.filter(
+        ImageFilter.UnsharpMask(
+            radius=1.8 + 1.6 * t,
+            percent=int(160 + 120 * t),
+            threshold=2
+        )
+    )
+    region = ImageEnhance.Contrast(region).enhance(1.0 + 0.15 * t)
+    region = ImageEnhance.Sharpness(region).enhance(1.0 + 0.45 * t)
+
+    blended = Image.blend(out.crop(box), region, alpha=min(0.55 + 0.35 * t, 0.9))
+    out.paste(blended, box)
+    return out
+
+
+def apply_mode_defaults(mode: str, denoise_steps: int, seed: int, crop_center: bool, garment_desc: str):
+    denoise_steps = clamp_int(denoise_steps, 10, 40)
+    seed = clamp_int(seed, 0, 999999)
+
+    if mode == "Balanced":
+        denoise_steps = max(denoise_steps, 28)
+    elif mode == "Quality":
+        denoise_steps = max(denoise_steps, 38)
+    elif mode == "Text/Logo":
+        denoise_steps = 40
+        crop_center = False
+        if seed == 0:
+            seed = 42
+        if garment_desc and "text" not in garment_desc.lower() and "logo" not in garment_desc.lower():
+            garment_desc = garment_desc.strip() + ", with clear text/logo print"
+
+    return denoise_steps, seed, crop_center, garment_desc
+
+
+def tryon_remote(
+    person_pil,
+    garment_pil,
+    garment_desc,
+    auto_mask,
+    crop_center,
+    denoise_steps,
+    seed,
+    mode,
+    text_boost,
+):
+    if person_pil is None:
+        return None, "❌ Загрузите фото человека"
+    if garment_pil is None:
+        return None, "❌ Загрузите одежду"
+
+    denoise_steps, seed, crop_center, garment_desc = apply_mode_defaults(
+        mode=mode,
+        denoise_steps=denoise_steps,
+        seed=seed,
+        crop_center=crop_center,
+        garment_desc=garment_desc or "a t-shirt"
+    )
+
+    p_path = pil_save_temp(person_pil.convert("RGB"), ".png")
+    g_path = pil_save_temp(garment_pil.convert("RGB"), ".png")
+
+    try:
+        client = get_client()
+
+        last_err = None
+        for attempt in range(1, 4):
+            try:
+                result = client.predict(
+                    dict={
+                        "background": handle_file(p_path),
+                        "layers": [],
+                        "composite": None
+                    },
+                    garm_img=handle_file(g_path),
+                    garment_des=garment_desc,
+                    is_checked=bool(auto_mask),
+                    is_checked_crop=bool(crop_center),
+                    denoise_steps=int(denoise_steps),
+                    seed=int(seed),
+                    api_name=API_NAME
+                )
+
+                if isinstance(result, (list, tuple)):
+                    result = result[0]
+
+                out = Image.open(result).convert("RGB")
+                out = postprocess_text_boost(out, text_boost)
+
+                return out, f"✅ Готово (mode={mode}, steps={denoise_steps}, seed={seed}, crop={crop_center})"
+
+            except Exception as e:
+                last_err = e
+                msg = str(e)
+
+                if "ZeroGPU" in msg or "Unlogged user" in msg or "quota" in msg.lower():
+                    return None, (
+                        "❌ ZeroGPU-квота/нагрузка Hugging Face.\n"
+                        "Если HF login = OK, то это лимит/очередь Space. Попробуйте позже или смените seed."
+                    )
+
+                time.sleep(1.2 * attempt)
+
+        return None, f"❌ Ошибка Space: {str(last_err)[:250]}"
+
+    finally:
+        for path in (p_path, g_path):
+            try:
+                os.remove(path)
+            except Exception:
+                pass
+
+
+def reset_ui():
+    return None, None, None, "Ожидание..."
+
+
+CUSTOM_CSS = """
+footer {display:none !important;}
+#api-info {display:none !important;}
+div[class*="footer"] {display:none !important;}
+button[aria-label="Settings"] {display:none !important;}
+"""
+
+with gr.Blocks(title="Virtual Try-On Rendez-vous", css=CUSTOM_CSS) as demo:
+    gr.Markdown(
+        "# Virtual Try-On Rendez-vous\n\n"
+        "**Для текста/логотипов:** режим *Text/Logo*, steps=40, попробуйте разные *seed*."
+    )
+
+    with gr.Row():
+        with gr.Column():
+            person = gr.Image(label="Фото человека", type="pil", height=420)
+            garment = gr.Image(label="Одежда", type="pil", height=320)
+
+            garment_desc = gr.Textbox(
+                label="Описание одежды (чем точнее — тем лучше)",
+                value="a t-shirt with clear text/logo print"
+            )
+
+            mode = gr.Radio(
+                ["Balanced", "Quality", "Text/Logo"],
+                value="Text/Logo",
+                label="Режим качества"
+            )
+
+            with gr.Accordion("Настройки", open=False):
+                auto_mask = gr.Checkbox(label="Auto-mask", value=True)
+                crop_center = gr.Checkbox(label="Crop по центру", value=True)
+                denoise_steps = gr.Slider(10, 40, value=28, step=1, label="Denoise steps")
+                seed = gr.Slider(0, 999999, value=42, step=1, label="Seed")
+                text_boost = gr.Slider(0, 100, value=55, step=1, label="Text boost (резкость/контраст принта)")
+
+            with gr.Row():
+                run = gr.Button("Примерить", variant="primary")
+                reset = gr.Button("Сбросить", variant="secondary")
+
+            status = gr.Textbox(value="Ожидание...", interactive=False)
+
+        with gr.Column():
+            out = gr.Image(label="Результат", type="pil", height=760)
+
+    run.click(
+        fn=tryon_remote,
+        inputs=[person, garment, garment_desc, auto_mask, crop_center, denoise_steps, seed, mode, text_boost],
+        outputs=[out, status]
+    )
+
+    reset.click(
+        fn=reset_ui,
+        inputs=[],
+        outputs=[person, garment, out, status]
+    )
+
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7863,
+        share=False
+    )