import sys
import time
from pathlib import Path

sys.path.insert(0, '05_Production_Pipeline')
if sys.platform == 'win32':
    sys.stdout.reconfigure(encoding='utf-8')
from invisible_playwright import InvisiblePlaywright

pdir = Path.home() / 'AntigravityProfiles' / 'phu2234_worker'
with InvisiblePlaywright(profile_dir=pdir, headless=True) as ctx:
    page = ctx.pages[0] if ctx.pages else ctx.new_page()
    page.goto('https://muse.ai/thread/new', wait_until='domcontentloaded', timeout=25000)
    time.sleep(3)
    
    # Click Shot 01 button
    btn = page.locator('button:has-text("làng quê 1980")').first
    if btn.count() > 0:
        btn.click()
        print('Clicked Shot 01 button!')
        time.sleep(4)
        page.screenshot(path='scratch/shot01_thread_view.png')
        print('Saved scratch/shot01_thread_view.png')
        
        # Check what buttons or links appeared
        elements = page.evaluate("""
            () => {
                const results = [];
                document.querySelectorAll('button, a, video').forEach(el => {
                    results.push({
                        tag: el.tagName,
                        text: (el.innerText || '').trim().substring(0, 40),
                        aria: el.getAttribute('aria-label') || '',
                        src: el.src || el.getAttribute('href') || ''
                    });
                });
                return results;
            }
        """)
        print(f'Total interactive elements: {len(elements)}')
        for el in elements:
            if any(w in (el['text'] + el['aria']).lower() for w in ['tải', 'download', 'xem', 'video', 'lưu']):
                print(' -', el)