import os
from huggingface_hub import HfApi
from tempfile import TemporaryDirectory
import importlib.util
import sys


def load_private_code():
    api = HfApi(token=os.environ["PRIVATE_REPOSITORY_HF_TOKEN"])
    
    with TemporaryDirectory() as tmp_dir:
        # プライベートスペースからファイルをダウンロード
        api.snapshot_download(
            repo_id="neoai-inc-private/pj-andromeda",
            repo_type="space",
            local_dir=tmp_dir
        )
        
        # tmp_dirをPythonパスに追加
        sys.path.append(tmp_dir)
        
        # app.pyをインポート
        spec = importlib.util.spec_from_file_location(
            "private_app",
            os.path.join(tmp_dir, "app.py")
        )
        private_module = importlib.util.module_from_spec(spec)
        sys.modules["private_app"] = private_module
        
        try:
            spec.loader.exec_module(private_module)
        except ModuleNotFoundError as e:
            print(f"Error loading module: {e}")
            print(f"Current sys.path: {sys.path}")
            print(f"Contents of tmp_dir: {os.listdir(tmp_dir)}")
            raise
        
        return private_module


# プライベートコードを読み込んでGradioアプリを取得
private_app = load_private_code()
app = private_app.app  # Gradioアプリのインスタンス

# private側の認証情報を使用してアプリを起動
app.launch(
    auth=(private_app.USERNAME, private_app.PASSWORD),  # private側で定義された認証情報を使用
    ssr_mode=False
)