import sys, time
from pathlib import Path
sys.path.append(str(Path("05_Production_Pipeline").resolve()))
from invisible_playwright import InvisiblePlaywright
from muse_invpw_driver import sync_worker_profile, DEFAULT_PROFILE

def check_thread_now():
    sync_worker_profile()
    with InvisiblePlaywright(profile_dir=DEFAULT_PROFILE, headless=True) as ctx:
        page = ctx.pages[0] if ctx.pages else ctx.new_page()
        page.goto("https://muse.ai", wait_until="domcontentloaded", timeout=30000)
        time.sleep(4)
        
        page.screenshot(path="04_Assets/thread_state_now.png")
        print("Screenshot saved to 04_Assets/thread_state_now.png")
        
        # Check all download buttons
        dl_btns = page.locator('button[aria-label="Tải video xuống"]').all()
        print(f"Total download buttons: {len(dl_btns)}")
        
        # Check all assistant messages
        msgs = page.locator('[data-message-author="assistant"], [role="article"]').all()
        print(f"Total assistant messages: {len(msgs)}")
        for i, m in enumerate(msgs):
            txt = m.inner_text().strip().replace('\n', ' ')
            print(f"Msg [{i}]: {txt[:150]}")

if __name__ == '__main__':
    check_thread_now()