Update app.py

app.py

@@ -11,7 +11,7 @@ from cellpose.io import imread, imsave
 
 from huggingface_hub import hf_hub_download
 
-img = np.zeros((384, 512))
+img = np.zeros((96, 128))
 fp0 = "0.png"
 imsave(fp0, img)
 
@@ -167,6 +167,7 @@ def run_model_gpu1000(img):
     masks, flows, _ = model.eval(img)#, channels = [0,0])
     return masks, flows
 
+
 from zipfile import ZipFile
 def cellpose_segment(filepath, resize = 1000):
 
@@ -249,31 +250,22 @@ def download_function():
     b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
     return b1, b2
 
+def tif_view(filepath):
+    fpath, fext = os.path.splitext(filepath)
+    if fext in ['tiff', 'tif']:
+        img = imread(filepath[-1])
+        filepath = fpath+'.png'
+        imsave(filepath, img)
+    return filepath
+    
 def update_image(filepath): 
-    #img = imread(filepath)    
-    #img = normalize99(img)
-    #img = np.clip(img, 0, 1)
-
-    #filegui  = os.path.splitext(filepath)[0]+"_gui.png"
-    #imsave(filegui, img)
-
-    #b1 = gr.DownloadButton("Download masks as TIFF", visible=False)
-    #b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
-    for f in filepath: 
-        print(f)
-
-    return filepath[-1], fp0, fp0
-
- 
-def update_button(filepath): 
-    #img = imread(filepath)    
-    #img = normalize99(img)
-    #img = np.clip(img, 0, 1)
-
-    #b1 = gr.DownloadButton("Download masks as TIFF", visible=False)
-    #b2 = gr.DownloadButton("Download outline image as PNG", visible=False)
+    for f in filepath:
+        f = tif_view(f)
+    return filepath, filepath[-1], fp0, fp0
 
-    return [filepath], fp0, fp0
+def update_button(filepath):
+    filepath = tif_view(filepath)
+    return filepath, [filepath], fp0, fp0
     
 with gr.Blocks(title = "Hello", 
                css=".gradio-container {background:purple;}") as demo:
@@ -312,8 +304,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, outlines, flows], examples_per_page=25, label = "Click on an example to try it")
     
-    input_image.upload(update_button, input_image, [up_btn, outlines, flows])
-    up_btn.upload(update_image, up_btn, [input_image, outlines, flows])
+    input_image.upload(update_button, input_image, [up_btn, input_image, outlines, flows])
+    up_btn.upload(update_image, up_btn, [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])