Update app.py

app.py

@@ -102,7 +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
+        extract_file_ids(step_details) #XXXXX
         
         show_json("step_details", step_details)
 
@@ -116,15 +116,26 @@ def get_messages(client, thread):
     return messages
 
 #XXXXX
-def extract_file_id(step_details):
-    for step in step_details:
-        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_file_ids(data: Dict[str, Any]) -> List[str]:
+    file_ids = []
+    steps = data.get('data', [])
+
+    for step in steps:
+        step_details = step.get('step_details', {})
+        tool_calls = step_details.get('tool_calls', [])
+
+        for tool_call in tool_calls:
+            code_interpreter = tool_call.get('code_interpreter', {})
+            outputs = code_interpreter.get('outputs', [])
+
+            for output in outputs:
+                if 'image' in output:
+                    file_id = output['image'].get('file_id')
+                    if file_id:
+                        file_ids.append(file_id)
+                        print(f"\n\n\n\n\n{file_id}")
+    
+    return file_ids
                         
 def extract_content_values(data):
     content_values = []