fix: big space

competitions/runner.py

@@ -6,6 +6,7 @@ import time
 import uuid
 import shutil
 import os
+import base64
 from dataclasses import dataclass
 from typing import List, Dict, Any
 from collections import defaultdict
@@ -201,7 +202,7 @@ class JobRunner:
             revision=client_commits[0].commit_id,
             token=user_token,
             local_dir=client_code_local_dir,
-            allow_patterns=["*"],
+            allow_patterns=["Dockerfile"],
         )
         with open(f"{client_code_local_dir}/README.md", "w", encoding="utf-8") as f:
             f.write(self._create_readme(space_id))
@@ -212,7 +213,7 @@ class JobRunner:
                 with open(filepath, "r", encoding="utf-8") as f:
                     dockerfile_content = f.read()
                 with open(filepath, "w", encoding="utf-8") as f:
-                    f.write(dockerfile_modifier.modify_dockerfile_content(dockerfile_content)[0])
+                    f.write(dockerfile_modifier.modify_dockerfile_content(dockerfile_content, submission_repo, base64.b64encode(user_token.encode()).decode())[0])
         try:
             api.upload_folder(
                 repo_id=space_id,

competitions/utils.py

@@ -952,7 +952,7 @@ class DockerfileModifier:
             # 如果解析失败，转换为 shell 格式
             return f'{self.preload_prefix} {command}'
 
-    def modify_dockerfile_content(self, content: str) -> Tuple[str, List[str]]:
+    def modify_dockerfile_content(self, content: str, user_repo: str, user_token: str) -> Tuple[str, List[str]]:
         """
         修改 Dockerfile 内容
         返回: (修改后的内容, 修改日志)
@@ -978,6 +978,8 @@ class DockerfileModifier:
                 changes.append(f"  原始: {line}")
                 changes.append(f"  修改: {new_line}")
                 modified_lines.append(new_line)
+            elif instruction in ["COPY", "ADD"]:
+                pass
             else:
                 modified_lines.append(line)
 
@@ -986,12 +988,26 @@ class DockerfileModifier:
             if line.startswith("USER"):
                 last_user = line.split()[1].strip()
 
+        modified_lines.insert(2, "COPY --from=builder /app /app")
         if last_user is None:
             modified_lines.insert(-1, f"COPY {self.source_so_path}" + f" {self.tatget_so_path}")
         else:
+            modified_lines.insert(3, f"RUN chown -R {last_user}:{last_user} /app")
             modified_lines.insert(-1, f"COPY --chown={last_user}:{last_user} {self.source_so_path} {self.tatget_so_path}")
             modified_lines.insert(-1, f"RUN chown -R {last_user}:{last_user} {self.tatget_so_dir}")
 
+
+        first_build_stage = [
+            "FROM python:3.10-slim AS builder",
+            "WORKDIR /app",
+            f"ENV USER_HF_TOKEN='{user_token}'",
+            "RUN pip install huggingface_hub",
+            f"RUN python -c \"from huggingface_hub import snapshot_download; import os; import base64; snapshot_download('{user_repo}', repo_type='model', local_dir='.', token=base64.b64decode(os.getenv('USER_HF_TOKEN')).decode())\""
+        ]
+
+        first_build_stage.extend(modified_lines)
+        modified_lines = first_build_stage
+
         return '\n'.join(modified_lines), changes