Update app.py

app.py

@@ -13,7 +13,7 @@
 # https://platform.openai.com/docs/assistants/tools
 
 import gradio as gr
-import json, openai, os, time
+import openai, os, time
 
 from openai import OpenAI
 from utils import show_json
@@ -117,14 +117,12 @@ def get_messages(client, thread):
 
 #XXXXX
 def extract_file_id(data):
-    data = json.loads(data)
-
-    for step in data['data']:
-        if hasattr(step['step_details'], 'tool_calls'):
+    for step in data["data"]:
+        if hasattr(step["step_details"], "tool_calls"):
             for tool_call in step['step_details'].tool_calls:
-                if hasattr(tool_call, 'code_interpreter') and hasattr(tool_call.code_interpreter, 'outputs'):
+                if hasattr(tool_call, "code_interpreter") and hasattr(tool_call.code_interpreter, "outputs"):
                     for output in tool_call.code_interpreter.outputs:
-                        if hasattr(output, 'image') and hasattr(output.image, 'file_id'):
+                        if hasattr(output, "image") and hasattr(output.image, "file_id"):
                             file_id = output.image.file_id
                             print(f"#############################################3Found file_id: {file_id}")