Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import earthview as ev
 from PIL import Image
+import numpy as np
+import random
 import os
 import json
 import utils
@@ -39,7 +41,12 @@ def get_next_sample():
 # --- Save Labeled Data ---
 def save_labeled_data(image, label, state):
     global labeled_data
-    image_bytes = image.convert("RGB").tobytes() if image else None
+
+    # Convert PIL Image to bytes before saving
+    if image is not None:
+        image_bytes = image.convert("RGB").tobytes()
+    else:
+        image_bytes = None
 
     labeled_data.append({
         "image": image_bytes,
@@ -52,15 +59,21 @@ def save_labeled_data(image, label, state):
 
     new_sample = get_next_sample()
     if new_sample is None:
+        print("Dataset exhausted.")
         return "Dataset exhausted.", None, DataFrame()
 
     new_image = new_sample["rgb"][0]
     new_metadata = new_sample["metadata"]
     new_metadata["map"] = f'<a href="{utils.get_google_map_link(new_sample, DATASET_SUBSET)}" target="_blank">🧭</a>'
-
     state["metadata"] = new_metadata  # Update metadata in state
 
-    return "", new_image, DataFrame([new_metadata])
+    # Convert new PIL Image to bytes for sending to client-side
+    if new_image is not None:
+        new_image_bytes = new_image.convert("RGB").tobytes()
+    else:
+        new_image_bytes = None
+
+    return "", new_image_bytes, DataFrame([new_metadata])
 
 # --- Gradio Interface ---
 # --- Labeling UI ---