Update app.py

app.py

@@ -80,20 +80,23 @@ def get_s2_cell_polygon(cell_id):
     vertices.append(vertices[0])  # Close the polygon
     return vertices
 
-# Function to generate Plotly map figure
 def create_map_figure(predictions, cell_ids):
     fig = go.Figure()
 
-    for cell_id in cell_ids:
+    # Assign colors based on rank
+    colors = ['rgba(0, 255, 0, 0.2)'] * 3 + ['rgba(255, 255, 0, 0.2)'] * 7
+
+    for rank, cell_id in enumerate(cell_ids):
         cell_id = int(cell_id)
         polygon = get_s2_cell_polygon(cell_id)
         lats, lons = zip(*polygon)
+        color = colors[rank]
         fig.add_trace(go.Scattermapbox(
             lat=lats,
             lon=lons,
             mode='lines',
             fill='toself',
-            fillcolor='rgba(0, 0, 255, 0.2)',
+            fillcolor=color,
             line=dict(color='blue'),
             name=f'Cell ID: {cell_id}'
         ))
@@ -114,6 +117,7 @@ def create_map_figure(predictions, cell_ids):
 
     return fig
 
+
 # Create label output function
 def create_label_output(predictions):
     results, cell_ids = predictions
@@ -125,6 +129,7 @@ def predict_and_plot(input_img):
     predictions = predict(input_img)
     return create_label_output(predictions)
 
+
 # Gradio app definition
 with gr.Blocks() as gradio_app:
     with gr.Column():