Update app.py

app.py

@@ -8,6 +8,7 @@ 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)
+CIVITAI_TOKEN = os.environ.get("CIVITAI_TOKEN", None)  # Civitai用トークン
 
 api = HfApi()
 
@@ -16,6 +17,9 @@ def download_and_upload(url, repo_type):
         return "URLまたはタイプが未指定です。"
 
     headers = {}
+    if CIVITAI_TOKEN:
+        headers["Authorization"] = f"Bearer {CIVITAI_TOKEN}"
+
     try:
         response = requests.get(url, headers=headers)
         response.raise_for_status()
@@ -35,7 +39,11 @@ def download_and_upload(url, repo_type):
     with open(filename, "wb") as f:
         f.write(response.content)
 
-    target_repo = MODEL_REPO if repo_type == "model" else LORA_REPO
+    if repo_type == "model":
+        target_repo = MODEL_REPO
+    else:
+        target_repo = LORA_REPO
+
     if not target_repo:
         return f"{repo_type}用リポジトリが環境変数で設定されていません。"
 
@@ -73,7 +81,7 @@ def apply_reorder_to_repo(data, repo_type):
         if not target_repo:
             return "エラー: リポジトリが設定されていません。"
         
-        # 並び替えられたファイルリストをリポジトリに反映 (例: ファイル順序を更新する処理を追加可能)
+        # 並び替えられたファイルリストをリポジトリに反映 (例としてMockで結果を出力)
         reordered_files = sorted(data, key=lambda x: int(x[0]))  # No.列の順序に基づいて並び替え
         reordered_filenames = [file[1] for file in reordered_files]