3rd revision

app.py

@@ -1,103 +1,121 @@
-import os, csv, json, random
+import os, io, csv, json, random
+from datetime import datetime
 import gradio as gr
+from huggingface_hub import HfApi, hf_hub_download
 
-# É  • X (D”Ü ì0Ì )
+# ===== 사용자 설정 =====
+REPO_ID = os.getenv("RESULTS_REPO", "SGTLIM/videoeval_results")  # 생성한 dataset repo
+RESULTS_FILE = "results.csv"  # repo 안에 저장될 파일명
+HF_TOKEN = os.getenv("HF_TOKEN")  # Space Variables에 저장
 EVAL_AXES = [
     ("motion_consistency", "Motion Consistency"),
     ("appearance_consistency", "Appearance Consistency"),
     ("anatomy_consistency", "Anatomy Consistency"),
 ]
+# ======================
 
-# °ü  ¥ ½\ (HF SpacesX |ܤ4¸ ¤ ¬À ¬© Œ¥)
-RESULTS_PATH = os.getenv("RESULTS_CSV", "/data/results.csv")
-
-# D$ ©]
+# 비디오 목록
 with open("videos.json", "r", encoding="utf-8") as f:
     VIDEOS = json.load(f)
 
-# °ü | äT ô¥
-os.makedirs(os.path.dirname(RESULTS_PATH), exist_ok=True)
-if not os.path.exists(RESULTS_PATH):
-    with open(RESULTS_PATH, "w", newline="", encoding="utf-8") as f:
-        writer = csv.writer(f)
-        writer.writerow(["ts_iso", "participant_id", "video_id"] + [k for k, _ in EVAL_AXES] + ["notes"])
-
-def next_video(state):
-    """state: dict  ¥©; 8 Ä œd   À"""
-    if not state or "order" not in state:
-        order = list(range(len(VIDEOS)))
-        random.shuffle(order)
-        state = {"order": order, "idx": 0}
+api = HfApi()
+
+def _read_current_results():
+    """Hub에서 기존 results.csv를 받아와 텍스트(byte)로 리턴. 없으면 None."""
+    try:
+        p = hf_hub_download(repo_id=REPO_ID, filename=RESULTS_FILE,
+                            repo_type="dataset", token=HF_TOKEN, local_dir="/tmp",
+                            local_dir_use_symlinks=False)
+        with open(p, "rb") as f:
+            return f.read()
+    except Exception:
+        return None
+
+def _append_csv_bytes(old_bytes: bytes | None, row: list[str]) -> bytes:
+    """기존 CSV bytes에 한 줄 append하여 새로운 bytes 리턴 (없으면 헤더 포함 생성)."""
+    output = io.StringIO()
+    writer = csv.writer(output)
+    if not old_bytes:
+        header = ["ts_iso", "participant_id", "video_id"] + [k for k, _ in EVAL_AXES] + ["notes"]
+        writer.writerow(header)
+    else:
+        output.write(old_bytes.decode("utf-8", errors="ignore"))
+    writer.writerow(row)
+    return output.getvalue().encode("utf-8")
+
+def push_row_to_hub(participant_id: str, video_id: str, scores: list[int], notes: str):
+    """results.csv를 다운로드→append→업로드 (commit)"""
+    old = _read_current_results()
+    row = [datetime.utcnow().isoformat(), participant_id or "", video_id] + [int(s) for s in scores] + [notes or ""]
+    new_bytes = _append_csv_bytes(old, row)
+    api.upload_file(
+        path_or_fileobj=io.BytesIO(new_bytes),
+        path_in_repo=RESULTS_FILE,
+        repo_id=REPO_ID,
+        repo_type="dataset",
+        token=HF_TOKEN,
+        commit_message="append eval row",
+    )
+
+# ===== Gradio UI =====
+def init_state():
+    order = list(range(len(VIDEOS)))
+    random.shuffle(order)
+    return {"order": order, "idx": 0}
+
+def get_current_video(state):
     idx = state["idx"]
     if idx >= len(state["order"]):
-        return state, None, "(completed)"
+        return None, f"(completed {idx}/{len(state['order'])})"
     v = VIDEOS[state["order"][idx]]
-    label = f'{v["id"]}  ({idx+1}/{len(state["order"])})'
-    return state, v["url"], label
-
-def submit(participant_id, video_id, *scores_and_notes):
-    *scores, notes = scores_and_notes
-    if not video_id or not all(scores):
-        return gr.update(visible=True, value="Please fill all required fields.")
-    from datetime import datetime
-    with open(RESULTS_PATH, "a", newline="", encoding="utf-8") as f:
-        writer = csv.writer(f)
-        writer.writerow([datetime.utcnow().isoformat(), participant_id or "", video_id] + list(map(int, scores)) + [notes or ""])
-    return gr.update(visible=True, value="Saved.")
-
-def save_and_next(participant_id, state, video_id, *scores_and_notes):
-    status = submit(participant_id, video_id, *scores_and_notes)
-    if "Saved." not in (status["value"] or ""):
-        return status, state, gr.update(), gr.update()
+    return v, f"{v['id']}  ({idx+1}/{len(state['order'])})"
+
+def load_first():
+    st = init_state()
+    v, label = get_current_video(st)
+    return st, (v["url"] if v else None), label
+
+def save_and_next(participant_id, state, video_label, *vals):
+    # video_id는 label에서 앞부분만 추출
+    video_id = (video_label or "").split("  (", 1)[0]
+    scores = [vals[i] for i in range(len(EVAL_AXES))]
+    notes = vals[-1]
+
+    # 필수값 체크
+    if not video_id or any(s is None for s in scores):
+        return gr.update(visible=True, value="Please rate all axes."), state, gr.update(), gr.update()
+
+    try:
+        push_row_to_hub(participant_id, video_id, scores, notes)
+        status = "Saved to Hub."
+    except Exception as e:
+        status = f"Save failed: {e}"
+
+    # 다음 영상
     state["idx"] += 1
-    state, url, label = next_video(state)
-    #  0T
-    reset_scores = [gr.update(value=None) for _ in EVAL_AXES]
-    return status, state, gr.update(value=url), gr.update(value=label), *reset_scores, gr.update(value="")
+    v, label = get_current_video(state)
+    resets = [gr.update(value=None) for _ in EVAL_AXES]
+    return gr.update(visible=True, value=status), state, gr.update(value=(v["url"] if v else None)), gr.update(value=label), *resets, gr.update(value="")
 
 with gr.Blocks(fill_width=True) as demo:
-    gr.Markdown("# <¥ Human Evaluation (Video)")
-    gr.Markdown("Please rate each video on the axes below. 1 = poor, 5 = excellent.")
+    gr.Markdown("# 🎥 Human Evaluation (Video)")
+    gr.Markdown("Rate each video (1=poor, 5=excellent).")
 
-    participant_id = gr.Textbox(label="Participant ID (optional)", placeholder="e.g., P001")
+    participant_id = gr.Textbox(label="Participant ID (optional)")
     state = gr.State({})
-
     video = gr.Video(label="Video")
     vid_label = gr.Markdown()
 
-    # ¬|tTä
     sliders = []
     for key, title in EVAL_AXES:
         sliders.append(gr.Slider(1, 5, value=None, step=1, label=title, interactive=True))
     notes = gr.Textbox(label="Notes (optional)", lines=3)
-
-    with gr.Row():
-        prev_btn = gr.Button("À Prev", variant="secondary")
-        save_next_btn = gr.Button("Save & Next ¶", variant="primary")
-
+    save_next = gr.Button("Save & Next ▶", variant="primary")
     status = gr.Alert(visible=False)
 
-    # 0 \Ü
-    def init():
-        st, url, label = next_video({})
-        return st, url, label
-    demo.load(init, [], [state, video, vid_label])
-
-    # Prev (è t<\ tÙ,  ¥@ XÀ JL)
-    def prev(st):
-        if st and st.get("idx", 0) > 0:
-            st["idx"] -= 1
-        st, url, label = next_video(st)
-        resets = [gr.update(value=None) for _ in EVAL_AXES]
-        return st, gr.update(value=url), gr.update(value=label), *resets, gr.update(value="")
-    prev_btn.click(prev, [state], [state, video, vid_label, *sliders, notes])
-
-    # Save & Next
-    save_next_btn.click(
-        save_and_next,
-        inputs=[participant_id, state, vid_label] + sliders + [notes],
-        outputs=[status, state, video, vid_label] + sliders + [notes],
-    )
+    demo.load(load_first, [], [state, video, vid_label])
+    save_next.click(save_and_next, [participant_id, state, vid_label] + sliders + [notes],
+                    [status, state, video, vid_label] + sliders + [notes])
 
 if __name__ == "__main__":
     demo.launch()

requirements.txt

@@ -1,2 +1,3 @@
 gradio==4.44.0
+huggingface_hub>=0.25.0