Update app.py

app.py

@@ -10,6 +10,7 @@ from skimage import color
 import torchvision.transforms as transforms
 from PIL import Image
 import torch
+import uuid
 
 os.system("pip install dlib")
 os.system('bash setup.sh')
@@ -124,32 +125,34 @@ def run_cmd(command):
         sys.exit(1)
 
 def run(image,Restoration_mode, Colorizaition_mode):
+    uid = uuid.uuid4()
     Restoration_mode == "BOPBTL"
-    if os.path.isdir("Temp"):
-        shutil.rmtree("Temp")
+    if os.path.isdir(f"Temp{uid}"):
+        shutil.rmtree(f"Temp{uid}")
     
-    os.makedirs("Temp")
-    os.makedirs("Temp/input")
+    os.makedirs(f"Temp{uid}")
+    os.makedirs(f"Temp{uid}/input")
     print(type(image))
-    cv2.imwrite("Temp/input/input_img.png", image)
+    cv2.imwrite(f"Temp{uid}/input/input_img.png", image)
 
     command = ("python run.py --input_folder "
-            + "Temp/input"
+            + f"Temp{uid}/input"
             + " --output_folder "
-            + "Temp"
+            + f"Temp{uid}"
             + " --GPU "
             + "-1"
             + " --with_scratch")
     run_cmd(command)
 
-    result_restoration = Image.open("Temp/final_output/input_img.png")
-    shutil.rmtree("Temp")
+    result_restoration = Image.open(f"Temp{uid}/final_output/input_img.png")
+    shutil.rmtree(f"Temp{uid}")
     
     result_colorization = inferColorization(result_restoration)
 
     return result_colorization
 with gr.Blocks() as app:
-    im = gr.Image(label="Input Image")
+    #im = gr.Image(label="Input Image")
+    im = gr.Textbox()
     im_btn=gr.Button(label="Restore")
     out_im = gr.Image(label="Restored Image")
     im_btn.click(run,im,out_im)