#!/usr/bin/env python3
"""Pulihkan blok preload_from_hub (2 shard Q4_K_M) di README Space."""
from __future__ import annotations

from pathlib import Path

BLOK_PRELOAD = """
preload_from_hub:
  - repo_id: Qwen/Qwen2.5-7B-Instruct-GGUF
    filename: qwen2.5-7b-instruct-q4_k_m-00001-of-00002.gguf
  - repo_id: Qwen/Qwen2.5-7B-Instruct-GGUF
    filename: qwen2.5-7b-instruct-q4_k_m-00002-of-00002.gguf
""".strip()


def main() -> int:
    path = Path("README.md")
    if not path.is_file():
        print("README.md tidak ditemukan")
        return 1

    baris = path.read_text(encoding="utf-8").splitlines()
    keluar: list[str] = []
    skip_boot = False
    punya_preload = False

    for baris_ini in baris:
        if baris_ini.strip().startswith("# preload_from_hub:"):
            skip_boot = True
            continue
        if skip_boot:
            if baris_ini.strip().startswith("- repo_id:") or baris_ini.strip().startswith(
                "filename:"
            ):
                continue
            if baris_ini.strip() and not baris_ini.startswith(" "):
                skip_boot = False
        if baris_ini.strip() == "preload_from_hub:":
            punya_preload = True
        keluar.append(baris_ini)

    if not punya_preload:
        baru: list[str] = []
        for baris_ini in keluar:
            if baris_ini.strip().startswith("startup_duration_timeout:"):
                baru.extend(BLOK_PRELOAD.splitlines())
                punya_preload = True
            baru.append(baris_ini)
        keluar = baru

    path.write_text("\n".join(keluar) + "\n", encoding="utf-8")
    print("README.md: preload_from_hub (2 shard) aktif")
    return 0


if __name__ == "__main__":
    raise SystemExit(main())