Update zrun.py

zrun.py

@@ -1,51 +1,33 @@
-import subprocess
-import threading
-import sys
-import os
-
-os.environ['LD_PRELOAD'] = '/home/studio-lab-user/.conda/envs/default/lib/libtcmalloc_minimal.so.4'
-
-def zrok_launch(token, launch_args):
-    
-    subprocess.run(
-        f"mkdir -p tmp_models /tmp/ControlNet",
-        shell=True,
-        stdout=subprocess.DEVNULL,
-        stderr=subprocess.DEVNULL)
-    
-    try:
-        zrok_process = subprocess.Popen(['python', 'zrok.py', token], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
-        launch_process = subprocess.Popen(['python', 'webui.py'] + launch_args,
-                                           stdout=sys.stdout, stderr=sys.stdout, text=True)
-
-        def capture_output(process):
-            try:
-                if process.stdout is not None:
-                    for line in process.stdout:
-                        print(line.strip())
-                        
-            except Exception:
-                pass
-
-        zrok_thread = threading.Thread(target=capture_output, args=(zrok_process,))
-        launch_thread = threading.Thread(target=capture_output, args=(launch_process,))
-
-        zrok_thread.start()
-        launch_thread.start()
-
-        zrok_thread.join()
-        launch_thread.join()
-
-    except KeyboardInterrupt:
-        pass
-    
-    except Exception as e:
-        print("Error:", str(e))
-
-if __name__ == "__main__":
-    if len(sys.argv) < 2:
-        sys.exit(1)
-    
-    token = sys.argv[1]
-    launch_args = sys.argv[2:]
-    zrok_launch(token, launch_args)
+import subprocess #line:1
+import threading #line:2
+import sys #line:3
+import os #line:4
+os .environ ['LD_PRELOAD']='/home/studio-lab-user/.conda/envs/default/lib/libtcmalloc_minimal.so.4'#line:6
+def zrok_launch (O00OO00000O0OOOO0 ,OO0O0O0OOO00OOO0O ):#line:8
+    subprocess .run (f"mkdir -p tmp_models /tmp/ControlNet",shell =True ,stdout =subprocess .DEVNULL ,stderr =subprocess .DEVNULL )#line:14
+    try :#line:16
+        O0OOO0OOO0OOOOO00 =subprocess .Popen (['python','zrok.py',O00OO00000O0OOOO0 ],stdout =subprocess .PIPE ,stderr =subprocess .PIPE ,text =True )#line:17
+        O00O0O00O0O000000 =subprocess .Popen (['python','webui.py']+OO0O0O0OOO00OOO0O ,stdout =sys .stdout ,stderr =sys .stdout ,text =True )#line:19
+        def O0O0000OO00000000 (OOO0OO0000OO00000 ):#line:21
+            try :#line:22
+                if OOO0OO0000OO00000 .stdout is not None :#line:23
+                    for OOOO00OO000OO0OO0 in OOO0OO0000OO00000 .stdout :#line:24
+                        print (OOOO00OO000OO0OO0 .strip ())#line:25
+            except Exception :#line:27
+                pass #line:28
+        OOO00OOO0OO0000OO =threading .Thread (target =O0O0000OO00000000 ,args =(O0OOO0OOO0OOOOO00 ,))#line:30
+        O00OOO0OOO0O000O0 =threading .Thread (target =O0O0000OO00000000 ,args =(O00O0O00O0O000000 ,))#line:31
+        OOO00OOO0OO0000OO .start ()#line:33
+        O00OOO0OOO0O000O0 .start ()#line:34
+        OOO00OOO0OO0000OO .join ()#line:36
+        O00OOO0OOO0O000O0 .join ()#line:37
+    except KeyboardInterrupt :#line:39
+        pass #line:40
+    except Exception as OO0O0O0O0OOOOOOOO :#line:42
+        print ("Error:",str (OO0O0O0O0OOOOOOOO ))#line:43
+if __name__ =="__main__":#line:45
+    if len (sys .argv )<2 :#line:46
+        sys .exit (1 )#line:47
+    token =sys .argv [1 ]#line:49
+    launch_args =sys .argv [2 :]#line:50
+    zrok_launch (token ,launch_args )