import sys
sys.stdout.reconfigure(encoding='utf-8')
from playwright.sync_api import sync_playwright

with sync_playwright() as p:
    browser = p.chromium.connect_over_cdp('http://localhost:9222')
    context = browser.contexts[0]
    page = context.pages[0]
    
    # 1. Click No button directly on the span
    print('Clicking No button...')
    page.locator("text='No'").nth(1).click()
    page.wait_for_timeout(2000)
    
    # 2. Click Home town
    print('Clicking Home town button...')
    home_btn = page.locator("div[role='button']:has-text('Home town'), span:has-text('Home town')").first
    home_btn.click()
    page.wait_for_timeout(2000)
    
    page.screenshot(path=r'C:\Projects\Historical\FacebookAutoPoster\hometown_opened.png')
    print('Saved hometown_opened.png!')
