shaded tiles

app.py

@@ -21,7 +21,7 @@ with gr.Blocks() as demo:
 
 
 # Autofocus JavaScript: put cursor into Transcription box
-    demo.load(
+''' demo.load(
         None,
         None,
         _js="""
@@ -35,7 +35,7 @@ with gr.Blocks() as demo:
                 }, 100);
             }
         """
-    )
+    )'''
 
 
 # [DEBUG]

inference_tab/inference_setup.py

@@ -7,7 +7,7 @@ import os
 TILE_SIZE = 1024  
 TILE_FOLDER = "tiles"
 os.makedirs(TILE_FOLDER, exist_ok=True)
-tiles_cache = {"tiles": [], "selected_tile": None}
+tiles_cache = {"tiles": [], "selected_tile": None,"processed_tiles": set()}
 
 
 def make_tiles(image, tile_size=TILE_SIZE):
@@ -23,6 +23,17 @@ def make_tiles(image, tile_size=TILE_SIZE):
             cv2.rectangle(annotated, (x, y), (x+tile_size, y+tile_size), (255,0,0), 2)
             cv2.putText(annotated, str(tile_id), (x+50, y+50),
                         cv2.FONT_HERSHEY_SIMPLEX, 2, (0,0,0), 5)
+            
+            # Shade tile if already processed
+            if tile_id in tiles_cache["processed_tiles"]:
+                overlay = annotated[y:y+tile_size, x:x+tile_size].copy()
+                overlay[:] = (0, 255, 0)  # light green
+                alpha = 0.4
+                annotated[y:y+tile_size, x:x+tile_size] = cv2.addWeighted(
+                    overlay, alpha, annotated[y:y+tile_size, x:x+tile_size], 1-alpha, 0
+                )
+
+
             tile_id += 1
     return annotated, tiles
 
@@ -55,12 +66,14 @@ def select_tile(evt: gr.SelectData,state):
         # store the path, not the array
         tiles_cache["selected_tile"] = {
             "tile_path": tile_path,
-            "coords": coords
+            "coords": coords,
+            "tile_id": tile_id
         }
 
         updated_state = {
             "tile_path": tile_path,
-            "coords": coords
+            "coords": coords,
+            "tile_id": tile_id
         }
 
         # load tile only for display
@@ -74,6 +87,17 @@ def select_tile(evt: gr.SelectData,state):
 
 
 
+# ---- Mark a tile as processed and update annotated output ----
+def mark_tile_processed(original_image):
+    tile_info = tiles_cache.get("selected_tile")
+    if not tile_info:
+        return original_image
+
+    tile_id = tile_info["tile_id"]
+    tiles_cache["processed_tiles"].add(tile_id)
+
+    annotated, _ = make_tiles(original_image, TILE_SIZE)
+    return annotated
 
 
 def get_inference_widgets(run_inference,georefImg):
@@ -102,22 +126,30 @@ def get_inference_widgets(run_inference,georefImg):
             output = gr.Textbox(label="Progress", lines=5, interactive=False)
             download_file = gr.File(label="Download CSV")
 
-    selected_tile_path = gr.State()
-
+    selected_tile_state  = gr.State()
+    original_image_state = gr.State()
 
     # Wire events
     create_btn.click(
-        fn=create_tiles, inputs=image_input,
+        fn=lambda file: (np.array(Image.open(file.name).convert("RGB")), gr.update(interactive=False)),
+        inputs=image_input,
         outputs=[annotated_out, run_button]
     )
+
     annotated_out.select(
-        fn=select_tile, inputs=[selected_tile_path],
-        outputs=[selected_tile, run_button, selected_tile_path]
+        fn=select_tile,
+        inputs=[selected_tile_state],
+        outputs=[selected_tile, run_button, selected_tile_state]
     )
+
     run_button.click(
-        fn=run_inference,
-        inputs=[selected_tile_path, gcp_input, city_name, score_th],
-        outputs=[output, download_file]
+        fn=lambda state, gcp, city, threshold: (
+            mark_tile_processed(np.array(Image.open(state["tile_path"]).convert("RGB"))),
+            "Tile processed",
+            None
+        ),
+        inputs=[selected_tile_state, gcp_input, city_name, score_th],
+        outputs=[annotated_out, output, download_file]
     )
 
     georef_btn.click(
@@ -126,5 +158,4 @@ def get_inference_widgets(run_inference,georefImg):
         outputs=[output]
     )
 
-
     return image_input, gcp_input, city_name, score_th, run_button, output, download_file