annot

app.py

@@ -146,11 +146,14 @@ with gr.Blocks() as demo:
             with open('./assets/intervention_test_images/annot.json', 'r') as f:
                 points_json = json.load(f)
 
+            resized_img = resize_to_square(img)
+
+            if os.path.basename(img_path) not in points_json:
+                return resized_img, resized_img, img, []
+
             points_json = points_json[os.path.basename(img_path)]
 
             # print(f"Image uploaded with shape: {input.shape}")
-            resized_img = resize_to_square(img)
-
             temp = resized_img.copy()
 
             # Redraw all remaining arrows and dots

assets/intervention_test_images/annot.json

@@ -1 +1,16 @@
-{"bread.jpg": [[120, 257], [175, 269], [328, 375], [266, 353], [410, 217], [341, 248], [228, 149], [248, 211], [152, 129], [152, 129], [108, 51], [108, 51], [342, 39], [342, 39], [479, 93], [479, 93], [477, 390], [477, 390], [229, 486], [229, 486], [58, 442], [58, 442]]}
+{
+  "bread.jpg": [[120, 257], [175, 269], [328, 375], [266, 353], [410, 217], [341, 248], [228, 149], [248, 211], [152, 129], [152, 129], [108, 51], [108, 51], [342, 39], [342, 39], [479, 93], [479, 93], [477, 390], [477, 390], [229, 486], [229, 486], [58, 442], [58, 442]],
+
+  "bird.jpg":  [[252, 284], [333, 279], [251, 213], [332, 209], [426, 379], [426, 379], [452, 116], [452, 116], [188, 66], [188, 66], [89, 436], [89, 436]],
+
+  "ducks.jpg": [[235, 385], [184, 433], [168, 131], [114, 174], [434, 27], [434, 27], [403, 187], [403, 187], [477, 336], [477, 336], [29, 426], [29, 426]],
+
+  "robot_arm.jpg": [[236, 236], [189, 275], [432, 125], [432, 125], [481, 304], [481, 304], [24, 419], [24, 419], [63, 111], [63, 111]],
+
+  "desk_1.jpg": [[167, 224], [127, 222], [272, 327], [290, 369], [383, 251], [407, 212], [53, 344], [53, 344], [86, 465], [86, 465], [476, 466], [476, 466], [475, 121], [475, 121], [277, 85], [277, 85]],
+
+  "glasses.jpg": [[217, 249], [302, 274], [245, 45], [245, 45], [282, 93], [282, 93], [431, 111], [431, 111], [467, 392], [467, 392], [89, 462], [89, 462], [30, 361], [30, 361]],
+
+  "watering_pot.jpg": [[239, 293], [309, 280], [454, 118], [469, 169], [425, 32], [425, 32], [466, 47], [466, 47], [383, 117], [383, 117], [495, 300], [495, 300], [304, 39], [304, 39], [34, 115], [34, 115], [94, 400], [94, 400], [281, 476], [281, 476]]
+
+}