Update app.py

app.py

@@ -11,6 +11,8 @@ import torchvision.transforms as transforms
 from PIL import Image
 import torch
 import dlib
+import uuid 
+uid=uuid.uuid4()
 #os.system("pip install dlib")
 os.system('bash setup.sh')
 
@@ -130,11 +132,11 @@ def run_cmd(command):
 def run(image,Restoration_mode, Colorizaition_mode):
     if 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))
         h,w,c=image.shape
         max = 600
@@ -147,45 +149,21 @@ def run(image,Restoration_mode, Colorizaition_mode):
         else:
             dim = (w,h)
             pass
-        '''
-        if width is None:
-            # calculate the ratio of the height and construct the
-            # dimensions
-            r = height / float(h)
-            dim = (int(w * r), height)
-    
-        # otherwise, the height is None
-        else:
-            # calculate the ratio of the width and construct the
-            # dimensions
-            r = width / float(w)
-            dim = (width, int(h * r))
-        '''
-
-
 
-
-        
-        #if h > 800:
-        #h1 = int(h/2)
-        #w1 = int(w/2)
         image=cv2.resize(image, dsize=(dim), interpolation=cv2.INTER_LANCZOS4)
-        #max_size=(800,800)
-        #image=image.thumbnail(max_size)
-        #image=np.array(image)
-        #image=cv2.imread(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")
+        result_restoration = Image.open(f"Temp{uid}/final_output/input_img.png")
         shutil.rmtree("Temp")
     
     elif Restoration_mode == "Pix2Pix":
@@ -202,8 +180,8 @@ with gr.Blocks() as app:
         gr.Column()
         with gr.Column():
             im = gr.Image(label="Input Image")
-            rad1 = gr.Radio(["BOPBTL", "Pix2Pix"])
-            rad2 = gr.Radio(["Deoldify", "Pix2Pix Resnet 9block","Pix2Pix Unet 256"])
+            rad1 = gr.Radio(["BOPBTL", "Pix2Pix"], value="BOPBTL")
+            rad2 = gr.Radio(["Deoldify", "Pix2Pix Resnet 9block","Pix2Pix Unet 256"],value="Deoldify")
                 
             im_btn=gr.Button(label="Restore")
             out_im = gr.Image(label="Restored Image")