new: argument to identify different deploying mode

.github/tools/deploy_to_space.py

@@ -32,10 +32,14 @@ if __name__ == "__main__":
     parser.add_argument("--branch_name", required=True)
     parser.add_argument("--username", required=True)
     parser.add_argument("--user_token", required=True)
+    parser.add_argument("--mode", required=True, choices=["prod", "dev"])
     args = parser.parse_args()
 
-    # TODO: Use different space config when deploy to dev environment.
-    space = Space(PRODUCTION_SPACE_CONFIG["default"])
+    if args.mode == "prod":
+        space = Space(PRODUCTION_SPACE_CONFIG["default"])
+    elif args.mode == "dev":
+        # TODO: Use dev space config when deploy to dev environment.
+        space = Space(PRODUCTION_SPACE_CONFIG["default"])
 
     push_result = force_push_to_remote(
         space, args.username, args.user_token, args.branch_name

.github/tools/generate_readme.py

@@ -1,3 +1,4 @@
+import argparse
 import sys
 
 from config import PRODUCTION_SPACE_CONFIG
@@ -22,7 +23,17 @@ Check out the configuration reference at https://huggingface.co/docs/hub/spaces-
 
 
 if __name__ == "__main__":
-    space = Space(PRODUCTION_SPACE_CONFIG["default"])
+    parser = argparse.ArgumentParser(
+        description="Generate Hugging Face space config file (i.e. README.md)."
+    )
+    parser.add_argument("--mode", required=True, choices=["prod", "dev"])
+    args = parser.parse_args()
+
+    if args.mode == "prod":
+        space = Space(PRODUCTION_SPACE_CONFIG["default"])
+    elif args.mode == "dev":
+        # TODO: Use dev space config when deploy to dev environment.
+        space = Space(PRODUCTION_SPACE_CONFIG["default"])
 
     readme = readme_template(space.readme)
     print(readme)

.github/workflows/deploy-to-production.yaml

@@ -22,7 +22,7 @@ jobs:
           fetch-depth: '0'
       - name: Update Repo README.md
         run: |
-          python3 .github/tools/generate_readme.py > README.md
+          python3 .github/tools/generate_readme.py --mode="prod" > README.md
           git config --global user.email "github-actions[bot]@users.noreply.github.com"
           git config --global user.name "GitHub Actions[bot]"
           git add ./README.md
@@ -30,4 +30,9 @@ jobs:
             git commit -m "[skip ci] update README.md"
           fi
       - name: Run deploy tool
-        run: python3 .github/tools/deploy_to_space.py --branch_name="${{ github.ref_name }}" --username="${{ vars.HF_USERNAME }}" --user_token="${{ secrets.HF_TOKEN }}"
+        run: |
+          python3 .github/tools/deploy_to_space.py \
+            --mode="prod" \
+            --branch_name="${{ github.ref_name }}" \
+            --username="${{ vars.HF_USERNAME }}" \
+            --user_token="${{ secrets.HF_TOKEN }}"