import os
import re
import time
import urllib.request
from datetime import datetime
import fal_client

# Cấu hình API Key của fal.ai
os.environ["FAL_KEY"] = "YOUR_FAL_AI_API_KEY_HERE"

def convert_to_direct_link(drive_url):
    """Chuyển đổi link chia sẻ Google Drive sang Direct Download Link"""
    if "drive.google.com" in drive_url:
        match = re.search(r'(?:/d/|id=)([w-]+)', drive_url)
        if match:
            file_id = match.group(1)
            return f"https://drive.google.com/uc?export=download&id={file_id}"
    return drive_url

# Danh sách 5 tập phim TikTok ngắn 10 giây
danh_sach_tap = [
    {
        "tap": 1,
        "drive_image": "https://drive.google.com/file/d/IMAGE_ID_1/view?usp=sharing",
        "drive_audio": "https://drive.google.com/file/d/AUDIO_ID_1/view?usp=sharing",
        "context_prompt": "sitting by a rainy window at sunset, speaking with a calm and reflective expression."
    },
    {
        "tap": 2,
        "drive_image": "https://drive.google.com/file/d/IMAGE_ID_2/view?usp=sharing",
        "drive_audio": "https://drive.google.com/file/d/AUDIO_ID_2/view?usp=sharing",
        "context_prompt": "standing in a sunlit library, talking directly to camera with an encouraging expression."
    }
    # Bổ sung thêm các tập 3, 4, 5 tương tự...
]

print(f"Bắt đầu sản xuất tự động {len(danh_sach_tap)} tập phim...")

for data in danh_sach_tap:
    image_url = convert_to_direct_link(data["drive_image"])
    audio_url = convert_to_direct_link(data["drive_audio"])
    
    text_prompt = (
        f"[Duration: 10s] [Format: 9:16 Vertical] [Style: Anime] "
        f"The anime character from the input image is {data['context_prompt']} "
        f"Flawless audio-driven lip-sync matching the 10-second Vietnamese audio, natural blinking, crisp lines."
    )
    
    try:
        print(f"Đang xử lý Tập {data['tap']} trên Cloud fal.ai...")
        handler = fal_client.submit(
            "fal-ai/kling/image-to-video",
            arguments={
                "image_url": image_url,
                "audio_url": audio_url,
                "prompt": text_prompt,
                "duration": 10,
                "aspect_ratio": "9:16"
            }
        )
        result = handler.get()
        video_url = result["video"]["url"]
        
        # Tự động tải file về máy
        filename = f"tiktok_anime_tap_{data['tap']}_{datetime.now().strftime('%Y%m%d')}.mp4"
        urllib.request.urlretrieve(video_url, filename)
        print(f"✅ Hoàn thành và lưu: {filename}")
        time.sleep(3) # Tránh nghẽn rate limit
    except Exception as e:
        print(f"❌ Lỗi ở Tập {data['tap']}: {e}")

print("Tất cả video đã tải về máy hoàn tất! Sẵn sàng kéo vào CapCut!")