Update su.py

su.py

@@ -1,53 +1,28 @@
 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"])
-os.system("sudo apt-get install git-lfs")
-os.system("git lfs install")
-
-print("开始下载 stable-diffusion-webui.zip")
+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/workspace")
-#"https://hf-mirror.com/dong625/stable-diffusion-webui/resolve/main/stable-diffusion-webui.zip"
+aria2(url, "stable-diffusion-webui.zip", "/mnt")
 subprocess.run([
-    "7z", "x", "-pQQ917724495", "/mnt/workspace/stable-diffusion-webui.zip",
+    "7z", "x", "-pQQ917724495", "/mnt/stable-diffusion-webui.zip",
     "-o/mnt/workspace", "-y"
 ])
-os.remove("/mnt/workspace/stable-diffusion-webui.zip")
-
+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"
 ])
-
-def move_files():
-    source_files = [
-        "/mnt/workspace/stable-diffusion-webui/点我启动SD.ipynb",
-        "/mnt/workspace/stable-diffusion-webui/下载大模型点这里.ipynb",
-        "/mnt/workspace/stable-diffusion-webui/下载Lora点这里.ipynb",
-        "/mnt/workspace/stable-diffusion-webui/Photo",
-        "/mnt/workspace/stable-diffusion-webui/更多服务点我",
-    ]
-    destination_dir = "/mnt/workspace/"
-    for file_path in source_files:
-        if os.path.exists(file_path):
-            try:
-                shutil.move(file_path, destination_dir)
-                pass
-            except Exception as e:
-                pass
-        else:
-            pass
-
+subprocess.run(["python", "move.py"])
 if __name__ == "__main__":
-    move_files()
-    print("\033[32m\033[4mSD服务部署成功！请点击左侧界面启动SD服务！\033[0m")
+    print("\n\n\n\n\033[32m\033[4mSD服务部署成功！请点击左侧界面启动SD服务！\033[0m")
     os.remove(__file__)