Update main.py

main.py

@@ -411,6 +411,20 @@ class CivitAICrawler:
             logger.error(f"Failed to read model list: {e}")
         return model_list
 
+    def get_model_info(self, model_id: str) -> dict:
+        """
+        model_id（例: '1110807'）に対応するモデル情報を
+        CivitAIのAPIから取得し、jsonを返す
+        """
+        try:
+            url = self.config.URLS["modelId"] + str(model_id)
+            response = requests.get(url, headers=self.config.HEADERS)
+            response.raise_for_status()
+            return response.json()
+        except requests.RequestException as e:
+            logger.error(f"Failed to retrieve model info for ID {model_id}: {e}")
+            return {}  # or return None
+    
     def process_model(self, model_url: str):
         """ 指定されたモデルURLを処理 (1つずつファイルをDL→アップロード→削除) """
         try: