Update app.py

app.py

@@ -80,7 +80,7 @@ class ComputerUseDemo:
                 )
             elif tool_name == "str_replace_editor":
                 return ToolResult(
-                    output=f"Executed text editor command: {tool_input}"
+                    output=f"Created/modified file with content: {tool_input}"
                 )
             elif tool_name == "bash":
                 return ToolResult(
@@ -111,12 +111,14 @@ class ComputerUseDemo:
             # Create system prompt for computer use
             base_system_prompt = """You have access to a computing environment through specialized tools.
             Use the computer tool to interact with the GUI, the text editor for file operations, 
-            and bash for command-line tasks. Take screenshots after significant actions and 
-            verify the results carefully. Always handle errors gracefully and inform the user 
-            of any issues."""
+            and bash for command-line tasks. After each action, explicitly describe what you did and what happened.
+            Always show the results of your actions."""
             
             full_system_prompt = f"{base_system_prompt}\n{system_prompt}" if system_prompt else base_system_prompt
 
+            conversation_log = []
+            conversation_log.append((message, ""))  # Add user message
+
             while True:
                 # Call Claude API with computer use enabled
                 response = anthropic.beta.messages.create(
@@ -128,17 +130,30 @@ class ComputerUseDemo:
                     betas=["computer-use-2024-10-22"]
                 )
 
+                # Get the assistant's response text
+                assistant_message = response.content[0].text if response.content else ""
+                
                 # 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)
 
+                # Add assistant's initial response to conversation
+                if assistant_message:
+                    conversation_log.append(("", assistant_message))
+
                 # 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 execution to conversation
+                    if tool_result.error:
+                        conversation_log.append(("", f"Error: {tool_result.error}"))
+                    else:
+                        conversation_log.append(("", f"Tool execution: {tool_result.output}"))
+
                     # Add tool interactions to message history
                     messages.append({
                         "role": "assistant",
@@ -149,12 +164,9 @@ class ComputerUseDemo:
                         "content": [tool_result.to_dict()]
                     })
                 else:
-                    # Final response received
-                    assistant_response = response.content[0].text
-                    chat_history.append((message, assistant_response))
                     break
 
-            return chat_history, None
+            return conversation_log, None
             
         except Exception as e:
             return chat_history, f"Error processing message: {str(e)}"