Update app.py

app.py

@@ -128,15 +128,21 @@ class ComputerUseDemo:
                     betas=["computer-use-2024-10-22"]
                 )
 
-                # Handle tool calls
-                if response.stop_reason == "tool_use" and response.tool_calls:
-                    tool_call = response.tool_calls[0]
-                    tool_result = self.execute_tool(tool_call.name, tool_call.arguments)
+                # Check for tool calls in the response
+                tool_calls = []
+                for content in response.content:
+                    if hasattr(content, 'tool_calls') and content.tool_calls:
+                        tool_calls.extend(content.tool_calls)
+
+                # Handle tool calls if present
+                if tool_calls:
+                    tool_call = tool_calls[0]  # Handle first tool call
+                    tool_result = self.execute_tool(tool_call.name, tool_call.parameters)
                     
                     # Add tool interactions to message history
                     messages.append({
                         "role": "assistant",
-                        "content": [{"type": "tool_use", "name": tool_call.name, "input": tool_call.arguments}]
+                        "content": [{"type": "tool_use", "name": tool_call.name, "input": tool_call.parameters}]
                     })
                     messages.append({
                         "role": "user",