update

app.py

@@ -1,7 +1,102 @@
+import os
 import gradio as gr
+import sys 
+sys.path.append(".")
 
-def greet(name):
-    return "Hello " + name + "!!"
+#@title Import stuff
+import gc
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+import subprocess
+import shutil
+from PIL import Image
+import time
+
+import imageio
+
+
+# def run(initial_image, mask, Backgrounds, Backgrounds_complexity, Size, Angle, Steps, num_of_Images):
+def run(source_img, Backgrounds, Backgrounds_complexity, Size, Angle, Steps, num_of_Images):
+    print('-------------------starting to process-------------------')
+    if os.path.exists('results'):
+        shutil.rmtree("results")
+    if os.path.exists('tmp'):
+        shutil.rmtree("tmp")
+    time.sleep(1)
+    os.makedirs('results', exist_ok=True)
+    os.makedirs('tmp/img', exist_ok=True)
+    os.makedirs('tmp/mask', exist_ok=True)
+    os.makedirs('tmp/bg', exist_ok=True)
+
+    '''
+    print('-----initial_image: ', initial_image)
+    init_image = Image.open(initial_image)
+    mask = Image.open(mask)
+    init_image = init_image.resize((256,256))
+    mask = mask.resize((256,256))
+    init_image.save("tmp/img/input.JPEG")
+    mask.save("tmp/mask/input.png")
+    '''
+    imageio.imwrite("tmp/img/input.JPEG", source_img["image"])
+    imageio.imwrite("tmp/mask/input.png", source_img["mask"])
+    
+    initial_image = Image.open('tmp/img/input.JPEG').resize((256,256))
+    initial_image.save('tmp/img/input.JPEG')
+    mask = Image.open('tmp/mask/input.png').resize((256,256))
+    mask.save('tmp/mask/input.png')
+
+
+    if Backgrounds:
+        background_specific = Backgrounds
+        if background_specific is not None:
+            background_specific = Image.open(background_specific).convert('RGB') # Specified background
+            background_specific = background_specific.resize((256,256))
+            background_specific.save('tmp/bg/bg.png')
+            background_specific = '../tmp/bg/bg.png'
+    else:
+        background_specific = ""
+
+    Backgrounds_complexity = Backgrounds_complexity
+    Size = Size
+    Angle = Angle
+    Steps = Steps
+    num_of_Images = num_of_Images
+    print(Backgrounds_complexity, background_specific, Size, Angle, Steps, num_of_Images)
+    p = subprocess.Popen(["sh", "run.sh", str(Backgrounds_complexity), background_specific, str(Size), str(Angle), str(Steps), str(num_of_Images)])
+    
+    # subprocess.Popen(["cd", "object_removal/TFill/"])
+    # subprocess.Popen(["python", "test.py"])
+    
+    return_code = p.wait()
+    print('----return_code: ', return_code)
+
+    if os.path.exists('results/edited.png'):
+        return Image.open('results/edited.png')
+    else:
+        return Image.open('tmp/img/input.JPEG')
+
+
+image = gr.outputs.Image(type="pil", label="Your result")
+css = ".output-image{height: 528px !important} .output-carousel .output-image{height:272px !important} a{text-decoration: underline}"
+iface = gr.Interface(fn=run, inputs=[
+    # gr.inputs.Image(type="filepath", label='initial_image'),
+    gr.Image(source="upload", type="numpy", tool="sketch", elem_id="source_container"),
+    # gr.inputs.Image(type="filepath", label='mask - object mask', optional=True),
+    gr.inputs.Image(type="filepath", label='Backgrounds - optional, specified backgrounds'),
+    gr.inputs.Slider(label="Backgrounds_complexity - How complicated you wish to the generated image to be", default=0, step=1, minimum=-30, maximum=30),
+    gr.inputs.Slider(label="Size - Object pixel rates", default=0.1, step=0.02, minimum=0.01, maximum=0.5),
+    gr.inputs.Slider(label="Angle - Object angle", default=0, step=10, minimum=-180, maximum=180),
+    gr.inputs.Slider(label="Steps - more steps can increase quality but will take longer to generate",default=10,maximum=100,minimum=1,step=1),
+    gr.inputs.Slider(label="num_of_Images - How many images you wish to generate", default=2, step=1, minimum=1, maximum=4),
+
+    # gr.inputs.Radio(label="Width", choices=[32,64,128,256],default=256),
+    # gr.inputs.Radio(label="Height", choices=[32,64,128,256],default=256),
+    # gr.inputs.Textbox(label="Prompt - try adding increments to your prompt such as 'oil on canvas', 'a painting', 'a book cover'",default="chalk pastel drawing of a dog wearing a funny hat"),
+    #gr.inputs.Slider(label="ETA - between 0 and 1. Lower values can provide better quality, higher values can be more diverse",default=0.0,minimum=0.0, maximum=1.0,step=0.1),
+    ], 
+    # outputs=[image,gr.outputs.Carousel(label="Individual images",components=["image"]),gr.outputs.Textbox(label="Error")],
+    outputs=["image"],
+    css=css,
+    title="Image Editing with Controls of Object Attributes including Backgrounds, Sizes, Positions and Directions",
+    description="Gradio Demo for Image Editing with Controls of Object Attributes including Backgrounds, Sizes, Positions and Directions",
+    article="ImageNet-E")
+iface.launch(enable_queue=True)

app_.py

@@ -1,102 +0,0 @@
-import os
-import gradio as gr
-import sys 
-sys.path.append(".")
-
-#@title Import stuff
-import gc
-
-import subprocess
-import shutil
-from PIL import Image
-import time
-
-import imageio
-
-
-# def run(initial_image, mask, Backgrounds, Backgrounds_complexity, Size, Angle, Steps, num_of_Images):
-def run(source_img, Backgrounds, Backgrounds_complexity, Size, Angle, Steps, num_of_Images):
-    print('-------------------starting to process-------------------')
-    if os.path.exists('results'):
-        shutil.rmtree("results")
-    if os.path.exists('tmp'):
-        shutil.rmtree("tmp")
-    time.sleep(1)
-    os.makedirs('results', exist_ok=True)
-    os.makedirs('tmp/img', exist_ok=True)
-    os.makedirs('tmp/mask', exist_ok=True)
-    os.makedirs('tmp/bg', exist_ok=True)
-
-    '''
-    print('-----initial_image: ', initial_image)
-    init_image = Image.open(initial_image)
-    mask = Image.open(mask)
-    init_image = init_image.resize((256,256))
-    mask = mask.resize((256,256))
-    init_image.save("tmp/img/input.JPEG")
-    mask.save("tmp/mask/input.png")
-    '''
-    imageio.imwrite("tmp/img/input.JPEG", source_img["image"])
-    imageio.imwrite("tmp/mask/input.png", source_img["mask"])
-    
-    initial_image = Image.open('tmp/img/input.JPEG').resize((256,256))
-    initial_image.save('tmp/img/input.JPEG')
-    mask = Image.open('tmp/mask/input.png').resize((256,256))
-    mask.save('tmp/mask/input.png')
-
-
-    if Backgrounds:
-        background_specific = Backgrounds
-        if background_specific is not None:
-            background_specific = Image.open(background_specific).convert('RGB') # Specified background
-            background_specific = background_specific.resize((256,256))
-            background_specific.save('tmp/bg/bg.png')
-            background_specific = '../tmp/bg/bg.png'
-    else:
-        background_specific = ""
-
-    Backgrounds_complexity = Backgrounds_complexity
-    Size = Size
-    Angle = Angle
-    Steps = Steps
-    num_of_Images = num_of_Images
-    print(Backgrounds_complexity, background_specific, Size, Angle, Steps, num_of_Images)
-    p = subprocess.Popen(["sh", "run.sh", str(Backgrounds_complexity), background_specific, str(Size), str(Angle), str(Steps), str(num_of_Images)])
-    
-    # subprocess.Popen(["cd", "object_removal/TFill/"])
-    # subprocess.Popen(["python", "test.py"])
-    
-    return_code = p.wait()
-    print('----return_code: ', return_code)
-
-    if os.path.exists('results/edited.png'):
-        return Image.open('results/edited.png')
-    else:
-        return Image.open('tmp/img/input.JPEG')
-
-
-image = gr.outputs.Image(type="pil", label="Your result")
-css = ".output-image{height: 528px !important} .output-carousel .output-image{height:272px !important} a{text-decoration: underline}"
-iface = gr.Interface(fn=run, inputs=[
-    # gr.inputs.Image(type="filepath", label='initial_image'),
-    gr.Image(source="upload", type="numpy", tool="sketch", elem_id="source_container"),
-    # gr.inputs.Image(type="filepath", label='mask - object mask', optional=True),
-    gr.inputs.Image(type="filepath", label='Backgrounds - optional, specified backgrounds'),
-    gr.inputs.Slider(label="Backgrounds_complexity - How complicated you wish to the generated image to be", default=0, step=1, minimum=-30, maximum=30),
-    gr.inputs.Slider(label="Size - Object pixel rates", default=0.1, step=0.02, minimum=0.01, maximum=0.5),
-    gr.inputs.Slider(label="Angle - Object angle", default=0, step=10, minimum=-180, maximum=180),
-    gr.inputs.Slider(label="Steps - more steps can increase quality but will take longer to generate",default=10,maximum=100,minimum=1,step=1),
-    gr.inputs.Slider(label="num_of_Images - How many images you wish to generate", default=2, step=1, minimum=1, maximum=4),
-
-    # gr.inputs.Radio(label="Width", choices=[32,64,128,256],default=256),
-    # gr.inputs.Radio(label="Height", choices=[32,64,128,256],default=256),
-    # gr.inputs.Textbox(label="Prompt - try adding increments to your prompt such as 'oil on canvas', 'a painting', 'a book cover'",default="chalk pastel drawing of a dog wearing a funny hat"),
-    #gr.inputs.Slider(label="ETA - between 0 and 1. Lower values can provide better quality, higher values can be more diverse",default=0.0,minimum=0.0, maximum=1.0,step=0.1),
-    ], 
-    # outputs=[image,gr.outputs.Carousel(label="Individual images",components=["image"]),gr.outputs.Textbox(label="Error")],
-    outputs=["image"],
-    css=css,
-    title="Image Editing with Controls of Object Attributes including Backgrounds, Sizes, Positions and Directions",
-    description="Gradio Demo for Image Editing with Controls of Object Attributes including Backgrounds, Sizes, Positions and Directions",
-    article="ImageNet-E")
-iface.launch(enable_queue=True,share=True)