Update app.py

app.py

@@ -3,35 +3,39 @@ from PIL import Image, ImageOps
 import gradio as gr
 
 
-def image_mod(image):
-
-    image = remove(image)
-    new_image = Image.new("RGBA", image.size, "MAGENTA")
-    new_image.paste(image, mask=image)
-    # Add a black border of 10 pixels on all sides
-    new_image = ImageOps.expand(new_image, border=5, fill='black')
-
-    # Get the image size
-    width, height = new_image.size
-
-    # Create a new image to hold the photos
-    output_image = Image.new("RGB", (width * 4, height * 2))
-
-    # Paste the source image into the new image
-    for row in range(2):
-        for col in range(4):
-            output_image.paste(new_image, (col * width, row * height))
-    # Save the output image
-    return output_image
-
+def change_color(passport='Passport Photo', color = 'Background Color'):
+  image = remove(passport)
+  new_image = Image.new("RGB", image.size, color)
+  new_image.paste(image, mask=image)
+  # Add a black border of 10 pixels on all sides
+  new_image = ImageOps.expand(new_image, border=5, fill='black')
+  new_image = ImageOps.expand(new_image, border=1, fill='white')
+
+  # Get the image size
+  width, height = new_image.size
+
+  # Create a new image to hold the photos
+  output_image = Image.new("RGB", (width * 4, height * 2))
+
+  # Paste the source image into the new image
+  for row in range(2):
+    for col in range(4):
+      output_image.paste(new_image, (col * width, row * height))
+  # Save the output image
+  return output_image
+
+
+inputs = [
+  gr.Image(label="Passport Photo", type="pil", image_mode="RGBA"),
+  gr.ColorPicker(label="Background Color",),
+]
+outputs = gr.Image(label="Download Image")
 
 demo = gr.Interface(
-    image_mod,
-    gr.Image(type='pil'),
-    "image",
+  fn=change_color,
+  inputs=inputs,
+  outputs=outputs,
 )
 
 if __name__ == "__main__":
-    demo.launch()
-
-
+  demo.launch()