Add decode_blob_data function to handle Gradio image blob conversion in app.py

app.py

@@ -7,6 +7,8 @@ Any new model should implement the following functions:
 
 import os
 import glob
+import hashlib
+import struct
 
 # import spaces
 import gradio as gr
@@ -20,7 +22,7 @@ from utils import (
 from flagging import HuggingFaceDatasetSaver
 
 import install_private_repos  # noqa: F401
-from seavision import load_model, imread
+from seavision import load_model
 
 
 TITLE = """
@@ -59,12 +61,113 @@ def inference(image):
     return results.draw(image)
 
 
+def decode_blob_data(image_data):
+    """
+    Decode blob data from Gradio image component.
+    Handles blob format and converts to proper image file format.
+    """
+    if not isinstance(image_data, dict):
+        return image_data
+    
+    print(f"DEBUG: Original input - image: {image_data}")
+    
+    # Check if this is blob data - more comprehensive check
+    is_blob = (
+        'path' in image_data and 
+        'blob' in image_data['path'] and
+        image_data.get('size') is None and
+        image_data.get('orig_name') is None and
+        image_data.get('mime_type') is None
+    )
+    
+    if is_blob:
+        print(f"DEBUG: Converting blob data: {image_data}")
+        print("DEBUG: Detected blob format, converting...")
+        
+        blob_path = image_data['path']
+        print(f"DEBUG: Blob path: {blob_path}")
+        
+        # Read the blob file
+        with open(blob_path, 'rb') as f:
+            blob_content = f.read()
+        
+        file_size = len(blob_content)
+        print(f"DEBUG: File size: {file_size}")
+        
+        # Check file header to determine format
+        if len(blob_content) >= 8:
+            header = blob_content[:8].hex()
+            print(f"DEBUG: File header: {header}")
+            
+            # PNG header: 89 50 4E 47 0D 0A 1A 0A
+            if header.startswith('89504e470d0a1a0a'):
+                extension = '.png'
+                mime_type = 'image/png'
+            # JPEG header: FF D8 FF
+            elif header.startswith('ffd8ff'):
+                extension = '.jpg'
+                mime_type = 'image/jpeg'
+            # GIF header: 47 49 46 38
+            elif header.startswith('47494638'):
+                extension = '.gif'
+                mime_type = 'image/gif'
+            else:
+                # Default to PNG if we can't determine
+                extension = '.png'
+                mime_type = 'image/png'
+        else:
+            extension = '.png'
+            mime_type = 'image/png'
+        
+        print(f"DEBUG: Detected extension: {extension}, MIME type: {mime_type}")
+        
+        # Generate a unique filename
+        content_hash = hashlib.md5(blob_content).hexdigest()[:8]
+        new_filename = f"flagged_image_{content_hash}{extension}"
+        print(f"DEBUG: Generated filename: {new_filename}")
+        
+        # Create new path in the same directory
+        import tempfile
+        temp_dir = os.path.dirname(blob_path)
+        new_path = os.path.join(temp_dir, new_filename)
+        print(f"DEBUG: New path: {new_path}")
+        
+        # Write the content to the new file
+        with open(new_path, 'wb') as f:
+            f.write(blob_content)
+        
+        print(f"DEBUG: Successfully renamed blob to: {new_path}")
+        
+        # Update the image data
+        converted_data = {
+            'path': new_path,
+            'url': image_data['url'].replace('blob', new_filename),
+            'size': file_size,
+            'orig_name': new_filename,
+            'mime_type': mime_type,
+            'is_stream': False,
+            'meta': image_data.get('meta', {})
+        }
+        
+        print(f"DEBUG: Converted data: {converted_data}")
+        return converted_data
+    else:
+        print("DEBUG: Not a blob, skipping conversion")
+    
+    print(f"DEBUG: Converted image: {image_data}")
+    return image_data
+
+
 def flag_img_input(
     image: gr.Image, flag_option: str = "misdetection", username: str = "anonymous"
 ):
     """Wrapper for flagging"""
     print(f"{image=}, {flag_option=}, {username=}")
-    hf_writer.flag([image], flag_option=flag_option, username=username)
+    
+    # Decode blob data if necessary
+    decoded_image = decode_blob_data(image)
+    
+    hf_writer.flag([decoded_image], flag_option=flag_option, username=username)
 
 
 # Flagging