Update app.py

app.py

@@ -9,6 +9,7 @@ from huggingface_hub import HfApi, hf_hub_download
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
 MODEL_REPO = os.environ.get("MODEL_REPO", None)
 LORA_REPO = os.environ.get("LORA_REPO", None)
+EMBE_REPO = os.environ.get("EMBE_REPO", None)  # embedding用リポジトリ
 CIVITAI_TOKEN = os.environ.get("CIVITAI_TOKEN", None)  # Civitai用トークン
 
 api = HfApi()
@@ -65,8 +66,12 @@ def download_and_upload(model_id, repo_type):
 
     if repo_type == "model":
         target_repo = MODEL_REPO
-    else:
+    elif repo_type == "lora":
         target_repo = LORA_REPO
+    elif repo_type == "embedding":
+        target_repo = EMBE_REPO
+    else:
+        return "不正なリポジトリタイプです。"
 
     if not target_repo:
         return f"{repo_type}用リポジトリが環境変数で設定されていません。"
@@ -135,7 +140,6 @@ def download_and_upload(model_id, repo_type):
     print(f"メタデータファイルを更新しました。ファイル名: {metadata_filename}")
 
     # ファイルのアップロード
-
     try:
         # 元のファイルをアップロード
         api.upload_file(
@@ -165,7 +169,7 @@ def download_and_upload(model_id, repo_type):
 # Gradio UI構築
 with gr.Blocks() as demo:
     gr.Markdown("## CivitaiモデルIDからファイルダウンロード＆アップロードツール")
-    gr.Markdown("CivitaiのモデルIDを指定して、'model'または'lora'リポジトリへアップロードします。")
+    gr.Markdown("CivitaiのモデルIDを指定して、'model'、'lora'、または'embedding'リポジトリへアップロードします。")
 
     # ダウンロード＆アップロード機能
     model_id_input = gr.Textbox(
@@ -173,7 +177,7 @@ with gr.Blocks() as demo:
         placeholder="例: 1102"
     )
     repo_choice = gr.Radio(
-        choices=["model", "lora"],
+        choices=["model", "lora", "embedding"],  # "embedding" を追加
         label="アップロード先タイプの選択",
         value="model"
     )