app.py

import os, sys, importlib, importlib.util, pkgutil
from pathlib import Path
from tempfile import TemporaryDirectory
from huggingface_hub import HfApi

def load_private_code():
    api = HfApi(token=os.environ["PRIVATE_REPOSITORY_HF_TOKEN"])

    with TemporaryDirectory() as tmp_dir:
        root = Path(tmp_dir)
        api.snapshot_download(
            repo_id="neoai-inc-private/Auriga",
            repo_type="space",
            local_dir=root
        )

        sys.path.insert(0, str(root))
        auriga = root / "Auriga"
        if auriga.exists():
            sys.path.insert(0, str(auriga))

        app_path = (root / "app.py") if (root / "app.py").exists() else next(root.rglob("app.py"), None)
        if not app_path:
            raise FileNotFoundError("app.py が見つかりませんでした")

        prev_cwd = os.getcwd()
        try:
            os.chdir(app_path.parent)

            spec = importlib.util.spec_from_file_location("private_app", app_path)
            private_module = importlib.util.module_from_spec(spec)
            sys.modules["private_app"] = private_module
            spec.loader.exec_module(private_module)

            # 事前インポート
            import neollm
            skipped = []
            for m in pkgutil.walk_packages(neollm.__path__, prefix="neollm."):
                try:
                    importlib.import_module(m.name)
                except (ModuleNotFoundError, ImportError) as e:
                    skipped.append((m.name, str(e)))

            importlib.import_module("neollm.llm.get_llm")
            importlib.import_module("neollm.llm.model_name._abstract_model_name")
            importlib.import_module("neollm.llm.model_name.azure_model_name")

        finally:
            os.chdir(prev_cwd)

        return private_module

private_app = load_private_code()
app = private_app.app
app.launch(
    auth=(private_app.HF_USERNAME, private_app.HF_PASSWORD),
    ssr_mode=False,
    share=True
)