Update app.py

app.py

@@ -117,14 +117,16 @@ def get_messages(client, thread):
 
 #XXXXX
 def extract_file_id(data):
-    for step in data["data"]:
-        if "step_details" in step and "tool_calls" in step["step_details"]:
-            for tool_call in step["step_details"]["tool_calls"]:
-                if "code_interpreter" in tool_call and "outputs" in tool_call["code_interpreter"]:
-                    for output in tool_call["code_interpreter"]["outputs"]:
-                        if "image" in output and "file_id" in output["image"]:
-                            file_id = output["image"]["file_id"]
-                            print(f"#########################################Found file_id: {file_id}")
+    data = json.loads(data)
+
+    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'):
+                    for output in tool_call.code_interpreter.outputs:
+                        if hasattr(output, 'image') and hasattr(output.image, 'file_id'):
+                            file_id = output.image.file_id
+                            print(f"#############################################3Found file_id: {file_id}")
                         
 def extract_content_values(data):
     content_values = []