feat: pull or clone

app.py

@@ -1,37 +1,65 @@
 import os
+import shutil
 import git
 import subprocess
-# from python-dotenv import load_dotenv
-from git.exc import GitCommandError
 
+REPO_DIR = 'repo'
+USER = os.getenv('HF_USER')
+REPO_URL = os.getenv('HF_REPO_URL')
+ACCESS_TOKEN = os.getenv('HF_ACCESS_TOKEN')
+AUTH_REPO_URL = REPO_URL.replace('https://', f'https://{USER}:{ACCESS_TOKEN}@')
 
-def clone_and_run():
-    # 設定環境變數中的倉庫地址和憑證
-    user = os.getenv('HF_USER')
-    repo_url = os.getenv('HF_REPO_URL')
-    access_token = os.getenv('HF_ACCESS_TOKEN')
 
-    if not repo_url or not access_token:
-        raise EnvironmentError("請設定環境變數中的 HUGGING_FACE_REPO_URL 和 HUGGING_FACE_ACCESS_TOKEN")
+def check_envs():
+    envs = [USER, REPO_URL, ACCESS_TOKEN]
+    missing_vars = [var for var in envs if var is None]
 
-    # 構建帶有憑證的 URL
-    auth_repo_url = repo_url.replace('https://', f'https://{user}:{access_token}@')
+    if missing_vars:
+        raise EnvironmentError(f"Missing environment variables: {', '.join(missing_vars)}")
 
+
+def pull() -> int:
     try:
-        # 克隆倉庫
-        git.Repo.clone_from(auth_repo_url, 'repo', branch='main')
-        print("倉庫克隆成功！")
+        repo = git.Repo(REPO_DIR)
+        repo.remotes.origin.pull()
+        return 200
+    except git.exc.InvalidGitRepositoryError:
+        # 404
+        raise ValueError('Invalid Git repository.')
+
+    except git.exc.NoSuchPathError:
+        return 404
+    
+    except git.ecx.GitCommandError as e:
+        print(f"拉取代碼時出錯：{e}")
+        return 500
+
+
+def clone(status_code):
+    if status_code == 200:
+        return
+    if status_code == 500:
+        if os.path.exists(REPO_DIR):
+            shutil.rmtree(REPO_DIR)
+        
+    try:
+        git.Repo.clone_from(AUTH_REPO_URL, REPO_DIR, branch='main')
+    except git.exc.GitCommandError as e:
+        print(f"克隆代碼時出錯：{e}")
+        raise e
+
+
+def run():
+    os.chdir(os.path.join(os.getcwd(), REPO_DIR))
+    subprocess.run(["python", "app.py"], check=True)
 
-        os.chdir(os.path.join(os.getcwd(), 'repo'))
-        subprocess.run(["python", "app.py"], check=True)
 
-        print("app.py 運行成功！")
+def main():
+    check_envs()
+    status_code = pull()
+    clone(status_code)
+    run()
 
-    except GitCommandError as e:
-        print(f"克隆倉庫時出錯：{e}")
-    except subprocess.CalledProcessError as e:
-        print(f"運行 app.py 時出錯：{e}")
 
 if __name__ == "__main__":
-    # load_dotenv()
-    clone_and_run()
+    main()

docker-compose.yml

@@ -8,15 +8,11 @@ services:
     - nginx-bridge
     environment:
     - CUDA_VISIBLE_DEVICES=all
-    # - LETSENCRYPT_HOST=vits-hat-tts.gohakka.org
-    # - VIRTUAL_HOST=vits-hat-tts.gohakka.org
     - VIRTUAL_PORT=7860
     ports:
     - 7862:7860
-#    expose:
- #   - "7860"
-#    env_file:
-#    - .env
+    env_file:
+    - .env
 
 networks:
   nginx-bridge:

requirements.txt

@@ -4,6 +4,7 @@ jieba
 opencc
 TTS
 omegaconf
+spaces
 # interface
 python-dotenv
 pyjwt