Update app.py

app.py

@@ -102,6 +102,7 @@ def get_run_steps(client, thread, run):
 def get_run_step_details(run_steps):
     for step in run_steps.data:
         step_details = step.step_details
+        extract_file_id(step_details) #XXXXX
         
         show_json("step_details", step_details)
 
@@ -113,7 +114,18 @@ def get_messages(client, thread):
     show_json("messages", messages)
     
     return messages
-    
+
+#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}")
+                        
 def extract_content_values(data):
     content_values = []