deploy app, storage, readme

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import random
 import threading
+import time
 import uuid
 import os
 import html
@@ -18,6 +19,8 @@ SUBMIT_KEY = os.getenv("RATINGS_SUBMIT_KEY")
 assert SUBMIT_KEY, "Missing required env var: RATINGS_SUBMIT_KEY"
 POOL_REPO_ID = "taigasan/e6-visual-ratings"
 VOTE_STORAGE = VoteStorage(mode="local" if DEBUG_MODE else "hf", local_dir=LOCAL_DATA_DIR, token=RATINGS_APP_TOKEN)
+STATS_PATH = Path(__file__).resolve().parent.parent / "stats.parquet"
+STATS_RELOAD_S = 30 * 60
 
 # -- Pool dataset -----------------------------------------------------------
 if DEBUG_MODE:
@@ -32,6 +35,32 @@ else:
     )
 _pool_df = pd.read_parquet(_pool_path)
 _pool_group_dfs = {g: gdf for g, gdf in _pool_df.groupby("group")}
+_stats_lock = threading.Lock()
+_stats_last_loaded_at = 0.0
+_stats_by_key: dict[str, tuple[int, int]] = {}
+
+
+def _reload_stats_if_due(force: bool = False):
+    global _stats_last_loaded_at, _stats_by_key
+    now = time.time()
+    if not force and (now - _stats_last_loaded_at) < STATS_RELOAD_S:
+        return
+    with _stats_lock:
+        now = time.time()
+        if not force and (now - _stats_last_loaded_at) < STATS_RELOAD_S:
+            return
+        if STATS_PATH.exists():
+            _stats_df = pd.read_parquet(STATS_PATH, columns=["item_key", "wins", "losses"])  # [n, 3]
+            _stats_by_key = {
+                str(r.item_key): (int(r.wins), int(r.losses))
+                for r in _stats_df.itertuples(index=False)
+            }
+        else:
+            _stats_by_key = {}
+        _stats_last_loaded_at = now
+
+
+_reload_stats_if_due(force=True)
 
 def _pool_fetch_pair(group_name: str) -> tuple:
     gdf = _pool_group_dfs[group_name]
@@ -73,10 +102,19 @@ def _commit_oldest_pending(state: dict):
     threading.Thread(target=VOTE_STORAGE.append_vote_row, args=(oldest.copy(), oldest.get("winner")), daemon=True).start()
 
 def _render_current(state: dict, submit_status: str = "") -> tuple:
-    img_a_html = f"<div class=\"rating-card\"><div class=\"rating-card-title\"><strong>Image A</strong></div><div class=\"rating-image-frame\"><img src=\"{html.escape(state['url_a'])}\" class=\"rating-image\" loading=\"eager\" referrerpolicy=\"no-referrer\"></div></div>"
-    img_b_html = f"<div class=\"rating-card\"><div class=\"rating-card-title\"><strong>Image B</strong></div><div class=\"rating-image-frame\"><img src=\"{html.escape(state['url_b'])}\" class=\"rating-image\" loading=\"eager\" referrerpolicy=\"no-referrer\"></div></div>"
-    link_a = f"Image A: https://e621.net/posts/{state['id_a']}"
-    link_b = f"Image B: https://e621.net/posts/{state['id_b']}"
+    _reload_stats_if_due()
+    wins_a, losses_a = _stats_by_key.get(str(state["key_a"]), (0, 0))
+    wins_b, losses_b = _stats_by_key.get(str(state["key_b"]), (0, 0))
+    total_a = wins_a + losses_a
+    total_b = wins_b + losses_b
+    win_pct_a = f"{(100.0 * wins_a / total_a):.1f}%" if total_a > 0 else "-"
+    win_pct_b = f"{(100.0 * wins_b / total_b):.1f}%" if total_b > 0 else "-"
+    title_a = "Image A"
+    title_b = "Image B"
+    img_a_html = f"<div class=\"rating-card\"><div class=\"rating-card-title\"><strong>{html.escape(title_a)}</strong></div><div class=\"rating-image-frame\"><img src=\"{html.escape(state['url_a'])}\" class=\"rating-image\" loading=\"eager\" referrerpolicy=\"no-referrer\"></div></div>"
+    img_b_html = f"<div class=\"rating-card\"><div class=\"rating-card-title\"><strong>{html.escape(title_b)}</strong></div><div class=\"rating-image-frame\"><img src=\"{html.escape(state['url_b'])}\" class=\"rating-image\" loading=\"eager\" referrerpolicy=\"no-referrer\"></div></div>"
+    link_a = f"Image A: https://e621.net/posts/{state['id_a']} | Wins: {wins_a} | Losses: {losses_a} | Win percentage: {win_pct_a}"
+    link_b = f"Image B: https://e621.net/posts/{state['id_b']} | Wins: {wins_b} | Losses: {losses_b} | Win percentage: {win_pct_b}"
     can_go_back = bool(state.get("can_go_back"))
     back_md = "[back](#back)" if can_go_back else "<span class='subtle-back-link-disabled'>back</span>"
     details = f"<span class='subtle-note'>Group: {state['group']}</span>"