chore: updates to app.py

app.py

@@ -1,23 +1,28 @@
 import gradio as gr
-import imageio
+import cv2
 
-def dummy(img):
-    imageio.imwrite("output_image.png", img["mask"])
-    return img["image"], img["mask"]
+def create_mask(image):
+   # Convert image to grayscale for easier segmentation
+   gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+
+   # Apply thresholding to create a binary mask (adjust threshold value as needed)
+   _, mask = cv2.threshold(gray, 127, 255, cv2.THRESH_BINARY)
+
+   return mask
 
 with gr.Blocks() as demo:
-    gr.Markdown(
-        """
-        # Image mask creator
-        This is a demo of a simple image mask creator. You can draw on the image and the mask will be created.
-        """
-    )
+   gr.Markdown(
+       """
+       # Image Mask Creator
+       Draw on the image to create a mask.
+       """
+   )
+
+   with gr.Row():
+       base_image = gr.Image(tool="sketch", label="Base Image", width=512, height=512, show_label=True)
+       mask_image = gr.Image(label="Mask Image", show_label=True)
 
-    with gr.Row():
-       img = gr.Image(tool="sketch", label="base image", width=512, height=512, show_label=True)
-       img1 = gr.Image()
-       img2 = gr.Image(label="mask image", show_label=True)
-    btn = gr.Button()
-    btn.click(dummy, img, [img1, img2])
+   btn = gr.Button("Create Mask")
+   btn.click(create_mask, inputs=base_image, outputs=mask_image)
 
 demo.launch(debug=True)