Update app.py

app.py

@@ -112,10 +112,9 @@ def create_zip_file(tiles: List[np.ndarray], prefix: str = "tile") -> str:
     return zip_path
 
 
-def process_image(image: np.ndarray, tile_size: int, annotation_type: str,
-                 annotation_position: Tuple[int, int]) -> Tuple[List[np.ndarray], str]:
+def process_image(image: np.ndarray, tile_size: int) -> Tuple[List[np.ndarray], str]:
     """
-    Split the input image into tiles, annotate them at specified positions, and create a zip file of the tiles.
+    Split the input image into tiles and create a zip file of the tiles.
 
     Parameters
     ----------
@@ -123,33 +122,23 @@ def process_image(image: np.ndarray, tile_size: int, annotation_type: str,
         Input image as a NumPy array.
     tile_size : int
         Size of each square tile in pixels.
-    annotation_type : str
-        Type of annotation to add ("Cross" or "Rectangle").
-    annotation_position : Tuple[int, int]
-        (x, y) position for the annotation.
 
     Returns
     -------
     Tuple[List[np.ndarray], str]
-        A tuple containing the list of annotated image tiles and the path to the zip file.
+        A tuple containing the list of image tiles and the path to the zip file.
+
+    Examples
+    --------
+    >>> tiles, zip_path = process_image(image, tile_size=500)
+    >>> len(tiles)
+    16
+    >>> os.path.exists(zip_path)
+    True
     """
     tiles = split_image(image, tile_size)
-    annotated_tiles = []
-
-    for tile in tiles:
-        x, y = annotation_position
-        if annotation_type == "Cross":
-            center = (x, y)
-            cv2.drawMarker(tile, center, color=(255, 0, 0), markerType=cv2.MARKER_CROSS,
-                           markerSize=20, thickness=2)
-        elif annotation_type == "Rectangle":
-            top_left = (x, y)
-            bottom_right = (tile.shape[1]-x, tile.shape[0]-y)
-            cv2.rectangle(tile, top_left, bottom_right, color=(0, 255, 0), thickness=2)
-        annotated_tiles.append(tile)
-
-    zip_path = create_zip_file(annotated_tiles) if annotated_tiles else ""
-    return annotated_tiles, zip_path
+    zip_path = create_zip_file(tiles) if tiles else ""
+    return tiles, zip_path
 
 
 def main():
@@ -170,17 +159,6 @@ def main():
                 minimum=100, maximum=1000, step=100, value=500, label="Tile Size"
             )
 
-        with gr.Row():
-            annotation_type = gr.Dropdown(
-                choices=["Cross", "Rectangle"],
-                value="Rectangle",
-                label="Annotation Type"
-            )
-            annotation_position = gr.Textbox(
-                label="Annotation Position (x,y)",
-                placeholder="e.g., 250,250"
-            )
-
         with gr.Row():
             submit_btn = gr.Button("Process Image")
 
@@ -190,7 +168,7 @@ def main():
 
         submit_btn.click(
             fn=process_image,
-            inputs=[input_image, tile_size, annotation_type, annotation_position],
+            inputs=[input_image, tile_size],
             outputs=[gallery, download_btn],
         )
 
@@ -198,4 +176,4 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    main()