Update app.py

app.py

@@ -34,11 +34,17 @@ def imread(filepath):
     return img
 
 def check_dims(filepath):
-    tif = tiff.TiffFile(filepath)
 
-    store = tif.aszarr()
-    img = zarr.open(store, mode='r', chunks=None) 
-    store.close()
+    fpath, fext = os.path.splitext(filepath)
+    
+    if fext in ['.tiff', '.tif']:
+        tif = tiff.TiffFile(filepath)
+
+        store = tif.aszarr()
+        img = zarr.open(store, mode='r', chunks=None) 
+        store.close()
+    else:
+        img = cv2.imread(filepath)
 
     if img.ndim==3:
         return img.shape[0], None
@@ -258,7 +264,7 @@ def update(filepath, filepath_result, filepath_coordinates, z, c):
     if filepath_coordinates is None:
         display_boxes = []
     else:        
-        display_boxes = filter_coordinates_alt(filepath_coordinates, z, check_dims(filepath_show))
+        display_boxes = filter_coordinates_alt(filepath_coordinates, z, check_dims(filepath_re))
     
     return (filepath_show, display_boxes), (filepath_result_show, display_boxes)