Update app.py

app.py

@@ -4,7 +4,6 @@ import re
 import requests
 import gradio as gr
 from huggingface_hub import HfApi
-from civitai import Civitai
 
 # 環境変数から取得
 HF_TOKEN = os.environ.get("HF_TOKEN", None)
@@ -15,9 +14,6 @@ CIVITAI_TOKEN = os.environ.get("CIVITAI_TOKEN", None)  # Civitai用トークン
 # Hugging Face APIの初期化
 api = HfApi()
 
-# Civitai APIの初期化
-civitai = Civitai(auth=CIVITAI_TOKEN)
-
 def extract_model_id(url):
     """
     CivitaiのダウンロードURLからモデルIDを抽出する関数。
@@ -30,23 +26,36 @@ def extract_model_id(url):
     else:
         return None
 
-def get_metadata_civitai(model_id):
+def get_metadata(model_id):
     """
-    civitai-pyを使用してCivitai APIからモデルのトリガーワードと推奨強度を取得する関数。
+    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:
-        model = civitai.models.get(model_id)
+        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 = model.trainedWords if hasattr(model, 'trainedWords') else "N/A"
+        # APIレスポンスのフィールド名を確認
+        # ここでは 'trainedWords' と仮定していますが、実際のレスポンスに合わせて修正してください
+        trigger_words = data.get('trainedWords', 'N/A')
         
         # 推奨強度の取得
-        # description フィールドから strength を抽出
-        description = model.description if hasattr(model, 'description') else ""
-        strength_match = re.search(r'strength:?\s*([\d.]+)', description, re.IGNORECASE)
+        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:
@@ -66,9 +75,13 @@ def download_and_upload(url, repo_type):
     if not model_id:
         return "URLからモデルIDを抽出できませんでした。URL形式を確認してください。"
     
+    headers = {}
+    if CIVITAI_TOKEN:
+        headers["Authorization"] = f"Bearer {CIVITAI_TOKEN}"
+    
     try:
         # ファイルのダウンロード
-        response = requests.get(url)
+        response = requests.get(url, headers=headers)
         response.raise_for_status()
     except Exception as e:
         return f"ファイルのダウンロードに失敗しました: {e}"
@@ -92,7 +105,7 @@ def download_and_upload(url, repo_type):
         return f"ファイルの保存に失敗しました: {e}"
     
     # メタデータの取得
-    strength, trigger_words = get_metadata_civitai(model_id)
+    strength, trigger_words = get_metadata(model_id)
     
     # メタデータのダウンロード
     metadata_filename = "metadata.json"