runner.py

import sys, os, json, time, threading, re
import undetected_chromedriver as uc
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import TimeoutException

# Cấu hình chung
CURRENT_VERSION = "1.0.3"
UPDATE_CHECK_URL = "https://drive.google.com/uc?export=download&id=1hkRegTsVohMNxM7o33Jj3FRix0GyEhUW"
TARGET_URL = "https://hailuoai.video/create"
COOKIES_FILE = "cookies.json"
IMAGE_FOLDER = "images"
INPUT_TEXT_FILE = "input.txt"
DOWNLOAD_TIMEOUT = 60  # giây

def normalize_text(text):
    replacements = {
        "‘": "'", "’": "'", "“": '"', "”": '"', "…": "..."
    }
    for old, new in replacements.items():
        text = text.replace(old, new)
    return re.sub(r'\s+', ' ', text).strip().lower()

class AutomationRunner:
    def __init__(self, headless, wait_time, avatar_image_path=None, video_folder="outputs", mode="subject", log_callback=None):
        self.headless = headless
        self.wait_time = wait_time
        self.avatar_image_path = avatar_image_path
        self.video_folder = video_folder
        self.mode = mode
        self.log = log_callback or (lambda msg: None)

    def setup_driver(self):
        self.log("🔧 Khởi tạo trình duyệt...")
        options = uc.ChromeOptions()
        if self.headless:
            options.add_argument('--headless=new')
            options.add_argument('--window-size=1920,1080')
        os.makedirs(self.video_folder, exist_ok=True)
        prefs = {
            "download.default_directory": os.path.abspath(self.video_folder),
            "download.prompt_for_download": False,
            "download.directory_upgrade": True,
        }
        options.add_experimental_option("prefs", prefs)
        driver = uc.Chrome(options=options)
        if not self.headless:
            driver.maximize_window()
        time.sleep(2)
        return driver

    def save_cookies(self, driver):
        with open(COOKIES_FILE, 'w', encoding='utf-8') as f:
            json.dump(driver.get_cookies(), f, ensure_ascii=False, indent=2)
        self.log(f"✅ Lưu cookies vào {COOKIES_FILE}")

    def load_cookies(self, driver):
        try:
            cookies = json.load(open(COOKIES_FILE, 'r', encoding='utf-8'))
            for c in cookies:
                # adjust SameSite
                if c.get('sameSite') == 'None':
                    c['sameSite'] = 'Strict'
                try:
                    driver.add_cookie(c)
                except Exception:
                    pass
            self.log("✅ Nạp cookies thành công")
        except FileNotFoundError:
            self.log("⚠️ Chưa có file cookies.json, sẽ login thủ công")

    def click_tab(self, driver, xpath, success_msg, fail_msg):
        try:
            el = WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, xpath)))
            el.click()
            self.log(success_msg)
        except Exception as e:
            self.log(f"{fail_msg}: {e}")

    def click_confirm(self, driver):
        self.click_tab(
            driver,
            "//*[contains(text(),'Confirm')]",
            "✅ Nhấn Confirm",
            "❌ Không tìm thấy Confirm"
        )

    def upload_images_silently(self, driver):
        try:
            btn = WebDriverWait(driver, 10).until(
                EC.element_to_be_clickable((
                    By.XPATH,
                    "//div[contains(text(),'Add reference character') and contains(@class,'hover:text-white')]"
                ))
            )
            btn.click()
            time.sleep(2)
            inp = driver.find_element(By.XPATH, "//input[@type='file']")
            driver.execute_script("arguments[0].style.display='block';", inp)
            paths = []
            if self.avatar_image_path:
                paths = [os.path.abspath(self.avatar_image_path)]
            else:
                imgs = [f for f in os.listdir(IMAGE_FOLDER) if f.lower().endswith((".png",".jpg",".jpeg"))]
                paths = [os.path.abspath(os.path.join(IMAGE_FOLDER,f)) for f in imgs]
            inp.send_keys("\n".join(paths))
            time.sleep(1)
            self.click_confirm(driver)  # modal upload
            self.log(f"✅ Upload ảnh: {paths}")
        except Exception as e:
            self.log(f"⚠️ Upload ảnh thất bại: {e}")

    def load_scenes(self):
        out = []
        with open(INPUT_TEXT_FILE, 'r', encoding='utf-8') as f:
            for line in f:
                line=line.strip()
                if not line: continue
                content = line.split(":",1)[1].strip() if ":" in line else line
                out.append(normalize_text(content))
        return out

    def process_prompts(self, driver, prompts):
        sent = []
        for p in prompts:
            if p in sent: 
                self.log(f"ℹ️ Bỏ qua prompt đã gửi: {p}")
                continue
            # nhập prompt
            tb = WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.ID, "video-create-textarea")))
            tb.click(); tb.send_keys(Keys.CONTROL, "a", Keys.BACKSPACE)
            tb.send_keys(p, Keys.ENTER)
            self.log(f"✍️ Nhập prompt: {p}")
            # click Create
            btn = WebDriverWait(driver, 10).until(EC.element_to_be_clickable((
                By.XPATH,
                "//button[.//img[@alt='AI Video create png by Hailuo AI Video Generator']]"
            )))
            btn.click()
            self.log("▶️ Click Create")
            # kiểm tra queue
            try:
                que = WebDriverWait(driver, 10).until(
                    EC.visibility_of_element_located((By.CSS_SELECTOR,"div.adm-auto-center-content"))
                )
                if "full" in que.text.lower():
                    self.log("⏳ Queue đầy, sẽ thử lại sau 3s")
                    time.sleep(3)
                    continue
            except TimeoutException:
                pass
            sent.append(p)
            self.log(f"✅ Prompt chấp nhận: {p}")
        return sent

    def download_all(self, driver, prompts):
        # đơn giản: load mine page, scroll, download
        driver.get("https://hailuoai.video/mine")
        time.sleep(5)
        files_before = set(os.listdir(self.video_folder))
        for _ in range(3):
            driver.execute_script("window.scrollTo(0,document.body.scrollHeight);")
            time.sleep(2)
        items = driver.find_elements(By.XPATH,"//div[contains(@class,'mine-video-item')]")
        for it in items:
            try:
                ActionChains(driver).move_to_element(it).perform()
                p = normalize_text(it.find_element(By.CSS_SELECTOR,".prompt-plain-span").text)
                if p in prompts:
                    btns = it.find_elements(By.TAG_NAME,"button")
                    for b in btns:
                        if "Download" in b.text or b.get_attribute("innerHTML").lower().find("recreate")<0:
                            b.click()
                            self.log(f"⬇️ Download prompt: {p}")
                            break
            except:
                pass
        # chờ download
        t0 = time.time()
        while time.time()-t0 < self.wait_time:
            if any(f.endswith(".crdownload") for f in os.listdir(self.video_folder)):
                time.sleep(1)
            else:
                break
        # đổi tên theo thứ tự
        for idx,p in enumerate(prompts,1):
            matches = [f for f in os.listdir(self.video_folder) if f.endswith(".mp4")]
            if matches:
                src = max(matches, key=lambda f:os.path.getctime(os.path.join(self.video_folder,f)))
                dst = os.path.join(self.video_folder, f"Prompt {idx}.mp4")
                os.rename(os.path.join(self.video_folder,src), dst)
                self.log(f"🔄 Đổi tên {src} → Prompt {idx}.mp4")

    def run(self):
        driver = self.setup_driver()
        driver.get(TARGET_URL)
        time.sleep(5)

        # cookies / login
        if os.path.exists(COOKIES_FILE):
            self.load_cookies(driver)
            driver.refresh()
            time.sleep(5)
        else:
            self.log("❗ Vui lòng login thủ công để lưu cookies rồi deploy lại.")
            driver.quit()
            return

        # chọn mode
        if self.mode=="subject":
            self.click_tab(driver,
                           "//*[contains(text(),'Subject Reference')]",
                           "✅ Chuyển Subject Reference",
                           "❌ Không tìm Subject Reference")
            time.sleep(2)
            self.click_confirm(driver); time.sleep(2)
            self.upload_images_silently(driver)
        else:
            self.click_tab(driver,
                           "//div[span[text()='Text to Video']]",
                           "✅ Chuyển Text to Video",
                           "❌ Không tìm Text to Video")
            time.sleep(2)
            self.click_confirm(driver)

        scenes = self.load_scenes()
        self.log(f"📋 Prompts: {scenes}")
        sent = self.process_prompts(driver, scenes)
        self.log(f"⏳ Chờ {self.wait_time}s để xử lý video...")
        time.sleep(self.wait_time)
        self.download_all(driver, sent)

        driver.quit()
        self.log("🏁 Automation hoàn tất.")