Update app.py

app.py

@@ -11,93 +11,22 @@ 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用トークン
 
-# Hugging Face APIの初期化
 api = HfApi()
 
-def extract_model_id(url):
-    """
-    CivitaiのダウンロードURLからモデルIDを抽出する関数。
-    例: https://civitai.com/api/download/models/546416@685247?type=Model&format=SafeTensor
-    から '546416' を抽出。
-    """
-    match = re.search(r'/api/download/models/([\w@]+)', url)
-    if match:
-        full_id = match.group(1)
-        # '@'が含まれている場合、前半部分を使用
-        if '@' in full_id:
-            return full_id.split('@')[0]
-        return full_id
-    else:
-        return None
-
-def get_metadata(model_id):
-    """
-    Civitai APIからモデルのトリガーワードと推奨強度を取得する関数。
-    """
-    if not model_id:
-        return "N/A", "N/A"
-
-    api_url = f"https://civitai.com/api/v1/models/{model_id}"
-    headers = {}
-    if CIVITAI_TOKEN:
-        headers["Authorization"] = f"Bearer {CIVITAI_TOKEN}"
-
-    try:
-        response = requests.get(api_url, headers=headers)
-        response.raise_for_status()
-        data = response.json()
-
-        # デバッグ: レスポンス全体を確認
-        print("Civitai API Response:")
-        print(json.dumps(data, indent=4, ensure_ascii=False))
-
-        # トリガーワードの取得
-        # 実際のフィールド名を確認し、適宜修正してください
-        trigger_words = data.get('trainedWords', 'N/A')  # 例: ['word1', 'word2']
-
-        # 推奨強度の取得
-        description = data.get('description', '')
-        # 正規表現を柔軟にして、様々な記述形式に対応
-        strength_match = re.search(r'strength[:=]?\s*([\d.]+)', description, re.IGNORECASE)
-        if strength_match:
-            strength = float(strength_match.group(1))
-        else:
-            strength = "N/A"
-
-        return strength, trigger_words
-    except Exception as e:
-        print(f"メタデータ取得エラー: {e}")
-        return "N/A", "N/A"
-
 def download_and_upload(url, repo_type):
     if not url or not repo_type:
         return "URLまたはタイプが未指定です。"
 
-    # URLのバリデーション
-    def is_valid_civitai_url(url):
-        pattern = r'https?://civitai\.com/api/download/models/[\w@]+\?type=(Model|LoRA)&format=SafeTensor'
-        return re.match(pattern, url) is not None
-
-    if not is_valid_civitai_url(url):
-        return "無効なCivitaiモデルAPIのURLです。形式を確認してください。"
-
-    # モデルIDの抽出
-    model_id = extract_model_id(url)
-    if not model_id:
-        return "URLからモデルIDを抽出できませんでした。URL形式を確認してください。"
-
     headers = {}
     if CIVITAI_TOKEN:
         headers["Authorization"] = f"Bearer {CIVITAI_TOKEN}"
 
     try:
-        # ファイルのダウンロード
         response = requests.get(url, headers=headers)
         response.raise_for_status()
     except Exception as e:
         return f"ファイルのダウンロードに失敗しました: {e}"
 
-    # ファイル名の取得
     content_disp = response.headers.get("content-disposition", "")
     filename = None
     if content_disp:
@@ -107,65 +36,19 @@ def download_and_upload(url, repo_type):
 
     if not filename:
         filename = "downloaded_file.bin"
+    
+    with open(filename, "wb") as f:
+        f.write(response.content)
 
-    # ファイルの保存
-    try:
-        with open(filename, "wb") as f:
-            f.write(response.content)
-    except Exception as e:
-        return f"ファイルの保存に失敗しました: {e}"
-
-    # メタデータの取得
-    strength, trigger_words = get_metadata(model_id)
-
-    # メタデータのダウンロード
-    metadata_filename = "metadata.json"
-    metadata = []
-    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}用リポジトリが環境変数で設定されていません。"
 
     try:
-        # メタデータファイルをリポジトリからダウンロード
-        api.repo_download_file(
-            repo_id=target_repo,
-            path=metadata_filename,
-            local_dir=".",
-            local_dir_use_symlinks=False,
-            token=HF_TOKEN
-        )
-
-        # ダウンロードしたメタデータファイルを読み込む
-        with open(metadata_filename, "r", encoding="utf-8") as f:
-            metadata = json.load(f)
-    except Exception as e:
-        # メタデータファイルが存在しない場合は新規作成
-        print(f"メタデータファイルのダウンロードまたは読み込みに失敗しました: {e}")
-        metadata = []
-
-    # 通し番号の決定
-    serial_number = len(metadata) + 1
-
-    # 新しいメタデータエントリの作成
-    new_entry = {
-        "serial_number": serial_number,
-        "filename": filename,
-        "strength": strength,
-        "trigger_words": trigger_words
-    }
-
-    metadata.append(new_entry)
-
-    # メタデータファイルの保存
-    try:
-        with open(metadata_filename, "w", encoding="utf-8") as f:
-            json.dump(metadata, f, ensure_ascii=False, indent=4)
-    except Exception as e:
-        return f"メタデータファイルの保存に失敗しました: {e}"
-
-    try:
-        # ファイルのアップロード
         api.upload_file(
             path_or_fileobj=filename,
             path_in_repo=filename,
@@ -174,22 +57,7 @@ def download_and_upload(url, repo_type):
             repo_type="dataset",
             commit_message=f"Add {filename} from {url}"
         )
-
-        # メタデータファイルのアップロード
-        api.upload_file(
-            path_or_fileobj=metadata_filename,
-            path_in_repo=metadata_filename,
-            repo_id=target_repo,
-            token=HF_TOKEN,
-            repo_type="dataset",
-            commit_message=f"Update metadata with {filename}"
-        )
-
-        # 不要ファイルの削除（オプション）
-        os.remove(filename)
-        os.remove(metadata_filename)
-
-        return f"ファイル '{filename}' とメタデータを '{target_repo}' にアップロードしました。"
+        return f"ファイル '{filename}' を '{target_repo}' にアップロードしました。"
     except Exception as e:
         return f"アップロード中にエラーが発生しました: {e}"
 
@@ -197,11 +65,11 @@ def download_and_upload(url, repo_type):
 with gr.Blocks() as demo:
     gr.Markdown("## Civitai経由ファイルダウンロード＆アップロードツール")
     gr.Markdown("Civitai APIのURLを指定して、'model'または'lora'リポジトリへアップロードします。")
-    
+
     # ダウンロード＆アップロード機能
     url_input = gr.Textbox(
         label="CivitaiモデルAPIのURL",
-        placeholder="例: https://civitai.com/api/download/models/546416@685247?type=Model&format=SafeTensor"
+        placeholder="例: https://civitai.com/api/download/models/xxxxx?type=Model&format=SafeTensor"
     )
     repo_choice = gr.Radio(
         choices=["model", "lora"],
@@ -211,10 +79,11 @@ with gr.Blocks() as demo:
     run_button = gr.Button("実行")
     output = gr.Textbox(label="結果メッセージ", interactive=False)
 
+    # 修正点: outputsにoutputを指定し、閉じ括弧を追加
     run_button.click(
         download_and_upload,
         inputs=[url_input, repo_choice],
         outputs=output
     )
 
-demo.launch()
+demo.launch()