BGR color code issue

app.py

@@ -151,7 +151,7 @@ def extract_and_paste_insect(original_image: np.ndarray, detection: DetectionRes
     background[y_offset:y_end, x_offset:x_end] = insect
 
 def create_yellow_background_with_insects(image: np.ndarray, detections: List[DetectionResult]) -> np.ndarray:
-    yellow_background = np.full((image.shape[0], image.shape[1], 3), (0, 255, 255), dtype=np.uint8)
+    yellow_background = np.full((image.shape[0], image.shape[1], 3), (0, 255, 255), dtype=np.uint8)  # BGR for yellow
     for detection in detections:
         if detection.mask is not None:
             extract_and_paste_insect(image, detection, yellow_background)
@@ -191,28 +191,18 @@ def detections_to_json(detections):
         detections_list.append(detection_dict)
     return detections_list
 
-def process_image(image, include_json, include_raw):
+def process_image(image, include_json):
     labels = ["insect"]
     original_image, detections = grounded_segmentation(image, labels, threshold=0.3, polygon_refinement=True)
     yellow_background_with_insects = create_yellow_background_with_insects(np.array(original_image), detections)
-    
-    output1 = yellow_background_with_insects
-    output2 = None
-    output3 = None
-    output4 = None
-    
     if include_json:
         detections_json = detections_to_json(detections)
         json_output_path = "insect_detections.json"
         with open(json_output_path, 'w') as json_file:
             json.dump(detections_json, json_file, indent=4)
-        output2 = json.dumps(detections_json, separators=(',', ':'))
-    
-    if include_raw:
-        output3 = original_image
-        output4 = detections
-    
-    return output1, output2, output3, output4
+        return yellow_background_with_insects, json.dumps(detections_json, separators=(',', ':'))
+    else:
+        return yellow_background_with_insects, None
 
 examples = [
     ["flower-night.jpg"]
@@ -220,8 +210,8 @@ examples = [
 
 gr.Interface(
     fn=process_image,
-    inputs=[gr.Image(type="pil"), gr.Checkbox(label="Include JSON", value=False), gr.Checkbox(label="Include Raw Outputs", value=False)],
-    outputs=[gr.Image(type="numpy"), gr.Textbox(), gr.Image(type="numpy"), gr.JSON()],
+    inputs=[gr.Image(type="pil"), gr.Checkbox(label="Include JSON", value=False)],
+    outputs=[gr.Image(type="numpy"), gr.Textbox()],
     title="InsectSAM 🐞",
     examples=examples
 ).launch()