Update sd_yun/launch_en.py

sd_yun/launch_en.py

@@ -9,6 +9,16 @@ import subprocess
 import cloudpickle as pickle
 from datetime import timedelta
 from IPython.display import clear_output
+import psutil 
+def is_process_running(script_name):
+    """检查脚本是否正在运行"""
+    for proc in psutil.process_iter(['pid', 'name', 'cmdline']):
+        try:
+            if proc.info['name'] == 'python' and script_name in ' '.join(proc.info['cmdline']):
+                return True
+        except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
+            pass
+    return False
 
 
 # Setup Env
@@ -113,7 +123,15 @@ with tunnel:
         commandline_arguments += ' --xformers'
     else:
         commandline_arguments += ' --disable-xformers --opt-sdp-attention --cuda-stream --pin-shared-memory'
-    get_ipython().system('COMMANDLINE_ARGS="{commandline_arguments}" python upload.py &> /dev/null &')
+    process_name = "python upload.py"  
+    try:
+        subprocess.check_output(["pgrep", "-f", process_name])
+        print("Upload 脚本已经在运行中")
+    except subprocess.CalledProcessError:
+        # 脚本没有运行，启动它
+        print("启动 Upload 脚本")
+        subprocess.Popen(["nohup", "python", "./upload.py", ">", "/dev/null", "2>&1", "&"])
+    # get_ipython().system('COMMANDLINE_ARGS="{commandline_arguments}" python upload.py &> /dev/null &')
     get_ipython().system('COMMANDLINE_ARGS="{commandline_arguments}" python launch.py')
 
 start_colab = float(open(f'{webui_path}/static/colabTimer.txt', 'r').read())