output original_image and detections

app.py

@@ -200,9 +200,9 @@ def process_image(image, include_json):
         json_output_path = "insect_detections.json"
         with open(json_output_path, 'w') as json_file:
             json.dump(detections_json, json_file, indent=4)
-        return yellow_background_with_insects, json.dumps(detections_json, separators=(',', ':'))
+        return yellow_background_with_insects, json.dumps(detections_json, separators=(',', ':')), original_image, detections
     else:
-        return yellow_background_with_insects, None
+        return yellow_background_with_insects, None, original_image, detections
 
 examples = [
     ["flower-night.jpg"]
@@ -211,7 +211,7 @@ examples = [
 gr.Interface(
     fn=process_image,
     inputs=[gr.Image(type="pil"), gr.Checkbox(label="Include JSON", value=False)],
-    outputs=[gr.Image(type="numpy"), gr.Textbox()],
+    outputs=[gr.Image(type="numpy"), gr.Textbox(), gr.Image(type="numpy"), gr.JSON()],
     title="InsectSAM 🐞",
     examples=examples
 ).launch()