#!/usr/bin/env python3
"""Export a standalone Hugging Face Space folder for the Task 1 dev portal."""

from __future__ import annotations

import argparse
import shutil
from pathlib import Path


APP_ROOT = Path(__file__).resolve().parent
PROJECT_ROOT = APP_ROOT.parent

SPACE_README = """---
title: __SPACE_TITLE__
emoji: 🧮
colorFrom: blue
colorTo: indigo
sdk: docker
app_port: 7860
pinned: false
---

# __SPACE_TITLE__

This Space hosts the official __SPACE_TITLE__ submission portal and live leaderboard.

Required Space secrets:

- `HF_TOKEN`: write token with access to the private dataset repo used for storage

Required Space variables:

- `TASK1_STORAGE_BACKEND=hf_dataset`
- `TASK1_HF_REPO_ID=<org-or-user>/<private-dataset-repo>`
- `TASK1_OFFICIAL_SITE_URL=https://mbzuai-nlp.github.io/CLEF-2026-FinMMEval-Lab/`
- `TASK1_DEVSET_FILENAME=<public-devset-jsonl>`
- `TASK1_TEMPLATE_FILENAME=<submission-template-json>`
- `TASK1_GOLD_FILENAME=<private-gold-jsonl>`
- `TASK1_OUTPUT_SUBDIR=<leaderboard-output-subdir>`
- `TASK1_PORTAL_VARIANT=<Arabic/Hindi/...>`
- `TASK1_PORTAL_TITLE=<full portal title>`
- `TASK1_PORTAL_SUBMISSION_TITLE=<submission page title>`
- `TASK1_PORTAL_LEADERBOARD_TITLE=<leaderboard page title>`
- `TASK1_PORTAL_DATASET_LABEL=<dataset label shown on homepage>`
"""

DOCKERFILE = """FROM python:3.11-slim

ENV PYTHONDONTWRITEBYTECODE=1
ENV PYTHONUNBUFFERED=1
ENV TASK1_RUNTIME_ROOT=/tmp/task1_dev_leaderboard_runtime

WORKDIR /app

COPY requirements.txt /app/requirements.txt
RUN pip install --no-cache-dir -r /app/requirements.txt

COPY task1_dev_leaderboard /app/task1_dev_leaderboard

EXPOSE 7860

CMD ["python", "-m", "uvicorn", "task1_dev_leaderboard.web_app:app", "--host", "0.0.0.0", "--port", "7860"]
"""

REQUIREMENTS = """fastapi
uvicorn
python-multipart
huggingface_hub
"""

PACKAGE_FILES = [
    "__init__.py",
    "bootstrap_hf_backend.py",
    "create_accounting_dev_set.py",
    "evaluate_submissions.py",
    "prepare_hf_space.py",
    "storage_backend.py",
    "web_app.py",
    "README.md",
]

PACKAGE_DIRS = [
    "web",
]


def parse_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser(description="Prepare a standalone HF Space folder.")
    parser.add_argument(
        "--out-dir",
        default=str(PROJECT_ROOT / "task1_dev_leaderboard_hf_space"),
        help="Output directory for the generated Space repo.",
    )
    parser.add_argument(
        "--space-title",
        default="FinMMEval Task 1 Arabic Dev Portal",
        help="Title written into the generated Space README metadata.",
    )
    parser.add_argument(
        "--devset-filename",
        default=None,
        help="Public dev set filename from task1_dev_leaderboard/dev_sets/ to include in the Space package.",
    )
    parser.add_argument(
        "--template-filename",
        default=None,
        help="Submission template filename from task1_dev_leaderboard/dev_sets/ to include in the Space package.",
    )
    return parser.parse_args()


def main() -> None:
    args = parse_args()
    out_dir = Path(args.out_dir).resolve()
    if out_dir.exists():
        shutil.rmtree(out_dir)
    out_package_dir = out_dir / "task1_dev_leaderboard"
    out_package_dir.mkdir(parents=True, exist_ok=True)

    for filename in PACKAGE_FILES:
        shutil.copy2(APP_ROOT / filename, out_package_dir / filename)

    for dirname in PACKAGE_DIRS:
        shutil.copytree(APP_ROOT / dirname, out_package_dir / dirname)

    dev_sets_out = out_package_dir / "dev_sets"
    dev_sets_out.mkdir(parents=True, exist_ok=True)
    selected_dev_files = [args.devset_filename, args.template_filename]
    if all(selected_dev_files):
        for filename in selected_dev_files:
            src = APP_ROOT / "dev_sets" / filename
            if not src.exists():
                raise FileNotFoundError(f"Missing requested dev set file: {src}")
            shutil.copy2(src, dev_sets_out / filename)
    elif any(selected_dev_files):
        raise ValueError("--devset-filename and --template-filename must be provided together.")
    else:
        shutil.copytree(APP_ROOT / "dev_sets", dev_sets_out, dirs_exist_ok=True)

    readme = SPACE_README.replace("__SPACE_TITLE__", args.space_title)
    (out_dir / "README.md").write_text(readme, encoding="utf-8")
    (out_dir / "Dockerfile").write_text(DOCKERFILE, encoding="utf-8")
    (out_dir / "requirements.txt").write_text(REQUIREMENTS, encoding="utf-8")

    print(f"Prepared HF Space repo at: {out_dir}")


if __name__ == "__main__":
    main()