Update app.py

app.py

@@ -5,7 +5,6 @@ import random
 import pandas as pd
 import os
 from itertools import islice
-import json
 import pyarrow as pa
 import pyarrow.parquet as pq
 
@@ -57,7 +56,7 @@ def get_next_image():
             shuffled_chunk = []
             chunk_iter = None
 
-def rate_image(image_id, bounds, rating):
+def rate_image(image_id, bounds, rating, google_maps_link):
     global labels_list
 
     labels_list.append(
@@ -65,7 +64,7 @@ def rate_image(image_id, bounds, rating):
             "image_id": image_id,
             "bounds": bounds,
             "rating": rating,
-            "google_maps_link": "", # Adding google maps link to the data to be downloaded
+            "google_maps_link": google_maps_link,
         }
     )
 
@@ -74,11 +73,14 @@ def rate_image(image_id, bounds, rating):
 
 def save_labels():
     global labels_list
-    table = pa.Table.from_pylist(labels_list)
-    pq.write_table(table, "labeled_data.parquet")
-    return "labeled_data.parquet"
-
-# Gradio interface 
+    if labels_list:
+       table = pa.Table.from_pylist(labels_list)
+       pq.write_table(table, "labeled_data.parquet")
+       return "labeled_data.parquet"
+    else:
+        return None
+
+# Gradio interface
 with gr.Blocks() as iface:
     with gr.Row():
         with gr.Column():
@@ -88,13 +90,14 @@ with gr.Blocks() as iface:
             google_maps_link_out = gr.Textbox(label="Google Maps Link", visible=True)
         with gr.Column():
             rating_radio = gr.Radio(["Cool", "Not Cool"], label="Rating")
-            submit_button = gr.Button("Submit Rating")
-            download_button = gr.Button("Download Labels")
+            with gr.Row():
+                submit_button = gr.Button("Submit Rating")
+                download_button = gr.Button("Download Labels")
             download_file = gr.File(label="Download")
 
     submit_button.click(
         rate_image,
-        inputs=[image_id_out, bounds_out, rating_radio],
+        inputs=[image_id_out, bounds_out, rating_radio, google_maps_link_out],
         outputs=[image_out, image_id_out, bounds_out, google_maps_link_out]
     )