with open("tests/test_visual_orchestration.py", "r", encoding="utf-8", newline="") as f:
    content = f.read()

# Add more waits and verify provider was set
content = content.replace(
    """# Setup provider
    client.set_value("current_provider", "gemini_cli")
    time.sleep(0.5)
    client.set_value(
        "gcli_path",
        f'"{sys.executable}" "{os.path.abspath("tests/mock_gemini_cli.py")}"',
    )
    time.sleep(0.5)
    client.set_value("manual_approve", True)
    time.sleep(0.5)""",
    """# Setup provider - verify it took
    client.set_value("current_provider", "gemini_cli")
    time.sleep(1)
    # Verify provider is set correctly
    mma = client.get_mma_status()
    print(f"[Test] After setting provider: {mma.get('mma_tier_usage', {})}")
    client.set_value(
        "gcli_path",
        f'"{sys.executable}" "{os.path.abspath("tests/mock_gemini_cli.py")}"',
    )
    time.sleep(1)
    client.set_value("manual_approve", True)
    time.sleep(1)
    print(f"[Test] After all setup: {client.get_mma_status().get('mma_tier_usage', {})}")""",
)

with open("tests/test_visual_orchestration.py", "w", encoding="utf-8", newline="") as f:
    f.write(content)

print("Added more debug to test")