Update app.py

app.py

@@ -61,11 +61,10 @@ def _patch_gradio_client_bool_schema():
     except Exception as e:
         print("gradio_client patch failed:", repr(e), flush=True)
 
-_patch_gradio_client_bool_schema()
 
+_patch_gradio_client_bool_schema()
 
 import torch
-import numpy as np
 from torchvision import transforms
 
 from huggingface_hub import login, snapshot_download
@@ -185,6 +184,8 @@ def clamp_int(x, lo, hi):
 
 
 _last_call_ts = 0.0
+
+
 def allow_call(min_interval_sec: float = 2.5) -> Tuple[bool, str]:
     global _last_call_ts
     now = time.time()
@@ -261,7 +262,6 @@ def start_tryon(
     denoise_steps: int = 25,
     seed: int = 42,
 ) -> Image.Image:
-
     device = "cuda" if torch.cuda.is_available() else "cpu"
     dtype = torch.float16 if device == "cuda" else torch.float32
 
@@ -302,16 +302,18 @@ def start_tryon(
     human_img_arg = _apply_exif_orientation(human_img.resize((384, 512)))
     human_img_arg = convert_PIL_to_numpy(human_img_arg, format="BGR")
 
-    args = apply_net.create_argument_parser().parse_args((
-        "show",
-        "./configs/densepose_rcnn_R_50_FPN_s1x.yaml",
-        "./ckpt/densepose/model_final_162be9.pkl",
-        "dp_segm",
-        "-v",
-        "--opts",
-        "MODEL.DEVICE",
-        "cuda" if device == "cuda" else "cpu",
-    ))
+    args = apply_net.create_argument_parser().parse_args(
+        (
+            "show",
+            "./configs/densepose_rcnn_R_50_FPN_s1x.yaml",
+            "./ckpt/densepose/model_final_162be9.pkl",
+            "dp_segm",
+            "-v",
+            "--opts",
+            "MODEL.DEVICE",
+            "cuda" if device == "cuda" else "cpu",
+        )
+    )
     pose_img = args.func(args, human_img_arg)
     pose_img = pose_img[:, :, ::-1]
     pose_img = Image.fromarray(pose_img).resize((768, 1024))
@@ -329,9 +331,14 @@ def start_tryon(
         if device == "cuda":
             autocast_ctx = torch.cuda.amp.autocast()
         else:
+
             class _NoCtx:
-                def __enter__(self): return None
-                def __exit__(self, *args): return False
+                def __enter__(self):
+                    return None
+
+                def __exit__(self, *args):
+                    return False
+
             autocast_ctx = _NoCtx()
 
         with autocast_ctx:
@@ -392,7 +399,7 @@ def start_tryon(
 
 
 # =========================
-# UI (API-like)
+# UI
 # =========================
 CUSTOM_CSS = """
 footer {display:none !important;}
@@ -401,6 +408,18 @@ div[class*="footer"] {display:none !important;}
 button[aria-label="Settings"] {display:none !important;}
 """
 
+PHOTO_TIPS_MD = """
+### Какое фото подойдёт
+
+✅ В полный рост или по пояс  
+✅ Руки и предметы не закрывают тело  
+✅ Одежда по фигуре  
+✅ Вы стоите прямо и смотрите в камеру  
+✅ Хорошее освещение  
+✅ В кадре нет других людей
+"""
+
+
 def refresh_catalog():
     ensure_garments_downloaded()
     files = list_garments()
@@ -408,6 +427,7 @@ def refresh_catalog():
     status = "✅ Каталог обновлён" if files else "⚠️ Каталог пуст (проверь dataset/токен)"
     return items, files, None, status
 
+
 def on_gallery_select(files_list: List[str], evt: gr.SelectData):
     if not files_list:
         return None, "⚠️ Каталог пуст"
@@ -415,6 +435,7 @@ def on_gallery_select(files_list: List[str], evt: gr.SelectData):
     idx = max(0, min(idx, len(files_list) - 1))
     return files_list[idx], f"👕 Выбрано: {files_list[idx]}"
 
+
 def tryon_ui(person_pil, selected_filename):
     yield None, "⏳ Обработка... (первый запуск может быть дольше)"
 
@@ -436,7 +457,7 @@ def tryon_ui(person_pil, selected_filename):
         return
 
     try:
-        out = start_tryon(
+        out_img = start_tryon(
             human_pil=person_pil,
             garm_img=garm,
             auto_mask=True,
@@ -444,7 +465,7 @@ def tryon_ui(person_pil, selected_filename):
             denoise_steps=25,
             seed=42,
         )
-        yield out, "✅ Готово"
+        yield out_img, "✅ Готово"
     except Exception as e:
         yield None, f"❌ Ошибка: {type(e).__name__}: {str(e)[:220]}"
 
@@ -464,6 +485,9 @@ with gr.Blocks(title="Virtual Try-On Rendez-vous", css=CUSTOM_CSS) as demo:
         with gr.Column():
             person = gr.Image(label="Фото человека", type="pil", height=420)
 
+            # ✅ Добавлено: подсказка сразу под загрузкой фото
+            gr.Markdown(PHOTO_TIPS_MD)
+
             with gr.Row():
                 refresh_btn = gr.Button("🔄 Обновить каталог одежды", variant="secondary")
                 selected_label = gr.Markdown("👕 Выберите одежду ниже")