Update app.py

app.py

@@ -130,7 +130,7 @@ def update_image(filepath, z):
     filepath_show = norm_path(filepath_show)
     print(filepath_show)
     print(filepath)
-    return filepath_show, filepath, fp0
+    return filepath_show, filepath, (fp0, [((5, 5, 10, 10), 'nothing')])
 
 def update_button(filepath, z):
     print('update_btn')
@@ -138,7 +138,7 @@ def update_button(filepath, z):
     filepath_show = tif_view(filepath, z)
     filepath_show = norm_path(filepath_show)
     print(filepath_show)
-    return filepath_show, [filepath], fp0
+    return filepath_show, [filepath], (fp0, [((5, 5, 10, 10), 'nothing')])
 
 def update_z(filepath, filepath_result, z): 
     print('update_img')   
@@ -155,7 +155,7 @@ def update_z(filepath, filepath_result, z):
     print(filepath_show)
     print(filepath)
     
-    return filepath_show, filepath_result_show
+    return filepath_show, (filepath_result_show, [((5, 5, 10, 10), 'nothing')])
 
 def detect_cells(filepath, z):
     model = tf.keras.models.load_model('./model_positions', compile=False)
@@ -180,7 +180,7 @@ def detect_cells(filepath, z):
 
     coordinates = peak_local_max(result, min_distance=2, threshold_abs=0.2,  exclude_border=False)
 
-    return filepath_result, filepath_result_show
+    return filepath_result, (filepath_result_show, [((5, 5, 10, 10), 'nothing')])
 
     
 with gr.Blocks(title = "Hello", 
@@ -217,7 +217,8 @@ with gr.Blocks(title = "Hello",
                     
         with gr.Column(scale=2):     
             #
-            output_image = gr.Image(label = "Output", type = "filepath")
+            #output_image = gr.Image(label = "Output", type = "filepath")
+            output_image = gr.AnnotatedImage(label = "Output")
 
     #sample_list = glob.glob("samples/*.png")
     #sample_list = []