Update app.py

app.py

@@ -104,8 +104,8 @@ class AnnotationManager:
     
     def process_upload(self, image_path):
         """Process uploaded image"""
-        if not image_path:
-            logger.warning("No image path provided")
+        if not isinstance(image_path, (str, bytes, os.PathLike)):
+            logger.warning(f"Invalid image path type: {type(image_path)}")
             return None
             
         try:
@@ -200,10 +200,14 @@ def create_interface():
         debug_output = gr.Textbox(label="Debug Info", interactive=False)
         
         # Event handlers
-        def handle_image_upload(image_path):
+        def handle_image_upload(bbox_data):
             try:
-                if not image_path:
+                if not bbox_data or not isinstance(bbox_data, tuple):
                     return None, "No image uploaded"
+                
+                image_path, annotations = bbox_data
+                if not image_path:
+                    return None, "No image path provided"
                     
                 logger.info(f"Handling upload for: {image_path}")
                 resized_path = annotation_mgr.process_upload(image_path)
@@ -211,7 +215,8 @@ def create_interface():
                 if resized_path and os.path.exists(resized_path):
                     debug_msg = f"Processed image path: {resized_path}"
                     logger.info(debug_msg)
-                    return resized_path, debug_msg
+                    # Return the resized path and keep any existing annotations
+                    return (resized_path, annotations), debug_msg
                 else:
                     error_msg = "Failed to process image"
                     logger.error(error_msg)