Update app.py

app.py

@@ -100,34 +100,16 @@ def get_run_steps(client, thread, run):
     return run_steps
 
 def get_run_step_details(run_steps):
+    run_step_details = []
+    
     for step in run_steps.data:
         step_details = step.step_details
+        run_step_details.append(step_details)
         
         show_json("step_details", step_details)
 
-        ###
-        if hasattr(step_details, "tool_calls"):
-            for tool_call in step_details.tool_calls:
-                print("\n\n")
-                show_json("tool_call", tool_call)
-                print("\n\n")
-                if hasattr(tool_call, "code_interpreter"):
-                    print(tool_call.code_interpreter)
-                    print("\n\n")
-                    if hasattr(tool_call.code_interpreter, "outputs"):
-                        print(tool_call.code_interpreter.outputs)
-                        # [CodeInterpreterOutputImage(image=CodeInterpreterOutputImageImage(file_id='file-qmr1d6ObICSjyGvmoncXvBPj'), type='image')]
-                        if len(tool_call.code_interpreter.outputs) > 0:
-                            if hasattr(tool_call.code_interpreter.outputs[0], "image"):
-                                print("\n111")
-                                file_id = tool_call.code_interpreter.outputs[0].image.file_id
-                                download = f"Download file here: https://platform.openai.com/storage/files/{file_id}"
-                                print("\n222")
-                                print(download)
-                                print("\n333")
-                                #img = generate_markdown_image(content.arrayBuffer())
-                                return download
-                
+    return run_step_details
+
 def get_messages(client, thread):
     messages = client.beta.threads.messages.list(
         thread_id=thread.id
@@ -143,9 +125,11 @@ def extract_content_values(data):
     for item in data.data:
         for content in item.content:
             if content.type == "text":
-                text_values.append(content.text.value)
+                text_value = content.text.value
+                text_values.append(text_value)
             if content.type == "image_file":
-                image_values.append(content.image_file.file_id)
+                image_value = content.image_file.file_id
+                image_values.append(image_value)
     
     return text_values, image_values
 
@@ -174,13 +158,17 @@ def chat(message, history):
 
     text_values, image_values = extract_content_values(messages)
 
+    download_link = ""
+    
     print("###")
-    print(text_values)
+    print(text_values[0])
     print("###")
-    print(image_values)
+    if len(image values) > 0:
+        print(image_values[0])
+        download_link = f"https://platform.openai.com/storage/files/{image_values[0]}"
     print("###")
     
-    return {text_values[1]}
+    return f"{text_values[0]}{download_link}"
 
 gr.ChatInterface(
         fn=chat,