Update app.py

app.py

@@ -7,7 +7,7 @@ from matplotlib.colors import hsv_to_rgb
 import matplotlib.pyplot as plt
 import os, io, base64
 from PIL import Image 
-from cellpose.io import imread
+from cellpose.io import imread, imsave
 
 # @title Data retrieval
 def download_weights():
@@ -165,15 +165,19 @@ def cellpose_segment(img_pil):
     flows = flows.resize((Lx, Ly), resample  = Image.BICUBIC)
 
     #masks = Image.fromarray(255. * crand[masks])
+    #pil_masks = Image.fromarray(masks.astype('int32'))
+    #pil_masks.save(fname_mask)
 
-    pil_masks = Image.fromarray(masks.astype('int32'))
-    pil_masks.save("masks.tiff")
-    
+    fname_out  = os.path.splitext(img_pil)[0]+"_outlines.png"
+    fname_masks = os.path.splitext(img_pil)[0]+"_masks.tif"
 
-    outpix.save(img_pil) #"outlines.png")
+    imsave(fname_masks, masks)
+    
+    
+    outpix.save(fname_out) #"outlines.png")
 
-    b1 = gr.DownloadButton(visible=True, value = "masks.tiff")
-    b2 = gr.DownloadButton(visible=True, value = img_pil) #"outlines.png")
+    b1 = gr.DownloadButton(visible=True, value = fname_masks)
+    b2 = gr.DownloadButton(visible=True, value = fname_out) #"outlines.png")
     
     return outpix, overlay, flows, b1, b2
 
@@ -203,7 +207,7 @@ with gr.Blocks(title = "Hello",
             input_image = gr.Image(label = "Input image", type = "filepath")
             send_btn = gr.Button("Run Cellpose-SAM")
             with gr.Row():
-                down_btn = gr.DownloadButton("Download masks (TIFF)", visible=False)            
+                down_btn = gr.DownloadButton("Download masks (TIF)", visible=False)            
                 down_btn2 = gr.DownloadButton("Download outlines (PNG)", visible=False)
 
             gr.HTML("""<a style="color:white;" href="https://github.com/MouseLand/cellpose" target="_blank">github page for cellpose</a>""")