No more update descriptions until I have a working version

app.py

@@ -46,17 +46,14 @@ def get_images(batch_size, state):
     images = []
     metadatas = []
     for _ in range(batch_size):
-      sample = get_next_sample()
-      if sample is None:
-        break
-
-      image = sample["rgb"][0]
-      metadata = sample["metadata"]
-      metadata["map"] = f'<a href="{utils.get_google_map_link(sample, subset)}" target="_blank">🧭</a>'
-
-      images.append(image)
-      metadatas.append(metadata)
-
+        sample = get_next_sample()
+        if sample is None:
+            break
+        image = sample["rgb"][0]
+        metadata = sample["metadata"]
+        metadata["map"] = f'<a href="{utils.get_google_map_link(sample, subset)}" target="_blank">🧭</a>'
+        images.append(image)
+        metadatas.append(metadata)
     return images, DataFrame(metadatas)
 
 # --- Save Labeled Data ---
@@ -66,8 +63,7 @@ def save_labeled_data(label, state):
     if sample is None:
         return "No image to label", None, DataFrame()
 
-    image = sample["rgb"][0]  # Get the PIL Image object
-
+    image = sample["rgb"][0]
     image_bytes = image.convert("RGB").tobytes()
     labeled_data.append({
         "image": image_bytes,
@@ -80,46 +76,38 @@ def save_labeled_data(label, state):
 
     new_sample = get_next_sample()
     if new_sample is None:
-      state["sample"] = None
-      return "Dataset exhausted.", None, DataFrame()
+        state["sample"] = None
+        return "Dataset exhausted.", None, DataFrame()
 
     state["sample"] = new_sample
-
     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>'
-    
     return "", [new_image], DataFrame([new_metadata])
 
 # --- Gradio Interface ---
 # --- Labeling UI ---
 def labeling_ui():
-    
     state = gr.State({"sample": None, "subset": DATASET_SUBSET})
 
     with gr.Row():
         with gr.Column():
             gallery = gr.Gallery(label="Satellite Image", interactive=False, columns=1, object_fit="scale-down")
-
             with gr.Row():
                 cool_button = gr.Button("Cool")
                 not_cool_button = gr.Button("Not Cool")
-
             table = gr.DataFrame(datatype="html")
 
             def initialize_labeling_ui():
-              sample = get_next_sample()
-              
-              image, metadata = get_images(1, {"sample": None, "subset": DATASET_SUBSET})
-
-              return sample, image, metadata
+                sample = get_next_sample()
+                images, metadata = get_images(1, {"sample": None, "subset": DATASET_SUBSET})
+                return sample, images, metadata
 
-            initial_sample, initial_image, initial_metadata = initialize_labeling_ui()
-            gallery.value = initial_image
+            initial_sample, initial_images, initial_metadata = initialize_labeling_ui()
+            gallery.value = initial_images
             table.value = initial_metadata
-            state.value["sample"] = initial_sample
+            state.value = {"sample": initial_sample, "subset": DATASET_SUBSET}
 
-            # Handle button clicks
             cool_button.click(
                 fn=lambda label, state: save_labeled_data(label, state),
                 inputs=[gr.Textbox(visible=False, value="cool"), state],
@@ -133,7 +121,6 @@ def labeling_ui():
 
 # --- Display UI ---
 def display_ui():
-
     def get_random_cool_images(n):
         cool_samples = [d for d in labeled_data if d["label"] == "cool"]
         return [Image.frombytes("RGB", (384, 384), s["image"]) for s in cool_samples] if len(cool_samples) >= n else []
@@ -158,7 +145,6 @@ def display_ui():
     with gr.Row():
         new_image_component = gr.Image(label="New Image", type="pil")
         metadata_display = gr.Textbox(label="Metadata (Bounds)")
-
     with gr.Row():
         cool_images_gallery = gr.Gallery(label="Cool Examples", value=[], columns=DISPLAY_N_COOL)
 
@@ -186,4 +172,4 @@ with gr.Blocks() as demo:
         with gr.TabItem("Display"):
             display_ui()
 
-demo.launch(debug=True, share=True)
+demo.launch(debug=True)