Upload 2 files

zrok.py

@@ -0,0 +1,34 @@
+from multiprocessing import Process ,Queue #line:1
+import subprocess #line:2
+import sys #line:3
+import re #line:4
+def hitozuma (O0O0O00OO0O0O00O0 ,OO0OO00O0OOO000OO ):#line:6
+    try :#line:7
+        OO000O0O000OOOOO0 =subprocess .run (['/home/studio-lab-user/.zrok/bin/zrok','enable',O0O0O00OO0O0O00O0 ],check =False ,stdout =subprocess .PIPE ,stderr =subprocess .STDOUT ,text =True )#line:9
+        O0O0OOOOO0000OO00 =subprocess .Popen (["/home/studio-lab-user/.zrok/bin/zrok","share","public","localhost:8188","--headless"],stdout =subprocess .PIPE ,stderr =subprocess .STDOUT ,text =True )#line:12
+        O000OO0O000OOOOOO =re .compile (r'https?://[^\s]*\.zrok\.io')#line:14
+        OOO0O0OOO000O0OOO =("◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼◼")#line:15
+        if OO000O0O000OOOOO0 .returncode ==0 :#line:17
+            OO0OO00O0OOO000OO .put (f"\n{OOO0O0OOO000O0OOO}\n[ZROK] environment enabled.\n")#line:18
+        else :#line:20
+            if "enabled environment"in OO000O0O000OOOOO0 .stdout :#line:21
+                OO0OO00O0OOO000OO .put (f"\n{OOO0O0OOO000O0OOO}\n[ZROK] environment already enabled.\n")#line:22
+            else :#line:23
+                OO0OO00O0OOO000OO .put (OO000O0O000OOOOO0 .stdout )#line:24
+        for O0O00O000O0O0O0O0 in O0O0OOOOO0000OO00 .stdout :#line:26
+            OO0O00O0000O0OOO0 =O000OO0O000OOOOOO .findall (O0O00O000O0O0O0O0 )#line:27
+            for O0O0O00O0OO0O000O in OO0O00O0000O0OOO0 :#line:28
+                OO0OO00O0OOO000OO .put (f"[ZROK] {O0O0O00O0OO0O000O}\n{OOO0O0OOO000O0OOO}\n\n")#line:29
+        O0O0OOOOO0000OO00 .wait ()#line:31
+    except :#line:33
+        pass #line:34
+if __name__ =='__main__':#line:36
+    if len (sys .argv )!=2 :#line:37
+        sys .exit ("")#line:38
+    zrok_out =Queue ()#line:40
+    process =Process (target =hitozuma ,args =(sys .argv [1 ],zrok_out ))#line:42
+    process .start ()#line:43
+    while process .is_alive ()or not zrok_out .empty ():#line:45
+        while not zrok_out .empty ():#line:46
+            print (zrok_out .get (),end ='',flush =True )#line:47
+    process .join ()

zrun.py

@@ -0,0 +1,33 @@
+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 (OO0OO0O00OOO00000 ,O0O0O00OO0O0OO00O ):#line:8
+    subprocess .run (f"mkdir -p /tmp/models /tmp/Lora /tmp/ControlNet",shell =True ,stdout =subprocess .DEVNULL ,stderr =subprocess .DEVNULL )#line:13
+    try :#line:14
+        O0O0000O00O000000 =subprocess .Popen (['python','zrok.py',OO0OO0O00OOO00000 ],stdout =subprocess .PIPE ,stderr =subprocess .PIPE ,text =True )#line:15
+        O000000OOO000OOOO =subprocess .Popen (['python','main.py']+O0O0O00OO0O0OO00O ,stdout =sys .stdout ,stderr =sys .stdout ,text =True )#line:17
+        def OOO0OO000OO0000OO (O00OOOOO0OOOOOOOO ):#line:19
+            try :#line:20
+                if O00OOOOO0OOOOOOOO .stdout is not None :#line:21
+                    for O00000O00O00OOOO0 in O00OOOOO0OOOOOOOO .stdout :#line:22
+                        print (O00000O00O00OOOO0 .strip ())#line:23
+            except Exception :#line:25
+                pass #line:26
+        O0O0OO000O00OO000 =threading .Thread (target =OOO0OO000OO0000OO ,args =(O0O0000O00O000000 ,))#line:28
+        OO000O000OOO0O0OO =threading .Thread (target =OOO0OO000OO0000OO ,args =(O000000OOO000OOOO ,))#line:29
+        O0O0OO000O00OO000 .start ()#line:31
+        OO000O000OOO0O0OO .start ()#line:32
+        O0O0OO000O00OO000 .join ()#line:34
+        OO000O000OOO0O0OO .join ()#line:35
+    except KeyboardInterrupt :#line:37
+        pass #line:38
+    except Exception as O0OO000OO0OOO0OOO :#line:40
+        print ("Error:",str (O0OO000OO0OOO0OOO ))#line:41
+if __name__ =="__main__":#line:43
+    if len (sys .argv )<2 :#line:44
+        sys .exit (1 )#line:45
+    token =sys .argv [1 ]#line:47
+    launch_args =sys .argv [2 :]#line:48
+    zrok_launch (token ,launch_args )