diff --git a/simulation/ping_pong.py b/simulation/ping_pong.py index 0b6f5d7..7c3da89 100644 --- a/simulation/ping_pong.py +++ b/simulation/ping_pong.py @@ -21,8 +21,7 @@ def main(): time.sleep(2) # Give it time to clear # 2. Initial message initial_msg = "Hello! I want to create a simple python script that prints 'Hello World'. Can you help me?" - print(f" - [USER]: {initial_msg}") + print(f" [USER]: {initial_msg}") client.set_value("ai_input", initial_msg) client.click("btn_gen_send") # 3. Wait for AI response @@ -37,14 +36,11 @@ def main(): # Something happened last_entry = entries[-1] if last_entry.get('role') == 'AI' and last_entry.get('content'): - print(f" - [AI]: {last_entry.get('content')[:100]}...") - print(" - Ping-pong successful!") + print(f" [AI]: {last_entry.get('content')[:100]}...") + print(" Ping-pong successful!") return last_entry_count = len(entries) - print(" - Timeout waiting for AI response") + print(" Timeout waiting for AI response") if __name__ == "__main__": main()