import os
import shutil
import subprocess
print("\033[33m欢迎使用阿里云SD一键部署服务！预计时长5分钟，请耐心等待\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"])
print("\n\n\n\n\033[33m开始下载 stable-diffusion-webui.zip\033[0m")
def aria2(url, filename, d):
    subprocess.run(["aria2c", "--console-log-level=error", "-c", "-x", "16", "-s", "16", url, "-o", filename, "-d", d])
url = "https://www.modelscope.cn/api/v1/models/dong625/stable-diffusion-webui/repo?Revision=master&FilePath=stable-diffusion-webui.zip"
aria2(url, "stable-diffusion-webui.zip", "/mnt")
subprocess.run([
    "7z", "x", "-pQQ917724495", "/mnt/stable-diffusion-webui.zip",
    "-o/mnt/workspace", "-y"
])
os.remove("/mnt/stable-diffusion-webui.zip")
os.chdir("/mnt/workspace/stable-diffusion-webui")
subprocess.run([
    "pip", "install", "-r", "requirements_versions.txt",
    "--extra-index-url", "https://download.pytorch.org/whl/cu117"
])
subprocess.run(["python", "move.py"])
if __name__ == "__main__":
    print("\n\n\n\n\033[32m\033[4mSD服务部署成功！请点击左侧界面启动SD服务！\033[0m")
    os.remove(__file__)