handle tiles with no labels

inference_tab/inference_logic.py

@@ -80,6 +80,14 @@ def run_inference(tile_dict, gcp_path,user_crs, city_name, score_th, hist_th, hi
         else:
             log += msg + "\n"
             yield log, None
+        if "No labels detected" in msg:
+            stop_pipeline = True
+            break
+
+    if stop_pipeline:
+        yield log + "Pipeline stopped: no text segments found.\n", None
+        return
+
     for msg in extractSegments(image_path):
         log += msg + "\n"
         yield log, None
@@ -238,6 +246,11 @@ def run_tile_inference():
 
         tile_array = np.array(Image.open(tile_path))
 
+        # If there are no boxes and no labels, stop execution
+        if not local_boxes and not point_coords and not point_labels:
+            yield f"Tile {tile['idx']} has no boxes or points/labels. Stopping inference."
+            return
+
         results = model(tile_array, bboxes=local_boxes,
                         points=point_coords, labels=point_labels)
 
@@ -265,6 +278,9 @@ def getSegments(image_path,iou=0.5,c_th=0.75,edge_margin=10):
     yield "Run inference on tiles..."
     for msg in run_tile_inference():
         yield msg
+        if "Stopping inference" in msg:
+            yield "No labels detected – halting getSegments."
+            return
     yield "Marge predicted masks into image..."
     merge_tile_masks(H,W)