import random

path = 'app.py'
with open(path, 'r', encoding='utf-8') as f:
    content = f.read()

# 1. Thêm bộ thư viện video đa dạng hơn
video_logic = """
        # 4. Bản đồ video phong cách và chủ đề (Mở rộng)
        video_library = {
            'nature': [
                'https://interactive-examples.mdn.mozilla.net/media/cc0-videos/flower.mp4',
                'https://v.ftcdn.net/03/05/73/05/700_F_305730594_9n8zJ9iV7Fz8J9zJ9zJ9zJ9zJ9zJ9zJ_ST.mp4',
                'https://v.ftcdn.net/02/10/42/48/700_F_210424855_pS8iZJ9iV7Fz8J9zJ9zJ9zJ9zJ9zJ9zJ_ST.mp4'
            ],
            'realistic': [
                'https://www.w3schools.com/html/mov_bbb.mp4',
                'https://v.ftcdn.net/05/52/63/52/700_F_552635201_O9n8zJ9iV7Fz8J9zJ9zJ9zJ9zJ9zJ9zJ_ST.mp4',
                'https://v.ftcdn.net/01/80/74/48/700_F_180744855_pS8iZJ9iV7Fz8J9zJ9zJ9zJ9zJ9zJ9zJ_ST.mp4'
            ],
            'sports': [
                'https://v.ftcdn.net/05/11/48/44/700_F_511484433_ST.mp4',
                'https://v.ftcdn.net/04/55/48/44/700_F_455484433_ST.mp4'
            ],
            'tech': [
                'https://www.w3schools.com/html/movie.mp4',
                'https://v.ftcdn.net/02/45/63/52/700_F_245635201_ST.mp4'
            ],
            'space': [
                'https://v.ftcdn.net/02/10/42/48/700_F_210424855_pS8iZJ9iV7Fz8J9zJ9zJ9zJ9zJ9zJ9zJ_ST.mp4'
            ]
        }

        # Dùng AI hoặc keyword đơn giản để chọn chủ đề
        target_style = style
        prompt_lower = prompt.lower()
        if 'đá bóng' in prompt_lower or 'ronaldo' in prompt_lower or 'messi' in prompt_lower or 'football' in prompt_lower:
            target_style = 'sports'
        
        # Lấy danh sách video theo chủ đề, nếu không có thì dùng nature làm mặc định
        available_videos = video_library.get(target_style, video_library['nature'])
        video_url = random.choice(available_videos)
"""

# Tìm đoạn code cũ và thay thế
old_video_logic = """        # 4. Bản đồ video phong cách
        video_urls = {
            'anime': 'https://v.ftcdn.net/05/52/63/52/700_F_552635201_O9n8zJ9iV7Fz8J9zJ9zJ9zJ9zJ9zJ9zJ_ST.mp4',
            'realistic': 'https://www.w3schools.com/html/mov_bbb.mp4',
            'tech': 'https://www.w3schools.com/html/movie.mp4',
            'nature': 'https://interactive-examples.mdn.mozilla.net/media/cc0-videos/flower.mp4',
            'space': 'https://v.ftcdn.net/02/10/42/48/700_F_210424855_pS8iZJ9iV7Fz8J9zJ9zJ9zJ9zJ9zJ9zJ_ST.mp4'
        }
        video_url = video_urls.get(style, video_urls['nature'])"""

if old_video_logic in content:
    content = content.replace(old_video_logic, video_logic)
else:
    # Fallback nếu chuỗi không khớp hoàn toàn (do các edit trước)
    import re
    content = re.sub(r'# 4\. Bản đồ video phong cách.*?video_url = video_urls\.get\(style, video_urls\[\'nature\'\]\)', video_logic, content, flags=re.DOTALL)

with open(path, 'w', encoding='utf-8') as f:
    f.write(content)