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.")