Update app.py

app.py

@@ -220,25 +220,12 @@ image_mask_list.sort()
 
 
 @spaces.GPU
-def run_local(base_image, base_mask, reference_image, ref_mask, seed, base_mask_option, ref_mask_option, text_prompt):
-
-
-    if base_mask_option == "Draw Mask":
-        tar_image = base_image["background"]
-        tar_mask = base_image["layers"][0]
-    else:
-        tar_image = base_image["background"]
-        tar_mask = base_mask["background"]
-
-    if ref_mask_option == "Draw Mask":
-        ref_image = reference_image["background"]
-        ref_mask = reference_image["layers"][0]
-    elif ref_mask_option == "Upload with Mask":
-        ref_image = reference_image["background"]
-        ref_mask = ref_mask["background"]
-    else:
-        ref_image = reference_image["background"]
-        ref_mask = get_mask(ref_image, text_prompt)
+def run_local(base_image, base_mask, reference_image, ref_mask):
+    seed = 666
+    tar_image = base_image
+    tar_mask = base_mask
+    ref_image = reference_image
+    ref_mask = ref_mask
 
     tar_image = tar_image.convert("RGB")
     tar_mask = tar_mask.convert("L")
@@ -348,7 +335,7 @@ def run_local(base_image, base_mask, reference_image, ref_mask, seed, base_mask_
     edited_image = Image.fromarray(edited_image)
 
   
-    return [edited_image,image_to_base64(edited_image)]
+    return image_to_base64(edited_image)
 
 def update_ui(option):
     if option == "Draw Mask":
@@ -358,71 +345,21 @@ def update_ui(option):
 
 
 with gr.Blocks() as demo:
+    gr.Markdown("## Upload 4 Images to Combine into a 2x2 Grid and Return Base64")
 
+    with gr.Row():
+        base_image = gr.Image(type="pil", label="base_image")
+        base_mask = gr.Image(type="pil", label="base_mask")
+        ref_image = gr.Image(type="pil", label="ref_image")
+        ref_mask = gr.Image(type="pil", label="ref_mask")
 
-    gr.Markdown("# Insert-Anything")
-    gr.Markdown("### Make sure to select the correct mask button!!")
-    gr.Markdown("### Click the output image to toggle between Diptych and final results!!")
+    output_base64 = gr.Textbox(label="Output Image (Base64 PNG)")
 
-    with gr.Row():
-        with gr.Column(scale=1):
-            with gr.Row():
-                base_image = gr.ImageEditor(label="Background Image", sources="upload", type="pil", brush=gr.Brush(colors=["#FFFFFF"],default_size = 30,color_mode = "fixed"),
-                                    layers = False,
-                                    interactive=True)
-
-                base_mask = gr.ImageEditor(label="Background Mask", sources="upload", type="pil", layers = False, brush=False, eraser=False)
-
-            with gr.Row():
-                base_mask_option = gr.Radio(["Draw Mask", "Upload with Mask"], label="Background Mask Input Option", value="Upload with Mask")
-
-            with gr.Row():
-                ref_image = gr.ImageEditor(label="Reference Image", sources="upload", type="pil", brush=gr.Brush(colors=["#FFFFFF"],default_size = 30,color_mode = "fixed"),
-                                    layers = False,
-                                    interactive=True)
-                                    
-                ref_mask = gr.ImageEditor(label="Reference Mask", sources="upload", type="pil", layers = False, brush=False, eraser=False)
-
-            with gr.Row():
-                ref_mask_option = gr.Radio(["Draw Mask", "Upload with Mask", "Label to Mask"], label="Reference Mask Input Option", value="Upload with Mask")
-
-            with gr.Row():
-                text_prompt = gr.Textbox(label="Label", placeholder="Enter the category of the reference object, e.g., car, dress, toy, etc.")
-
-        with gr.Column(scale=1):
-            baseline_gallery = gr.Gallery(label='Output', show_label=True, elem_id="gallery", height=695, columns=1)
-            base64_box = gr.Textbox(label="Base64 Output", visible=False)
-            with gr.Accordion("Advanced Option", open=True):
-                seed = gr.Slider(label="Seed", minimum=-1, maximum=999999999, step=1, value=666)
-                gr.Markdown("### Guidelines")
-                gr.Markdown(" Users can try using different seeds. For example, seeds like 42 and 123456 may produce different effects.")
-                gr.Markdown(" Draw Mask means manually drawing a mask on the original image.")
-                gr.Markdown(" Upload with Mask means uploading a mask file.")
-                gr.Markdown(" Label to Mask means simply inputting a label to automatically extract the mask and obtain the result.")
-                
-
-    run_local_button = gr.Button(value="Run")
-
-    # #### example #####
-    num_examples = len(image_list)
-    for i in range(num_examples):
-        with gr.Row():
-            if i == 0:
-                gr.Examples([image_list[i]], inputs=[base_image], label="Examples - Background Image", examples_per_page=1)
-                gr.Examples([image_mask_list[i]], inputs=[base_mask], label="Examples - Background Mask", examples_per_page=1)
-                gr.Examples([ref_list[i]], inputs=[ref_image], label="Examples - Reference Object", examples_per_page=1)
-                gr.Examples([ref_mask_list[i]], inputs=[ref_mask], label="Examples - Reference Mask", examples_per_page=1)
-            else:
-                gr.Examples([image_list[i]], inputs=[base_image], examples_per_page=1, label="")
-                gr.Examples([image_mask_list[i]], inputs=[base_mask], examples_per_page=1, label="")
-                gr.Examples([ref_list[i]], inputs=[ref_image], examples_per_page=1, label="")
-                gr.Examples([ref_mask_list[i]], inputs=[ref_mask], examples_per_page=1, label="")
-        if i < num_examples - 1:
-            gr.HTML("<hr>")
-    # #### example #####
-
-    run_local_button.click(fn=run_local,
-                            inputs=[base_image, base_mask, ref_image, ref_mask, seed, base_mask_option, ref_mask_option, text_prompt],
-                            outputs=[baseline_gallery, base64_box]
-                            )
+    submit_btn = gr.Button("Process Images")
+
+    submit_btn.click(
+        fn=run_local,
+        inputs=[base_image, base_mask, ref_image, ref_mask, ],
+        outputs=output_base64
+    )
 demo.launch()