import gradio as gr
import os
import json
import random
from datetime import datetime, timedelta, timezone
from pathlib import Path
from huggingface_hub import CommitScheduler
# --- ① 統計データ保存用の設定 ---
# 自分のリポジトリ名に合わせて変更、または環境変数から取得
DATASET_REPO_ID = "Nyanpre/hasunosora-stats"
DATA_FILE = Path("data") / "stats.json"
DATA_FILE.parent.mkdir(parents=True, exist_ok=True)
# 初回起動時にファイルがない場合は空のデータを作成
if not DATA_FILE.exists():
with open(DATA_FILE, "w", encoding="utf-8") as f:
json.dump({"total": 0, "history": []}, f)
# Hugging Face Datasetへ自動保存するスケジューラー
# ※SpaceのSettingsでCP_ORACLE_TOKEN (write権限) の設定が必要です
scheduler = CommitScheduler(
repo_id=DATASET_REPO_ID,
repo_type="dataset",
folder_path=DATA_FILE.parent,
path_in_repo="stats.json",
token=os.getenv("CP_ORACLE_TOKEN")
)
def load_stats():
if DATA_FILE.exists():
try:
with open(DATA_FILE, "r", encoding="utf-8") as f:
return json.load(f)
except:
pass
return {"total": 0, "history": []}
# --- ② 既存ロジックと統計の統合 ---
members = ["かほ", "さや", "こず", "るり", "めぐ", "つづ", "ぎん", "すず", "ひめ", "せら", "いず", "さち"]
def get_personal_daily_oracle(device_id):
seed_base = device_id if device_id else "default_fate"
jst = timezone(timedelta(hours=9))
now = datetime.now(jst)
today_str = now.strftime("%Y-%m-%d")
random.seed(f"{seed_base}_{today_str}")
selected = random.sample(members, 2)
random.shuffle(selected)
pair_name = f"{selected[0]}{selected[1]}"
# --- ③ 統計の記録処理 ---
with scheduler.lock:
stats = load_stats()
stats["total"] += 1
stats["history"].append({"date": today_str, "pair": pair_name})
with open(DATA_FILE, "w", encoding="utf-8") as f:
json.dump(stats, f, ensure_ascii=False)
oracle_html = (
f"{pair_name}
"
f"本日の神託
"
f"{pair_name}
"
)
peace_msg = "これにより、不毛なカップリング論争は終結しました。"
return oracle_html, peace_msg, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
# --- 神モード：集計ロジック ---
def update_stats_view():
stats = load_stats()
hist = stats.get("history", [])
jst = timezone(timedelta(hours=9))
today_str = datetime.now(jst).strftime("%Y-%m-%d")
counts_all = {}
counts_today = {}
for h in hist:
p = h['pair']
counts_all[p] = counts_all.get(p, 0) + 1
if h.get('date') == today_str:
counts_today[p] = counts_today.get(p, 0) + 1
def gen_rank_html(title, data_dict):
sorted_items = sorted(data_dict.items(), key=lambda x: x[1], reverse=True)[:5]
rows = "".join([f"{i+1}位: {p} ({c}回)" for i, (p, c) in enumerate(sorted_items)])
return f"{title}
{rows if rows else '- データなし
'}
"
html_all = gen_rank_html("全人類神託カップリングTOP5【累計】", counts_all)
html_today = gen_rank_html("全人類神託カップリングTOP5【本日】", counts_today)
return f"{html_all}
{html_today}
総神託数: {stats['total']}
統一されます。", elem_id="doctrine")
result_display = gr.HTML(elem_id="oracle-box", visible=False)
peace_display = gr.Markdown(elem_id="peace-msg")
draw_btn = gr.Button("神託を受ける", variant="primary", elem_id="draw-btn", elem_classes="action-btn")
share_btn_bsky = gr.Button("Blueskyで信仰を広める", variant="secondary", elem_id="share-bsky", elem_classes="action-btn", visible=False)
share_btn_x = gr.Button("Xで信仰を広める", variant="secondary", elem_id="share-x", elem_classes="action-btn", visible=False)
# --- ② 神モードタブ ---
with gr.TabItem("神モード"):
with gr.Column(elem_classes="center-content"):
gr.Markdown("### 全人類の信仰状況")
stats_display = gr.HTML("ボタンを押して読み込み
")
refresh_stats = gr.Button("全人類の神託を見る", variant="primary", elem_classes="action-btn")
# --- 既存のクリックイベント ---
draw_btn.click(
fn=get_personal_daily_oracle,
inputs=[device_id_storage],
outputs=[result_display, peace_display, share_btn_bsky, share_btn_x, result_display],
js=js_logic
).then(
fn=None,
inputs=[result_display, peace_display, share_btn_bsky, share_btn_x, result_display],
outputs=None,
js=js_save_result
)
share_btn_bsky.click(fn=None, inputs=None, outputs=None, js=js_share_bluesky)
share_btn_x.click(fn=None, inputs=None, outputs=None, js=js_share_x)
# --- 新機能のクリックイベント ---
refresh_stats.click(fn=update_stats_view, outputs=stats_display)
if __name__ == "__main__":
demo.launch()