Update app.py

app.py

@@ -268,18 +268,28 @@ def tif_view(filepath):
         nn = np.minimum(3, img.shape[-1])
         imgi[:,:,:nn] = img[:,:,:nn]
         
-        filepath = fpath+'.png'
+        #filepath = fpath+'.png'
         imsave(filepath, imgi)
     return filepath
+
+def norm_path(filepath):
+    img = imread(filepath)
+    img = normalize99(img)
+    fpath, fext = os.path.splitext(filepath)
+    filepath = fpath+'.png'
+    imsave(filepath, img)
+    return filepath 
     
 def update_image(filepath): 
     for f in filepath:
         f = tif_view(f)
-    return filepath[-1], filepath, fp0, fp0
+    filepath_show = norm_path(filepath[-1])
+    return filepath_show, filepath, fp0, fp0
 
 def update_button(filepath):
     filepath = tif_view(filepath)
-    return filepath, [filepath], fp0, fp0
+    filepath_show = norm_path(filepath[-1])
+    return filepath_show, [filepath], fp0, fp0
     
 with gr.Blocks(title = "Hello", 
                css=".gradio-container {background:purple;}") as demo:
@@ -316,10 +326,11 @@ with gr.Blocks(title = "Hello",
     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 = [input_image, up_btn, outlines, flows], examples_per_page=25, label = "Click on an example to try it")
-    
     input_image.upload(update_button, input_image, [input_image, up_btn, outlines, flows])
     up_btn.upload(update_image, up_btn, [input_image, up_btn, 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])