Update app.py

app.py

@@ -254,7 +254,11 @@ def update_image(filepath):
     #b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
     for f in filepath: 
         print(f)
-    return filepath[-1] #, b1, b2
+
+    im1 = gr.Image(label = "Outlines", type = "pil", format = 'png') #, width = "50vw", height = "20vw")
+    im2 = gr.Image(label = "Cellpose flows", type = "pil", format = 'png') #, width = "50vw", height = "20vw")
+
+    return filepath[-1] , im1, im2
 
  
 def update_button(filepath): 
@@ -264,7 +268,10 @@ 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]#, b1, b2  
+    im1 = gr.Image(label = "Outlines", type = "pil", format = 'png') #, width = "50vw", height = "20vw")
+    im2 = gr.Image(label = "Cellpose flows", type = "pil", format = 'png') #, width = "50vw", height = "20vw")
+
+    return [filepath] , im1, im2
     
 with gr.Blocks(title = "Hello", 
                css=".gradio-container {background:purple;}") as demo:
@@ -287,15 +294,10 @@ with gr.Blocks(title = "Hello",
                     
                     #gr.HTML("""<h4 style="color:white;"> Note2: Only the first image of a tif will display the segmentations, but you can download segmentations for all planes. </h4>""")
                     
-                    
-                
                 with gr.Column(scale=1):
                     send_btn = gr.Button("Run Cellpose-SAM")
                     down_btn = gr.DownloadButton("Download masks (TIF)", visible=False)            
                     down_btn2 = gr.DownloadButton("Download outlines (PNG)", visible=False)  
-            
-            
-
                     
         with gr.Column(scale=2):     
             img_outlines = gr.Image(label = "Outlines", type = "pil", format = 'png') #, width = "50vw", height = "20vw")
@@ -308,8 +310,8 @@ with gr.Blocks(title = "Hello",
         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")
     
-    input_image.upload(update_button, input_image, up_btn)
-    up_btn.upload(update_image, up_btn, input_image)
+    input_image.upload(update_button, input_image, [up_btn, img_outlines, flows])
+    up_btn.upload(update_image, up_btn, [input_image, img_outlines, flows])
     send_btn.click(cellpose_segment, [up_btn, resize], [img_outlines, flows, down_btn, down_btn2])
 
     #down_btn.click(download_function, None, [down_btn, down_btn2])