Update app.py

app.py

@@ -406,14 +406,17 @@ def on_gallery_select(files_list: List[str], evt: gr.SelectData):
 
 
 # =========================
-# Feedback (like/dislike) -> SAVE TO SPACE REPO (no persistent storage)
+# Feedback (like/dislike) -> SAVE TO DATASET REPO (NO SPACE REBUILD)
 # =========================
 FEEDBACK_DIR = "./feedback"
 FEEDBACK_PATH = os.path.join(FEEDBACK_DIR, "feedback.jsonl")
+
+# теперь FEEDBACK_REPO_ID указывает на отдельный Dataset repo (например ArmanRV/idm-vton-feedback)
+FEEDBACK_REPO_ID = os.getenv("FEEDBACK_REPO_ID", "").strip()
+FEEDBACK_REPO_TYPE = "dataset"
 FEEDBACK_REPO_FILEPATH = "feedback/feedback.jsonl"
-SPACE_REPO_ID = (os.getenv("FEEDBACK_REPO_ID", "").strip() or os.getenv("SPACE_ID", "").strip())
 
-_FEEDBACK_UI_MSG = "Спасибо за отзыв! Мы продолжаем улучшать качество примерки."
+_FEEDBACK_UI_MSG = "Спасибо за отзыв, нейросеть еще учится"
 
 
 def _read_local_feedback_text() -> str:
@@ -430,12 +433,12 @@ def _write_local_feedback_text(text: str) -> None:
 
 
 def _download_repo_feedback(api: HfApi, token: str) -> str:
-    if not SPACE_REPO_ID:
+    if not FEEDBACK_REPO_ID:
         return ""
     try:
         local_path = hf_hub_download(
-            repo_id=SPACE_REPO_ID,
-            repo_type="space",
+            repo_id=FEEDBACK_REPO_ID,
+            repo_type=FEEDBACK_REPO_TYPE,
             filename=FEEDBACK_REPO_FILEPATH,
             token=token,
         )
@@ -446,8 +449,8 @@ def _download_repo_feedback(api: HfApi, token: str) -> str:
 
 
 def _upload_repo_feedback(api: HfApi, token: str, text: str) -> None:
-    if not SPACE_REPO_ID:
-        raise RuntimeError("SPACE_ID (repo_id) not found")
+    if not FEEDBACK_REPO_ID:
+        raise RuntimeError("FEEDBACK_REPO_ID not set")
     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:
@@ -456,8 +459,8 @@ def _upload_repo_feedback(api: HfApi, token: str, text: str) -> None:
     api.upload_file(
         path_or_fileobj=tmp_path,
         path_in_repo=FEEDBACK_REPO_FILEPATH,
-        repo_id=SPACE_REPO_ID,
-        repo_type="space",
+        repo_id=FEEDBACK_REPO_ID,
+        repo_type=FEEDBACK_REPO_TYPE,
         token=token,
         commit_message="Add try-on feedback",
     )
@@ -472,7 +475,7 @@ def save_feedback(is_like: bool, garment_name: str):
     }
     line = json.dumps(record, ensure_ascii=False) + "\n"
 
-    # 1) Локально (на всякий случай, внутри контейнера)
+    # 1) локально (на всякий)
     try:
         local_text = _read_local_feedback_text()
         local_text += line
@@ -480,15 +483,15 @@ def save_feedback(is_like: bool, garment_name: str):
     except Exception as e:
         print("Feedback local write failed:", repr(e), flush=True)
 
-    # 2) В репозиторий Space (чтобы не пропадало при рестарте)
+    # 2) в Dataset repo (НЕ вызывает rebuild Space)
     try:
         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 _FEEDBACK_UI_MSG
 
-        if not SPACE_REPO_ID:
-            print("Feedback repo sync skipped: SPACE_ID not set", flush=True)
+        if not FEEDBACK_REPO_ID:
+            print("Feedback repo sync skipped: FEEDBACK_REPO_ID not set", flush=True)
             return _FEEDBACK_UI_MSG
 
         api = HfApi()