# Hàm kiểm tra trạng thái task
def check_task_status(task_id):
    try:
        status_url = f"https://api.vidu.com/ent/v2/tasks/{task_id}/creations"
        headers = {
            "Authorization": f"Token {VIDU_API_KEY}",
            "Content-Type": "application/json"
        }
        response = requests.get(status_url, headers=headers)
        if not response.ok:
            error_data = response.json()
            raise Exception(error_data.get("message", "Lỗi khi kiểm tra trạng thái"))
        
        response_data = response.json()
        state = response_data.get("state")
        video_url = None
        if state == "success" and response_data.get("creations"):
            video_url = response_data["creations"][0].get("url")
        elif state == "failed":
            err_code = response_data.get("err_code", "Không có mã lỗi")
            logging.error(f"Task {task_id} thất bại - Error Code: {err_code}")
            raise Exception(f"Task thất bại - Error Code: {err_code}")
        
        logging.info(f"Task {task_id} - State: {state}")
        return state, video_url
    except Exception as e:
        logging.error(f"Lỗi khi kiểm tra trạng thái task {task_id}: {str(e)}")
        raise

# Hàm xử lý tạo video và theo dõi trạng thái
def generate_and_monitor_video(images, prompt, duration, seed, aspect_ratio, resolution, movement_amplitude):
    try:
        # Validate inputs (giữ nguyên)
        if not images or len(images) < 1 or len(images) > 3:
            raise ValueError("Vui lòng chọn từ 1 đến 3 file ảnh")
        if not prompt or len(prompt) > 1500:
            raise ValueError("Mô tả nội dung phải có tối đa 1500 ký tự")
        if duration != 4:
            raise ValueError("Thời lượng phải là 4 giây")
        if resolution not in ["360p", "720p"]:
            raise ValueError("Độ phân giải phải là 360p hoặc 720p")

        # Upload các ảnh (giữ nguyên)
        image_urls = []
        for image in images:
            validate_image(image)
            image_url = upload_image(image)
            image_urls.append(image_url)

        # Gửi yêu cầu tạo video
        payload = {
            "model": "vidu2.0",
            "images": image_urls,
            "prompt": prompt,
            "duration": str(duration),
            "seed": str(seed),
            "aspect_ratio": aspect_ratio,
            "resolution": resolution,
            "movement_amplitude": movement_amplitude
        }
        api_url = "https://api.vidu.com/ent/v2/reference2video"
        headers = {
            "Authorization": f"Token {VIDU_API_KEY}",
            "Content-Type": "application/json"
        }
        response = requests.post(api_url, headers=headers, json=payload)
        if not response.ok:
            error_data = response.json()
            raise Exception(error_data.get("message", "Lỗi khi tạo video"))

        response_data = response.json()
        task_id = response_data.get("task_id")
        logging.info(f"Task created - Task ID: {task_id}")

        # Theo dõi trạng thái
        for _ in range(30):  # 5 phút
            state, video_url = check_task_status(task_id)
            yield f"Task ID: {task_id}\nTrạng thái: {state}", None
            if state == "success" and video_url:
                return f"Video hoàn thành!\nTask ID: {task_id}\nURL: {video_url}", video_url
            elif state == "failed":
                raise Exception(f"Task thất bại - Task ID: {task_id}")
            time.sleep(10)
        
        raise Exception(f"Timeout: Task {task_id} chưa hoàn thành sau 5 phút")

    except Exception as e:
        logging.error(f"Lỗi trong quá trình tạo video: {str(e)}")
        yield f"Lỗi: {str(e)}", None
        return None, None