import sys, os, time
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
from youtube_studio_manager import YouTubeStudioManager

manager = YouTubeStudioManager(mode='chrome_stealth')
manager.start(headless=False)

try:
    edit_url = "https://studio.youtube.com/video/MtB7VLHfcDk/edit"
    manager.page.goto(edit_url, wait_until="domcontentloaded", timeout=45000)
    time.sleep(5)

    # Use mouse wheel to scroll down smoothly
    manager.page.mouse.move(500, 500)
    manager.page.mouse.wheel(0, 600)
    time.sleep(2)
    manager.capture_status("downloads/studio_short_edit_pagedown.png")

    # Get outerHTML of the thumbnail section
    thumb_html = manager.page.evaluate("""
    () => {
        const h3 = Array.from(document.querySelectorAll('h3, div, span')).find(el => el.textContent.trim() === 'Hình thu nhỏ' || el.textContent.trim() === 'Thumbnail');
        if (h3 && h3.parentElement) {
            return h3.parentElement.parentElement ? h3.parentElement.parentElement.outerHTML : h3.parentElement.outerHTML;
        }
        return 'Not found';
    }
    """)
    print("Thumbnail section HTML snippet:")
    print(thumb_html[:1500])

    # Get outerHTML of related video section
    related_html = manager.page.evaluate("""
    () => {
        const label = Array.from(document.querySelectorAll('*')).find(el => (el.textContent || '').includes('Video có liên quan') || (el.textContent || '').includes('Related video'));
        if (label) {
            let p = label;
            for (let i = 0; i < 3; i++) {
                if (p.parentElement) p = p.parentElement;
            }
            return p.outerHTML;
        }
        return 'Not found';
    }
    """)
    print("\nRelated video section HTML snippet:")
    print(related_html[:1500])

finally:
    manager.close()