Upload main.py

main.py

@@ -48,6 +48,35 @@ class Config:
     # 暗号化されたファイルが出力されるローカルディレクトリ（cryptLocal: の実体）
     ENCRYPTED_DIR = "/home/user/app/encrypted"
 
+class CivitAICrawler:
+    """CivitAIからモデルをダウンロードし、Hugging Faceにアップロードするクラス（フォルダ名も暗号化対応版）"""
+
+    def __init__(self, config: Config):
+        self.config = config
+        self.api = HfApi()
+        self.app = FastAPI()
+        self.repo_ids = self.config.REPO_IDS.copy()
+        self.jst = self.config.JST
+
+        # rclone のセットアップ（rclone.conf を環境変数から生成）
+        self.setup_rclone_conf()
+
+        self.setup_routes()
+
+    def setup_routes(self):
+        @self.app.get("/")
+        def read_root():
+            now = str(datetime.datetime.now(self.jst))
+            description = f"""
+            CivitAIを定期的に周回し新規モデルを {self.repo_ids['current']} にバックアップするSpaceです。
+            詳細は https://huggingface.co/{self.repo_ids['model_list']}/blob/main/model_list.log からどうぞ。
+            Status: {now} + currently running :D
+            """
+            return description
+
+        @self.app.on_event("startup")
+        async def startup_event():
+            asyncio.create_task(self.crawl())
 
     # =============================================================================
     # rclone まわり