Update app.py

app.py

@@ -136,8 +136,8 @@ def execute_tool_call(tool_call):
 def execute_tool_calls(run_steps):
     run_step_details = []
 
-    tool_call_id = []
-    tool_call_result = []
+    tool_call_ids = []
+    tool_call_results = []
     
     for step in run_steps.data:
         step_details = step.step_details
@@ -149,10 +149,10 @@ def execute_tool_calls(run_steps):
                 show_json("tool_call", tool_call)
                 
                 if hasattr(tool_call, "function"):
-                    tool_call_id.append(tool_call.id)
-                    tool_call_result.append(execute_tool_call(tool_call))
+                    tool_call_ids.append(tool_call.id)
+                    tool_call_results.append(execute_tool_call(tool_call))
 
-    return tool_call_id, tool_call_result
+    return tool_call_ids, tool_call_results
 
 def get_messages(client, thread):
     messages = client.beta.threads.messages.list(
@@ -177,6 +177,28 @@ def extract_content_values(data):
     
     return text_values, image_values
 
+###
+def generate_tool_outputs(tool_call_ids, tool_call_results):
+    """
+    Generate tool outputs from tool call IDs and results.
+
+    Args:
+        tool_call_ids (list): List of tool call IDs.
+        tool_call_results (list): List of tool call results.
+
+    Returns:
+        list: List of tool outputs.
+    """
+    tool_outputs = []
+    for tool_call_id, tool_call_result in zip(tool_call_ids, tool_call_results):
+        tool_output = {
+            "tool_call_id": tool_call_id,
+            "output": tool_call_result
+        }
+        tool_outputs.append(tool_output)
+    return tool_outputs
+###
+
 def chat(message, history):
     if not message:
         raise gr.Error("Message is required.")
@@ -196,23 +218,15 @@ def chat(message, history):
 
     run_steps = get_run_steps(client, thread, run)
 
-    tool_call_id, tool_call_result = execute_tool_calls(run_steps)
+    tool_call_ids, tool_call_results = execute_tool_calls(run_steps)
 
     ### TODO
-    if tool_call_id:
-        print(tool_call_id)
-        print(tool_call_result.to_json())
-
+    if tool_call_ids[0]:
         # https://platform.openai.com/docs/api-reference/runs/submitToolOutputs
         run = client.beta.threads.runs.submit_tool_outputs(
             thread_id=thread.id,
             run_id=run.id,
-            tool_outputs=[
-                {
-                  "tool_call_id": tool_call_id,
-                  "output": tool_call_result
-                }
-            ]
+            tool_outputs=generate_tool_outputs(tool_call_ids, tool_call_results)
         )
     
         run = wait_on_run(client, thread, run)