Upload 37 files

src/components/MultiSourceCaptioningView.tsx

@@ -17,28 +17,29 @@ function parseFlatBoxArray(arr: any[]): { label: string, bbox_2d: number[] }[] {
 }
 
 function normalizeBoxes(raw: any): { label: string, bbox_2d: number[] }[] {
+  if (!raw) return [];
   let boxes = [];
   if (Array.isArray(raw)) {
     boxes = raw;
   } else if (typeof raw === "object" && raw !== null) {
     boxes = [raw];
   }
-  // Normalize bbox_2d to flat [x1, y1, x2, y2] only if valid
-  return boxes.map(obj => {
-    let bbox = obj.bbox_2d;
-    if (
-      Array.isArray(bbox) &&
-      bbox.length === 2 &&
-      Array.isArray(bbox[0]) &&
-      Array.isArray(bbox[1]) &&
-      bbox[0].length === 2 &&
-      bbox[1].length === 2
-    ) {
-      // Convert [[x1, y1], [x2, y2]] to [x1, y1, x2, y2]
-      bbox = [bbox[0][0], bbox[0][1], bbox[1][0], bbox[1][1]];
-    }
-    return { ...obj, bbox_2d: bbox };
-  });
+  return boxes
+    .filter(obj => obj && obj.bbox_2d)
+    .map(obj => {
+      let bbox = obj.bbox_2d;
+      if (
+        Array.isArray(bbox) &&
+        bbox.length === 2 &&
+        Array.isArray(bbox[0]) &&
+        Array.isArray(bbox[1]) &&
+        bbox[0].length === 2 &&
+        bbox[1].length === 2
+      ) {
+        bbox = [bbox[0][0], bbox[0][1], bbox[1][0], bbox[1][1]];
+      }
+      return { ...obj, bbox_2d: bbox };
+    });
 }
 
 function isImageFile(file: File) {
@@ -91,7 +92,9 @@ export default function MultiSourceCaptioningView() {
       }
       boxes = normalizeBoxes(boxes);
       if (boxes.length === 0) setInferenceStatus("No boxes detected or model output invalid.");
-      drawBoundingBoxesOnCanvas(ctx, boxes);
+      if (Array.isArray(boxes) && boxes.length > 0) {
+        drawBoundingBoxesOnCanvas(ctx, boxes);
+      }
     });
   };
 
@@ -212,7 +215,9 @@ export default function MultiSourceCaptioningView() {
       }
       boxes = normalizeBoxes(boxes);
       if (boxes.length === 0) setInferenceStatus("No boxes detected or model output invalid.");
-      drawBoundingBoxesOnCanvas(ctx, boxes);
+      if (Array.isArray(boxes) && boxes.length > 0) {
+        drawBoundingBoxesOnCanvas(ctx, boxes);
+      }
       setImageProcessed(true);
     });
     setProcessing(false);