Update app.py

app.py

@@ -119,21 +119,19 @@ def get_messages(client, thread):
 def extract_file_ids(step_details):
     file_ids = []
 
-    for step in step_details:
-        step_details = step.get('step_details', {})
-        tool_calls = step_details.get('tool_calls', [])
+    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 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}")
 
-            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):