Create app.py

app.py

@@ -0,0 +1,77 @@
+import os
+import requests
+import gradio as gr
+from huggingface_hub import HfApi
+
+# 環境変数から取得
+HF_TOKEN = os.environ.get("HF_TOKEN", None)
+MODEL_REPO = os.environ.get("MODEL_REPO", None)
+LORA_REPO = os.environ.get("LORA_REPO", None)
+
+# HfApiインスタンス生成
+api = HfApi()
+
+def download_and_upload(url, repo_type):
+    # 入力チェック
+    if not url or not repo_type:
+        return "URLまたはタイプが未指定です。"
+    
+    # ファイル名をURLから取得する簡易的な処理
+    # URLの末尾がファイル名を想定
+    filename = url.split("/")[-1]
+    if filename == "":
+        filename = "downloaded_file"
+
+    # ファイルダウンロード
+    try:
+        response = requests.get(url)
+        response.raise_for_status()  # エラー時例外発生
+    except Exception as e:
+        return f"ファイルのダウンロードに失敗しました: {e}"
+
+    # 一時的なローカルファイルに保存
+    with open(filename, "wb") as f:
+        f.write(response.content)
+
+    # 選択されたリポジトリを決定
+    if repo_type == "model":
+        target_repo = MODEL_REPO
+    else:
+        target_repo = LORA_REPO
+
+    if not target_repo:
+        return f"{repo_type}用リポジトリが設定されていません。"
+
+    # Hugging Face Hub にアップロード
+    try:
+        api.upload_file(
+            path_or_fileobj=filename,
+            path_in_repo=filename,  # リポジトリ内でのファイル名
+            repo_id=target_repo,
+            token=HF_TOKEN,
+            repo_type="model", # modelリポジトリとして扱う想定
+            commit_message=f"Add {filename} from {url}"
+        )
+        # 処理が成功したら成功メッセージ
+        return f"ファイル '{filename}' を '{target_repo}' にアップロードしました。"
+    except Exception as e:
+        return f"アップロード中にエラーが発生しました: {e}"
+    finally:
+        # ローカルに残ったファイルを削除するなどの後処理は必要に応じて
+        pass
+
+# Gradio UI構築
+with gr.Blocks() as demo:
+    gr.Markdown("## ファイルダウンロード＆アップロードツール")
+    gr.Markdown("URLを指定して、'model'または'lora'リポジトリへアップロードします。")
+
+    url_input = gr.Textbox(label="ダウンロードファイルURL", placeholder="例: https://example.com/file.bin")
+    repo_choice = gr.Radio(choices=["model", "lora"], label="アップロード先タイプの選択", value="model")
+    run_button = gr.Button("実行")
+    output = gr.Textbox(label="結果メッセージ", interactive=False)
+
+    run_button.click(download_and_upload, inputs=[url_input, repo_choice], outputs=output)
+
+# アプリ起動
+if __name__ == "__main__":
+    demo.launch()