Update app.py

app.py

@@ -2,13 +2,21 @@ import gradio as gr
 from PIL import Image
 
 apol=[]
+pixels=[]
 
 with gr.Blocks() as iface:
     sketchpad=gr.ImageEditor(type="pil",height=512,width=512,container=False,show_share_button=False,show_download_button=False)
     gal=gr.Gallery(columns=1)
-    @gr.on(triggers=[sketchpad.change],inputs=[sketchpad],outputs=[gal])
-    def plex(sketchpad):
+    @gr.on(triggers=[sketchpad.select, sketchpad.change],inputs=[sketchpad],outputs=[gal])
+    def plex(sketchpad, evt: gr.SelectData):
+        pixels.append(evt.index)
+        inpts = np.array(pixels)
+        intwos = Image.fromarray(inpts)
+        inpts_inv = np.ones(shape=(inpts.shape[0]))
+        inv_twos = Image.fromarray(inpts_inv)
         sketchpad=sketchpad["composite"].convert("RGB")
+        apol.append(intwos)
+        apol.append(inv_twos)
         apol.append(sketchpad)
         return apol