Update app.py

app.py

@@ -200,7 +200,10 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 DTYPE = torch.float16 if DEVICE == "cuda" else torch.float32
 print("DEVICE:", DEVICE, "DTYPE:", DTYPE, flush=True)
 
-tensor_transfrom = transforms.Compose([transforms.ToTensor(), transforms.Normalize([0.5], [0.5])])
+tensor_transfrom = transforms.Compose([
+    transforms.ToTensor(),
+    transforms.Normalize([0.5], [0.5]),
+])
 
 unet = UNet2DConditionModel.from_pretrained(base_path, subfolder="unet", torch_dtype=DTYPE)
 unet.requires_grad_(False)
@@ -309,7 +312,6 @@ def start_tryon(
         if device == "cuda":
             autocast_ctx = torch.cuda.amp.autocast()
         else:
-
             class _NoCtx:
                 def __enter__(self):
                     return None
@@ -393,6 +395,7 @@ button[aria-label="Settings"] {display:none !important;}
     border-radius: 12px;
     padding: 10px 12px;
     font-size: 14px;
+    margin-top: 8px;
 }
 
 .feedback-idle {
@@ -402,6 +405,7 @@ button[aria-label="Settings"] {display:none !important;}
     border-radius: 12px;
     padding: 10px 12px;
     font-size: 14px;
+    margin-top: 8px;
 }
 """
 
@@ -441,7 +445,7 @@ def on_gallery_select(files_list: List[str], evt: gr.SelectData):
 
 
 # =========================
-# Feedback
+# Feedback storage
 # =========================
 FEEDBACK_DIR = "./feedback"
 FEEDBACK_PATH = os.path.join(FEEDBACK_DIR, "feedback.jsonl")
@@ -486,6 +490,7 @@ def _upload_repo_feedback(api: HfApi, token: str, text: str) -> None:
 
     os.makedirs(FEEDBACK_DIR, exist_ok=True)
     tmp_path = os.path.join(FEEDBACK_DIR, "_feedback_upload.jsonl")
+
     with open(tmp_path, "w", encoding="utf-8") as f:
         f.write(text)
 
@@ -499,19 +504,7 @@ def _upload_repo_feedback(api: HfApi, token: str, text: str) -> None:
     )
 
 
-def save_feedback(is_like: bool, garment_name: str, comment: str):
-    os.makedirs(FEEDBACK_DIR, exist_ok=True)
-
-    clean_comment = (comment or "").strip()
-    if len(clean_comment) > 1500:
-        clean_comment = clean_comment[:1500]
-
-    record = {
-        "timestamp": datetime.utcnow().isoformat(),
-        "like": bool(is_like),
-        "garment": garment_name or "",
-        "comment": clean_comment,
-    }
+def _append_feedback_record(record: dict) -> None:
     line = json.dumps(record, ensure_ascii=False) + "\n"
 
     try:
@@ -525,11 +518,11 @@ def save_feedback(is_like: bool, garment_name: str, comment: str):
         token = (os.getenv("HF_TOKEN", "").strip() or os.getenv("HUGGINGFACEHUB_API_TOKEN", "").strip())
         if not token:
             print("Feedback repo sync skipped: HF_TOKEN not set", flush=True)
-            return True
+            return
 
         if not FEEDBACK_REPO_ID:
             print("Feedback repo sync skipped: FEEDBACK_REPO_ID not set", flush=True)
-            return True
+            return
 
         api = HfApi()
         repo_text = _download_repo_feedback(api, token)
@@ -538,63 +531,132 @@ def save_feedback(is_like: bool, garment_name: str, comment: str):
     except Exception as e:
         print("Feedback repo sync failed:", repr(e), flush=True)
 
-    return True
+
+def save_rating_feedback(is_like: bool, garment_name: str) -> None:
+    record = {
+        "timestamp": datetime.utcnow().isoformat(),
+        "event": "rating",
+        "like": bool(is_like),
+        "garment": garment_name or "",
+    }
+    _append_feedback_record(record)
+
+
+def save_comment_feedback(garment_name: str, comment: str) -> None:
+    clean_comment = (comment or "").strip()
+    if len(clean_comment) > 1500:
+        clean_comment = clean_comment[:1500]
+
+    record = {
+        "timestamp": datetime.utcnow().isoformat(),
+        "event": "comment",
+        "garment": garment_name or "",
+        "comment": clean_comment,
+    }
+    _append_feedback_record(record)
+
+
+# =========================
+# Feedback UI helpers
+# =========================
+def _rating_notice_idle_html():
+    return """
+    <div class="feedback-idle">
+        Оцените результат: нажмите лайк или дизлайк.
+    </div>
+    """
 
 
-def _feedback_idle_html():
+def _comment_notice_idle_html():
     return """
     <div class="feedback-idle">
-        Оцените результат и при желании оставьте комментарий.
+        При желании напишите комментарий и нажмите кнопку отправки.
     </div>
     """
 
 
-def _feedback_ok_html(action_text: str, has_comment: bool):
-    comment_text = " и комментарий" if has_comment else ""
+def _rating_notice_ok_html(action_text: str):
     return f"""
     <div class="feedback-ok">
-        ✅ Спасибо! Оценка <b>{action_text}</b>{comment_text} сохранены.
+        ✅ Ваша оценка <b>{action_text}</b> сохранена.
+    </div>
+    """
+
+
+def _comment_notice_ok_html():
+    return """
+    <div class="feedback-ok">
+        ✅ Комментарий отправлен. Спасибо за обратную связь.
     </div>
     """
 
 
-def submit_feedback_ui(is_like: bool, garment_name: str, comment: str):
-    action_text = "«Нравится»" if is_like else "«Не нравится»"
+# =========================
+# Feedback actions
+# =========================
+def submit_like_feedback(garment_name: str):
+    if not garment_name:
+        return "⚠️ Сначала выполните примерку и выберите одежду", _rating_notice_idle_html()
+
+    try:
+        save_rating_feedback(True, garment_name)
+        return "✅ Оценка сохранена: «Нравится»", _rating_notice_ok_html("«Нравится»")
+    except Exception as e:
+        return (
+            f"❌ Ошибка сохранения оценки: {type(e).__name__}: {str(e)[:200]}",
+            _rating_notice_idle_html(),
+        )
+
+
+def submit_dislike_feedback(garment_name: str):
+    if not garment_name:
+        return "⚠️ Сначала выполните примерку и выберите одежду", _rating_notice_idle_html()
+
+    try:
+        save_rating_feedback(False, garment_name)
+        return "✅ Оценка сохранена: «Не нравится»", _rating_notice_ok_html("«Не нравится»")
+    except Exception as e:
+        return (
+            f"❌ Ошибка сохранения оценки: {type(e).__name__}: {str(e)[:200]}",
+            _rating_notice_idle_html(),
+        )
+
+
+def submit_comment_feedback(garment_name: str, comment: str):
+    clean_comment = (comment or "").strip()
 
     if not garment_name:
         return (
             "⚠️ Сначала выполните примерку и выберите одежду",
-            _feedback_idle_html(),
+            _comment_notice_idle_html(),
+            gr.update(value=comment),
+        )
+
+    if not clean_comment:
+        return (
+            "⚠️ Напишите комментарий перед отправкой",
+            _comment_notice_idle_html(),
             gr.update(value=comment),
         )
 
     try:
-        save_feedback(is_like=is_like, garment_name=garment_name, comment=comment)
+        save_comment_feedback(garment_name, clean_comment)
         return (
-            f"✅ Отзыв сохранён: {action_text}",
-            _feedback_ok_html(action_text, bool((comment or "").strip())),
+            "✅ Комментарий отправлен",
+            _comment_notice_ok_html(),
             gr.update(value=""),
         )
     except Exception as e:
         return (
-            f"❌ Ошибка сохранения отзыва: {type(e).__name__}: {str(e)[:200]}",
-            """
-            <div class="feedback-idle">
-                Не удалось сохранить отзыв. Попробуйте ещё раз.
-            </div>
-            """,
+            f"❌ Ошибка отправки комментария: {type(e).__name__}: {str(e)[:200]}",
+            _comment_notice_idle_html(),
             gr.update(value=comment),
         )
 
 
-def submit_like_feedback(garment_name: str, comment: str):
-    return submit_feedback_ui(True, garment_name, comment)
-
-
-def submit_dislike_feedback(garment_name: str, comment: str):
-    return submit_feedback_ui(False, garment_name, comment)
-
-
+# =========================
+# Try-on UI
+# =========================
 def tryon_ui(person_pil, selected_filename):
     for msg in [
         "🧵 Анализируем посадку ткани…",
@@ -609,7 +671,8 @@ def tryon_ui(person_pil, selected_filename):
             msg,
             gr.update(visible=False),
             gr.update(value=""),
-            gr.update(value=_feedback_idle_html()),
+            gr.update(value=_rating_notice_idle_html()),
+            gr.update(value=_comment_notice_idle_html()),
         )
         time.sleep(2.3)
 
@@ -620,7 +683,8 @@ def tryon_ui(person_pil, selected_filename):
             msg,
             gr.update(visible=False),
             gr.update(value=""),
-            gr.update(value=_feedback_idle_html()),
+            gr.update(value=_rating_notice_idle_html()),
+            gr.update(value=_comment_notice_idle_html()),
         )
         return
 
@@ -630,7 +694,8 @@ def tryon_ui(person_pil, selected_filename):
             "❌ Загрузите фото человека",
             gr.update(visible=False),
             gr.update(value=""),
-            gr.update(value=_feedback_idle_html()),
+            gr.update(value=_rating_notice_idle_html()),
+            gr.update(value=_comment_notice_idle_html()),
         )
         return
 
@@ -640,7 +705,8 @@ def tryon_ui(person_pil, selected_filename):
             "❌ Выберите одежду (клик по превью)",
             gr.update(visible=False),
             gr.update(value=""),
-            gr.update(value=_feedback_idle_html()),
+            gr.update(value=_rating_notice_idle_html()),
+            gr.update(value=_comment_notice_idle_html()),
         )
         return
 
@@ -651,7 +717,8 @@ def tryon_ui(person_pil, selected_filename):
             "❌ Не удалось загрузить выбранную одежду",
             gr.update(visible=False),
             gr.update(value=""),
-            gr.update(value=_feedback_idle_html()),
+            gr.update(value=_rating_notice_idle_html()),
+            gr.update(value=_comment_notice_idle_html()),
         )
         return
 
@@ -659,10 +726,11 @@ def tryon_ui(person_pil, selected_filename):
         out_img = start_tryon(human_pil=person_pil, garm_img=garm)
         yield (
             out_img,
-            "✅ Готово — оцените результат ниже",
+            "✅ Готово — оцените результат и при желании оставьте комментарий",
             gr.update(visible=True),
             gr.update(value=""),
-            gr.update(value=_feedback_idle_html()),
+            gr.update(value=_rating_notice_idle_html()),
+            gr.update(value=_comment_notice_idle_html()),
         )
     except Exception as e:
         yield (
@@ -670,11 +738,14 @@ def tryon_ui(person_pil, selected_filename):
             f"❌ Ошибка: {type(e).__name__}: {str(e)[:220]}",
             gr.update(visible=False),
             gr.update(value=""),
-            gr.update(value=_feedback_idle_html()),
+            gr.update(value=_rating_notice_idle_html()),
+            gr.update(value=_comment_notice_idle_html()),
         )
 
 
-# boot
+# =========================
+# Boot
+# =========================
 ensure_garments_available()
 _default_gender = "Женская"
 _initial_files = list_garments(gender=_default_gender)
@@ -734,18 +805,23 @@ with gr.Blocks(title="Virtual Try-On Rendez-vous", css=CUSTOM_CSS) as demo:
             with gr.Column(visible=False) as feedback_box:
                 gr.HTML('<div class="feedback-box">')
 
+                gr.Markdown("### Оценка результата")
                 with gr.Row():
                     like_btn = gr.Button("👍 Нравится")
                     dislike_btn = gr.Button("👎 Не нравится")
+                rating_notice = gr.HTML(_rating_notice_idle_html())
 
+                gr.Markdown("### Комментарий")
                 feedback_comment = gr.Textbox(
-                    label="Комментарий",
+                    label="",
                     placeholder="Напишите, что понравилось или что стоит улучшить...",
                     lines=3,
-                    max_lines=5,
+                    max_lines=6,
+                    show_label=False,
                 )
+                submit_comment_btn = gr.Button("Отправить комментарий")
+                comment_notice = gr.HTML(_comment_notice_idle_html())
 
-                feedback_notice = gr.HTML(_feedback_idle_html())
                 gr.HTML("</div>")
 
     gender.change(
@@ -763,21 +839,28 @@ with gr.Blocks(title="Virtual Try-On Rendez-vous", css=CUSTOM_CSS) as demo:
     run.click(
         fn=tryon_ui,
         inputs=[person, selected_garment_state],
-        outputs=[out, status, feedback_box, feedback_comment, feedback_notice],
+        outputs=[out, status, feedback_box, feedback_comment, rating_notice, comment_notice],
         concurrency_limit=1,
     )
 
     like_btn.click(
         fn=submit_like_feedback,
-        inputs=[selected_garment_state, feedback_comment],
-        outputs=[status, feedback_notice, feedback_comment],
+        inputs=[selected_garment_state],
+        outputs=[status, rating_notice],
         concurrency_limit=1,
     )
 
     dislike_btn.click(
         fn=submit_dislike_feedback,
+        inputs=[selected_garment_state],
+        outputs=[status, rating_notice],
+        concurrency_limit=1,
+    )
+
+    submit_comment_btn.click(
+        fn=submit_comment_feedback,
         inputs=[selected_garment_state, feedback_comment],
-        outputs=[status, feedback_notice, feedback_comment],
+        outputs=[status, comment_notice, feedback_comment],
         concurrency_limit=1,
     )