Update app.py

app.py

@@ -1,6 +1,8 @@
 # -*- coding: utf-8 -*-
 import os
 import time
+import json
+from datetime import datetime
 from typing import List, Optional, Tuple
 
 import spaces
@@ -242,12 +244,6 @@ pipe.unet_encoder = UNet_Encoder
 
 # =========================
 # Inference (BASELINE params)
-# - num_inference_steps: 30
-# - guidance_scale: 2.0
-# - strength: 1.0
-# - seed: 42
-# - auto_mask: True (upper_body)
-# - crop_center: True
 # =========================
 @spaces.GPU
 def start_tryon(
@@ -405,42 +401,64 @@ def on_gallery_select(files_list: List[str], evt: gr.SelectData):
         return None, "⚠️ Каталог пуст"
     idx = int(evt.index) if evt.index is not None else 0
     idx = max(0, min(idx, len(files_list) - 1))
-    return files_list[idx], f"👕 Выбрано: {files_list[idx]}"
+    # НЕ показываем имя файла
+    return files_list[idx], "👕 Одежда выбрана"
+
+
+# =========================
+# Feedback (like/dislike)
+# =========================
+FEEDBACK_DIR = "./feedback"
+FEEDBACK_PATH = os.path.join(FEEDBACK_DIR, "feedback.jsonl")
+
+
+def save_feedback(is_like: bool, garment_name: str):
+    os.makedirs(FEEDBACK_DIR, exist_ok=True)
+    record = {
+        "timestamp": datetime.utcnow().isoformat(),
+        "like": bool(is_like),
+        "garment": garment_name or "",
+    }
+    with open(FEEDBACK_PATH, "a", encoding="utf-8") as f:
+        f.write(json.dumps(record, ensure_ascii=False) + "\n")
+    return "Спасибо за отзыв 🙏"
 
 
 def tryon_ui(person_pil, selected_filename):
     # динамическое ожидание для UX (каждые 2–3 сек)
     for msg in [
-        "⏳ Анализируем позу…",
-        "⏳ Настраиваем посадку…",
-        "⏳ Переносим ткань…",
-        "⏳ Финальные штрихи…",
+        "🧵 Анализируем посадку ткани…",
+        "📏 Подбираем пропорции…",
+        "🎨 Сохраняем цвет и фактуру…",
+        "🪡 Прорисовываем швы и детали…",
+        "✨ Наводим финальный лоск…",
+        "🪄 Почти готово…",
     ]:
-        yield None, msg
+        yield None, msg, gr.update(visible=False)
         time.sleep(2.3)
 
     ok, msg = allow_call(2.5)
     if not ok:
-        yield None, msg
+        yield None, msg, gr.update(visible=False)
         return
 
     if person_pil is None:
-        yield None, "❌ Загрузите фото человека"
+        yield None, "❌ Загрузите фото человека", gr.update(visible=False)
         return
     if not selected_filename:
-        yield None, "❌ Выберите одежду (клик по превью)"
+        yield None, "❌ Выберите одежду (клик по превью)", gr.update(visible=False)
         return
 
     garm = load_garment_pil(selected_filename)
     if garm is None:
-        yield None, "❌ Не удалось загрузить выбранную одежду"
+        yield None, "❌ Не удалось загрузить выбранную одежду", gr.update(visible=False)
         return
 
     try:
         out_img = start_tryon(human_pil=person_pil, garm_img=garm)
-        yield out_img, "✅ Готово"
+        yield out_img, "✅ Готово", gr.update(visible=True)
     except Exception as e:
-        yield None, f"❌ Ошибка: {type(e).__name__}: {str(e)[:220]}"
+        yield None, f"❌ Ошибка: {type(e).__name__}: {str(e)[:220]}", gr.update(visible=False)
 
 
 # boot
@@ -474,9 +492,8 @@ with gr.Blocks(title="Virtual Try-On Rendez-vous", css=CUSTOM_CSS) as demo:
                 label="Раздел каталога",
             )
 
-            with gr.Row():
-                refresh_btn = gr.Button("🔄 Обновить каталог одежды", variant="secondary")
-                selected_label = gr.Markdown("👕 Выберите одежду ниже")
+            # Убрали кнопку обновления каталога и вывод имени файла
+            selected_label = gr.Markdown("👕 Выберите одежду ниже")
 
             garment_gallery = gr.Gallery(
                 label="Одежда для примерки",
@@ -492,6 +509,11 @@ with gr.Blocks(title="Virtual Try-On Rendez-vous", css=CUSTOM_CSS) as demo:
         with gr.Column():
             out = gr.Image(label="Результат", type="pil", height=760)
 
+            # Кнопки фидбэка появляются после успешной примерки
+            with gr.Row(visible=False) as feedback_row:
+                like_btn = gr.Button("👍 Нравится")
+                dislike_btn = gr.Button("👎 Не нравится")
+
     gender.change(
         fn=refresh_catalog,
         inputs=[gender],
@@ -504,16 +526,24 @@ with gr.Blocks(title="Virtual Try-On Rendez-vous", css=CUSTOM_CSS) as demo:
         outputs=[selected_garment_state, selected_label],
     )
 
-    refresh_btn.click(
-        fn=refresh_catalog,
-        inputs=[gender],
-        outputs=[garment_gallery, garment_files_state, selected_garment_state, status, selected_label],
-    )
-
     run.click(
         fn=tryon_ui,
         inputs=[person, selected_garment_state],
-        outputs=[out, status],
+        outputs=[out, status, feedback_row],
+        concurrency_limit=1,
+    )
+
+    like_btn.click(
+        fn=lambda garment: save_feedback(True, garment),
+        inputs=[selected_garment_state],
+        outputs=[status],
+        concurrency_limit=1,
+    )
+
+    dislike_btn.click(
+        fn=lambda garment: save_feedback(False, garment),
+        inputs=[selected_garment_state],
+        outputs=[status],
         concurrency_limit=1,
     )