import os
import shutil
import subprocess

print("\033[36m\033[4mFooocus部署服务启动中，下载模型共65G，预计部署时长为50分钟\033[0m")
subprocess.run(["python3", "-m", "pip", "install", "--upgrade", "pip"])
subprocess.run(["apt", "update"])
subprocess.run(["apt", "install", "-y", "aria2"])
subprocess.run(["apt-get", "install", "sudo"])
subprocess.run(["sudo", "apt", "install", "p7zip-full", "-y"])
os.system("sudo apt-get install git-lfs")
os.system("git lfs install")

def aria2_download(url, filename, d):
    os.system(f"aria2c --console-log-level=error -c -x 16 -s 16 {url} -o {filename} -d {d}")

def main():
    url = "https://hf-mirror.com/dong625/Fooocus/resolve/main/Fooocus.zip"
    # url = "https://modelscope.cn/api/v1/models/dong625/Fooocus/repo?Revision=master&FilePath=Fooocus.zip
    aria2_download(url, "Fooocus.zip", "/mnt/workspace/")
    os.system("7z x -pQQ917724495 /mnt/workspace/Fooocus.zip -o/mnt/workspace -y")
    os.remove("/mnt/workspace/Fooocus.zip")
    os.chdir("/mnt/workspace/Fooocus")
    os.system("pip install -r requirements_versions.txt")
    print("Environment success")
    subprocess.run(["python", "Predownload.py"])
    subprocess.run(["python", "move.py"])

if __name__ == "__main__":
    main()
    print("\033[32m\033[4mFooocus服务部署成功！请点击左侧界面启动Fooocus服务！\033[0m")
    os.remove(__file__)