working on hf integration

demo.py → app.py

@@ -8,7 +8,7 @@ import pathlib
 import sys
 from main import process_face
 from PIL import Image
-
+from install import install
 PORT = 7860
 CACHE_DIR = "./cache"
 
@@ -149,17 +149,12 @@ def create_gradio_interface():
     demo.queue(max_size=99)
     
     try:
-        demo.launch(
-            share=True, 
-            server_name="0.0.0.0",
-            server_port=PORT,
-            quiet=True,
-            show_error=True,
-        )
+        demo.launch()
     except OSError as e:
         print(f"Error starting server: {e}")
         sys.exit(1)
 
 
 if __name__ == "__main__":
+    install()
     create_gradio_interface() 

install.py

@@ -4,7 +4,7 @@ import os
 BASE_PATH = "./"
 COMFYUI_PATH = os.path.join(BASE_PATH, "ComfyUI")
 MODEL_PATH = os.path.join(COMFYUI_PATH, "models")
-CACHE_PATH = "/workspace/huggingface_cache"
+CACHE_PATH = "/work/huggingface_cache"
 
 os.environ["HF_HOME"] = CACHE_PATH
 os.makedirs(CACHE_PATH, exist_ok=True)
@@ -213,9 +213,12 @@ def install_custom_nodes():
     print("✅ Installed and updated all ComfyUI nodes.")
 
 
-if __name__ == "__main__":
+def install():
     install_comfyui()
     install_custom_nodes()
     download_huggingface_models()
     download_and_extract_antelopev2()
-    print("🎉 Setup Complete! Run `run.py` to start ComfyUI.")
+    print("🎉 Setup Complete! Run `run.py` to start ComfyUI.")  
+
+if __name__ == "__main__":
+    install()

requirements.txt

@@ -1,8 +1,176 @@
-fastapi[standard]
-huggingface_hub[hf_transfer]
-comfy-cli
-python-dotenv
-requests
-fal-client
-gradio>=3.50.2
-pillow>=10.0.0
+aiofiles==24.1.0
+aiohappyeyeballs==2.6.1
+aiohttp==3.11.16
+aiosignal==1.3.2
+albucore==0.0.23
+albumentations==2.0.5
+annotated-types==0.7.0
+anyio==4.9.0
+arrow==1.3.0
+attrs==25.3.0
+av==14.3.0
+binaryornot==0.4.4
+certifi==2025.1.31
+cffi==1.17.1
+chardet==5.2.0
+charset-normalizer==3.4.1
+click==8.1.8
+coloredlogs==15.0.1
+comfy-cli==1.3.8
+comfyui_frontend_package==1.15.13
+contourpy==1.3.2
+cookiecutter==2.6.0
+cryptography==44.0.2
+cycler==0.12.1
+Cython==3.0.12
+Deprecated==1.2.18
+distro==1.9.0
+dnspython==2.7.0
+easydict==1.13
+einops==0.8.1
+email_validator==2.2.0
+facexlib==0.3.0
+fal_client==0.5.9
+fastapi==0.115.12
+fastapi-cli==0.0.7
+ffmpy==0.5.0
+filelock==3.18.0
+filterpy==1.4.5
+flatbuffers==25.2.10
+fonttools==4.57.0
+frozenlist==1.5.0
+fsspec==2025.3.2
+ftfy==6.3.1
+gitdb==4.0.12
+GitPython==3.1.44
+gradio==5.25.2
+gradio_client==1.8.0
+groovy==0.1.2
+h11==0.14.0
+hf_transfer==0.1.9
+httpcore==1.0.8
+httptools==0.6.4
+httpx==0.28.1
+httpx-sse==0.4.0
+huggingface-hub==0.30.2
+humanfriendly==10.0
+idna==3.10
+imageio==2.37.0
+insightface==0.7.3
+Jinja2==3.1.6
+jiter==0.9.0
+joblib==1.4.2
+kiwisolver==1.4.8
+kornia==0.8.0
+kornia_rs==0.1.8
+lazy_loader==0.4
+llvmlite==0.44.0
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+matplotlib==3.10.1
+matrix-client==0.4.0
+mdurl==0.1.2
+mixpanel==4.10.1
+mpmath==1.3.0
+multidict==6.4.3
+networkx==3.4.2
+numba==0.61.2
+numpy==2.2.4
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-cusparselt-cu12==0.6.2
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
+onnx==1.17.0
+onnxruntime==1.21.0
+onnxruntime-gpu==1.21.0
+openai==1.75.0
+opencv-python==4.11.0.86
+opencv-python-headless==4.11.0.86
+orjson==3.10.16
+packaging==24.2
+pandas==2.2.3
+pathspec==0.12.1
+pillow==11.2.1
+prettytable==3.16.0
+prompt_toolkit==3.0.51
+propcache==0.3.1
+protobuf==6.30.2
+psutil==7.0.0
+pycparser==2.22
+pydantic==2.11.3
+pydantic_core==2.33.1
+pydub==0.25.1
+PyGithub==2.6.1
+Pygments==2.19.1
+PyJWT==2.10.1
+PyNaCl==1.5.0
+pyparsing==3.2.3
+python-dateutil==2.9.0.post0
+python-dotenv==1.1.0
+python-multipart==0.0.20
+python-slugify==8.0.4
+pytz==2025.2
+PyYAML==6.0.2
+questionary==2.1.0
+regex==2024.11.6
+requests==2.32.3
+rich==14.0.0
+rich-toolkit==0.14.1
+ruff==0.11.5
+safehttpx==0.1.6
+safetensors==0.5.3
+scikit-image==0.25.2
+scikit-learn==1.6.1
+scipy==1.15.2
+semantic-version==2.10.0
+semver==3.0.4
+sentencepiece==0.2.0
+shellingham==1.5.4
+simsimd==6.2.1
+six==1.17.0
+smmap==5.0.2
+sniffio==1.3.1
+soundfile==0.13.1
+spandrel==0.4.1
+starlette==0.46.2
+stringzilla==3.12.4
+sympy==1.13.1
+text-unidecode==1.3
+threadpoolctl==3.6.0
+tifffile==2025.3.30
+timm==1.0.15
+tokenizers==0.21.1
+toml==0.10.2
+tomlkit==0.13.2
+torch==2.6.0
+torchaudio==2.6.0
+torchsde==0.2.6
+torchvision==0.21.0
+tqdm==4.67.1
+trampoline==0.1.2
+transformers==4.51.3
+triton==3.2.0
+typer==0.15.2
+types-python-dateutil==2.9.0.20241206
+typing-inspection==0.4.0
+typing_extensions==4.13.2
+tzdata==2025.2
+urllib3==1.26.20
+uv==0.6.14
+uvicorn==0.34.1
+uvloop==0.21.0
+watchfiles==1.0.5
+wcwidth==0.2.13
+websocket-client==1.8.0
+websockets==15.0.1
+wrapt==1.17.2
+yarl==1.20.0