Update app.py

app.py

@@ -11,6 +11,10 @@ from cellpose.io import imread, imsave
 
 from huggingface_hub import hf_hub_download
 
+img = np.zeros((384, 512))
+fp0 = "0.png"
+imsave(fp0, img)
+
 # @title Data retrieval
 def download_weights():    
     return hf_hub_download(repo_id="mouseland/cellpose-sam", filename="cpsam")
@@ -258,7 +262,7 @@ def update_image(filepath):
     for f in filepath: 
         print(f)
 
-    return filepath[-1] 
+    return filepath[-1], fp0, fp0
 
  
 def update_button(filepath): 
@@ -269,7 +273,7 @@ def update_button(filepath):
     #b1 = gr.DownloadButton("Download masks as TIFF", visible=False)
     #b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
 
-    return [filepath]
+    return [filepath], fp0, fp0
     
 with gr.Blocks(title = "Hello", 
                css=".gradio-container {background:purple;}") as demo:
@@ -298,19 +302,19 @@ with gr.Blocks(title = "Hello",
                     down_btn2 = gr.DownloadButton("Download outlines (PNG)", visible=False)  
                     
         with gr.Column(scale=2):     
-            img_outlines = gr.Image(label = "Outlines", type = "filepath", format = 'png') #, width = "50vw", height = "20vw")
+            outlines = gr.Image(label = "Outlines", type = "filepath", format = 'png', value = fp0) #, width = "50vw", height = "20vw")
             #img_overlay = gr.Image(label = "Overlay", type = "pil", format = 'png') #, width = "50vw", height = "20vw")
-            flows = gr.Image(label = "Cellpose flows", type = "filepath", format = 'png') #, width = "50vw", height = "20vw")
+            flows = gr.Image(label = "Cellpose flows", type = "filepath", format = 'png', value = fp0) #, width = "50vw", height = "20vw")
 
             
     sample_list = []
     for j in range(23):
         sample_list.append("samples/img%0.2d.png"%j)
-    gr.Examples(sample_list, fn = update_button, inputs=input_image, outputs = [up_btn], examples_per_page=25, label = "Click on an example to try it")
+    gr.Examples(sample_list, fn = update_button, inputs=input_image, outputs = [up_btn, outlines, flows], examples_per_page=25, label = "Click on an example to try it")
     
-    input_image.upload(update_button, input_image, [up_btn])
-    up_btn.upload(update_image, up_btn, [input_image])
-    send_btn.click(cellpose_segment, [up_btn, resize], [img_outlines, flows, down_btn, down_btn2])
+    input_image.upload(update_button, input_image, [up_btn, outlines, flows])
+    up_btn.upload(update_image, up_btn, [input_image, outlines, flows])
+    send_btn.click(cellpose_segment, [up_btn, resize], [outlines, flows, down_btn, down_btn2])
 
     #down_btn.click(download_function, None, [down_btn, down_btn2])