Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import os | |
| import shutil | |
| from huggingface_hub import HfApi, HfFolder, snapshot_download | |
| # ローカル / デプロイ共通の環境トークン（あれば） | |
| ENV_HF_TOKEN = os.getenv("HF_TOKEN") | |
| api = HfApi() | |
| target_space = "s-4-s/pre-s4s-editor" | |
| local_dir = "./space_copy" | |
| def run_clone(source_space: str, profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None): | |
| """ | |
| 注意: Gradio は profile と oauth_token を自動注入します。 | |
| - profile が None => 未ログイン | |
| - oauth_token が None でも、ENV_HF_TOKEN があればそれを使う（ローカルデバッグ用） | |
| """ | |
| logs = "" | |
| def log(msg: str): | |
| nonlocal logs | |
| logs += msg + "\n" | |
| yield logs | |
| try: | |
| hf_token = oauth_token.token if oauth_token is not None else ENV_HF_TOKEN | |
| if profile is None: | |
| yield from log("❌ 未ログインです。先にログインしてください。") | |
| return | |
| if getattr(profile, "username", None) != "soiz1": | |
| yield from log(f"❌ アクセス拒否: {getattr(profile, 'username', '不明')}") | |
| return | |
| if hf_token is None: | |
| yield from log("❌ 実行に必要な Hugging Face トークンが見つかりません。") | |
| return | |
| HfFolder.save_token(hf_token) | |
| yield from log("=== Hugging Face Space クローン開始 ===") | |
| if os.path.exists(local_dir): | |
| yield from log(f"既存の '{local_dir}' を削除します...") | |
| shutil.rmtree(local_dir) | |
| yield from log(f"'{source_space}' をローカルにダウンロード中...") | |
| snapshot_download( | |
| repo_id=source_space, | |
| repo_type="space", | |
| local_dir=local_dir, | |
| token=hf_token | |
| ) | |
| yield from log(f"'{source_space}' を '{local_dir}' にダウンロードしました。") | |
| yield from log(f"'{target_space}' へアップロード中...") | |
| api.upload_folder( | |
| folder_path=local_dir, | |
| repo_id=target_space, | |
| repo_type="space", | |
| token=hf_token, | |
| commit_message="自動クローン" | |
| ) | |
| yield from log(f"'{source_space}' の内容を '{target_space}' にアップロードしました。") | |
| yield from log("=== 処理完了 ===") | |
| except Exception as e: | |
| yield from log(f"エラー発生: {e}") | |
| with gr.Blocks() as demo: | |
| gr.Markdown("## 🚀 Hugging Face Space 自動クローンツール") | |
| source_space_input = gr.Textbox( | |
| label="ソース Space", | |
| value="soiz1/soiz1-s4s-editor", | |
| lines=1, | |
| ) | |
| gr.LoginButton() | |
| run_button = gr.Button("クローン & アップロード開始", interactive=True) | |
| log_output = gr.Textbox(label="ログ出力", lines=20, interactive=False) | |
| # source_space_input も inputs に渡す | |
| run_button.click( | |
| fn=run_clone, | |
| inputs=[source_space_input], | |
| outputs=log_output, | |
| ) | |
| demo.launch() | |