Update main.py

main.py

@@ -155,14 +155,35 @@ class CivitAICrawler:
                             logger.error("Failed after %d attempts: %s", max_retries, enc_file_path)
                             raise
 
-    def upload_folder_encrypted(self, folder_path: str, repo_id: Optional[str] = None, path_in_repo: str = ""):
-        if not repo_id:
+    def upload_folder_encrypted(self, folder_path: str, repo_id: Optional[str] = None, path_in_repo: str = "") -> str:
+        """フォルダを暗号化してアップロードし、暗号化後のトップフォルダ名を返す。"""
+        if repo_id is None:
             repo_id = self.repo_ids['current']
+        
+        # rclone でローカル -> cryptLocal:
         self.encrypt_with_rclone(folder_path)
-        self.upload_encrypted_files(repo_id, path_in_repo=path_in_repo)
+
+        # 暗号化先ディレクトリ直下を見て、暗号化後のトップレベルフォルダ名を取得
+        # （rclone の directory_name_encryption=standard 等ならランダム文字列になっているはず）
+        top_levels = [
+            d for d in os.listdir(self.config.ENCRYPTED_DIR)
+            if os.path.isdir(os.path.join(self.config.ENCRYPTED_DIR, d))
+        ]
+        if not top_levels:
+            raise RuntimeError("No top-level folder found after encryption.")
+        # もし複数あったら最初の1つだけ使う
+        encrypted_top_name = top_levels[0]
+
+        # Upload
+        self.upload_encrypted_files(repo_id, path_in_repo)
+
+        # 後片付け
         if os.path.isdir(self.config.ENCRYPTED_DIR):
             shutil.rmtree(self.config.ENCRYPTED_DIR, ignore_errors=True)
 
+        # 呼び出し元がURLを作れるよう返す
+        return encrypted_top_name
+
     def upload_file_encrypted_one_by_one(self, file_path: str, repo_id: Optional[str] = None, path_in_repo: str = ""):
         """古いバージョン用: ファイルを暗号化アップロード → ローカル削除。"""
         if not repo_id:
@@ -409,11 +430,11 @@ class CivitAICrawler:
             encrypted_top_name = self.upload_folder_encrypted(folder)
             shutil.rmtree(folder, ignore_errors=True)
 
-            # model_list.log に書き込み (例: "111123: https://huggingface.co/xxx/tree/main")
-            #   → 実際にフォルダ暗号化でトップレベル名が変わってるはずなので、
-            #      ざっくり "/tree/main" を書く
-            hf_url = f"https://huggingface.co/{self.repo_ids['current']}/tree/main/{encrypted_top_name}"
-            self.append_model_list(model_id, hf_url)
+            # ここで、暗号化後のトップフォルダ名をURLに含める
+            model_hf_url = f"https://huggingface.co/{self.repo_ids['current']}/tree/main/{encrypted_top_name}"
+
+            # あるいは、さらにサブフォルダ構造があるなら適宜URLを調整
+            self.append_model_list(model_id, model_hf_url)
 
         except Exception as e:
             logger.error("Unexpected error in process_model(%s): %s", model_url, e)