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:
    shorts_url = "https://studio.youtube.com/channel/UClZnv08RDTD4vgLOw1jmOuA/videos/short"
    print("Navigating to Shorts tab:", shorts_url)
    manager.page.goto(shorts_url, wait_until="domcontentloaded", timeout=45000)
    time.sleep(6)

    bypass = manager.page.locator('a:has-text("CHUYỂN THẲNG ĐẾN YOUTUBE STUDIO"), a:has-text("CONTINUE TO YOUTUBE STUDIO")').first
    if bypass.is_visible(timeout=3000):
        bypass.click()
        time.sleep(6)

    manager.capture_status("downloads/studio_shorts_final_overview.png")
    print("Screenshot saved to downloads/studio_shorts_final_overview.png")

finally:
    manager.close()