"""LOOPBACK TEST ONLY: public reports with synthetic non-user statistics."""
import json
import sys
import tempfile
from datetime import datetime, timedelta
from pathlib import Path
from types import SimpleNamespace

import gradio as gr

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
import admin_stats
from generation_stats import JST


if __name__ == "__main__":
    today = datetime.now(JST).date()
    yesterday = today - timedelta(days=1)
    retired = {f"RETIRED-{i}-" + "very-long-model-name-" * 3: i for i in range(30)}
    with tempfile.TemporaryDirectory(prefix="admin-stats-test-") as directory:
        path = Path(directory) / "daily-counts.json"
        path.write_text(json.dumps({"version": 1, "timezone": "Asia/Tokyo", "counts": {
            yesterday.isoformat(): {"anima": {"PRIVATE-TEST-MODEL": 26}},
            today.isoformat(): {"anima": {"PRIVATE-TEST-MODEL": 94, **retired}, "illustrious": {"ILL-TEST": 2}},
        }}), encoding="utf-8")
        with gr.Blocks(analytics_enabled=False) as demo:
            admin_stats.build_admin_statistics(demo, SimpleNamespace(path=path), {"anima": ["PRIVATE-TEST-MODEL"], "illustrious": ["ILL-TEST", "UNUSED"]})
        demo.launch(server_name="127.0.0.1", server_port=7869)