Update app.py

app.py

@@ -220,13 +220,16 @@ def cellpose_segment(filepath, resize = 1000):
     fname_out  = os.path.splitext(filepath[-1])[0]+"_outlines.png"
     outpix.save(fname_out) #"outlines.png")
 
+    fname_flows  = os.path.splitext(filepath[-1])[0]+"_flows.png"
+    flows.save(fname_flows) #"outlines.png")
+
     if len(filepath)>1:
         b1 = gr.DownloadButton(visible=True, value = zip_path)
     else:
         b1 = gr.DownloadButton(visible=True, value = fname_masks)
     b2 = gr.DownloadButton(visible=True, value = fname_out) #"outlines.png")
     
-    return outpix, flows, b1, b2
+    return fname_out, fname_flows, b1, b2
 
 # Gradio Interface
 #iface = gr.Interface(
@@ -255,10 +258,7 @@ def update_image(filepath):
     for f in filepath: 
         print(f)
 
-    im1 = gr.Image(label = "Outlines", type = "pil", format = 'png', visible=False) #, width = "50vw", height = "20vw")
-    im2 = gr.Image(label = "Cellpose flows", type = "pil", format = 'png', visible=False) #, width = "50vw", height = "20vw")
-
-    return filepath[-1] , im1, im2
+    return filepath[-1] , None, None
 
  
 def update_button(filepath): 
@@ -268,10 +268,8 @@ def update_button(filepath):
 
     #b1 = gr.DownloadButton("Download masks as TIFF", visible=False)
     #b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
-    im1 = gr.Image(label = "Outlines", type = "pil", format = 'png', visible=False) #, width = "50vw", height = "20vw")
-    im2 = gr.Image(label = "Cellpose flows", type = "pil", format = 'png', visible=False) #, width = "50vw", height = "20vw")
 
-    return [filepath] , im1, im2
+    return [filepath] , None, None
     
 with gr.Blocks(title = "Hello", 
                css=".gradio-container {background:purple;}") as demo:
@@ -300,9 +298,9 @@ 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 = "pil", format = 'png', visible=False) #, width = "50vw", height = "20vw")
+            img_outlines = gr.Image(label = "Outlines", type = "filepath", format = 'png', visible=False) #, 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 = "pil", format = 'png', visible=False) #, width = "50vw", height = "20vw")
+            flows = gr.Image(label = "Cellpose flows", type = "filepath", format = 'png', visible=False) #, width = "50vw", height = "20vw")
 
             
     sample_list = []