Update app.py

app.py

@@ -1,3 +1,4 @@
+import json
 import os
 
 import gradio as gr
@@ -33,18 +34,15 @@ def parse_json(json_output):
             json_output = "\n".join(lines[i+1:])  # Remove everything before "```json"
             json_output = json_output.split("```")[0]  # Remove everything after the closing "```"
             break  # Exit the loop once "```json" is found
-    return json_output
+    return json.loads(json_output)
 
 def parse_info(image, json_data):
     width, height = image.size
     df_data = []
     boxes_with_labels = []
 
-    if not isinstance(json_data, list):
-        return boxes_with_labels, pd.DataFrame(df_data)
-
     # Iterate over each detected action actions
-    for action in info:
+    for action in json_data:
         box_2d = action.get("box_2d")
         label = action.get("label")
         co2_grams = action.get("co2_grams")