File size: 2,859 Bytes
7c34b66 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 | import os
import sys
def main():
root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
os.chdir(root_dir)
print("--- Starting Deployment Package Check ---")
errors = []
# 1. Required files
required_files = [
"Dockerfile",
".dockerignore",
"requirements.txt",
"README.md",
"model.joblib",
"model_metadata.json",
"predict.py",
"features.py",
"backend/app.py",
"frontend/package.json"
]
for f in required_files:
if not os.path.exists(f):
errors.append(f"Missing required file/folder: {f}")
# 2. Check README.md for Hugging Face Frontmatter
if os.path.exists("README.md"):
with open("README.md", "r", encoding="utf-8") as f:
content = f.read()
if "sdk: docker" not in content or "app_port: 7860" not in content:
errors.append("README.md is missing 'sdk: docker' or 'app_port: 7860' YAML frontmatter.")
# 3. Check Dockerfile
if os.path.exists("Dockerfile"):
with open("Dockerfile", "r", encoding="utf-8") as f:
content = f.read()
if "EXPOSE 7860" not in content:
errors.append("Dockerfile does not contain 'EXPOSE 7860'.")
if "7860" not in content:
errors.append("Dockerfile does not seem to bind uvicorn to port 7860.")
# 4. Check requirements.txt
if os.path.exists("requirements.txt"):
with open("requirements.txt", "r", encoding="utf-8") as f:
content = f.read()
if "opencv-python-headless" not in content:
errors.append("requirements.txt must use 'opencv-python-headless' instead of 'opencv-python'.")
# 5. Check for forbidden large files (we won't fail strictly on zip if it's outside the build context, but good to warn)
forbidden_dirs = [
"dataset/Data",
"node_modules",
"venv",
".venv"
]
for d in forbidden_dirs:
if os.path.exists(d) and os.path.isdir(d):
print(f"[WARNING] Large directory exists locally: {d} (Ensure it is in .dockerignore!)")
if os.path.exists(".dockerignore"):
with open(".dockerignore", "r", encoding="utf-8") as f:
content = f.read()
if "dataset" not in content:
errors.append(".dockerignore must ignore 'dataset' folder.")
if "node_modules" not in content:
errors.append(".dockerignore must ignore 'node_modules' folder.")
if errors:
print("\n[FAIL] Deployment check failed with the following errors:")
for e in errors:
print(f" - {e}")
sys.exit(1)
else:
print("\n[PASS] Deployment package checks passed! Ready for Hugging Face Spaces.")
if __name__ == "__main__":
main()
|