import json
from pathlib import Path

prompts_path = Path("02_AI_Prompts/muse_ai_video_prompts.json")

with open(prompts_path, "r", encoding="utf-8") as f:
    data = json.load(f)

motion_prompts = data.setdefault("motion_prompts", {})

new_shots = {
    "ep01_scene07_shot01": {
        "scene_title": "EP01 Scene 07 Shot 01: Thư Phòng Kim Trọng - Đêm Trằn Trọc Tương Tư",
        "mode": "Image-to-Video",
        "character_asset_ref": "04_Assets/characters/01_Main_Protagonists/kim_trong_18yo_720p.png",
        "reference_start_frame": "Master asset Kim Trọng 18 tuổi khôi ngô tuấn tú trong thư phòng đêm.",
        "motion_prompt": "Tao video 10 giay dien anh lang man: Can canh Kim Trong 18 tuoi guong mat nho nha tuan tu ngoi tram tu ben ban thu phong go lim trong dem. Anh nen lung linh hat bong chang len buc tuong giay do. Tay chang cam but long ngung tren to giay hoa tien, muc chua ha. Chang khe tho dai nhe, anh mat chan chua noi niem tuong tu doi mat 'lan thu thuy' cua nang Kieu chieu hom truoc. Gio dem thoang qua lam ngon nen chao nghieng nhe. 24fps dien anh 16:9, tuyet doi khong chen chu. Âm thanh: Tiếng thở dài nhẹ tương tư của Kim Trọng, tiếng ngọn nến bập bùng khe khẽ, tiếng gió đêm hiu hắt lay rèm trúc ngoài hiên. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rác.",
        "camera_motion": "Slow subtle push-in toward thoughtful face and poised calligraphy brush",
        "duration_sec": 10,
        "audio_prompt": "Âm thanh: Tiếng thở dài nhẹ tương tư của Kim Trọng, tiếng ngọn nến bập bùng khe khẽ, tiếng gió đêm hiu hắt lay rèm trúc ngoài hiên. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rác."
    },
    "ep01_scene07_shot02": {
        "scene_title": "EP01 Scene 07 Shot 02: Ba Thu Dọn Lại Một Ngày - Độc Thoại Bên Khung Cửa Sổ",
        "mode": "Image-to-Video",
        "reference_start_frame": "Frame cuối của ep01_scene07_shot01: Kim Trọng ngồi trầm tư bên bàn trà.",
        "motion_prompt": "Tao video 10 giay tiep noi muot ma tu khung hinh dinh kem: Kim Trong buong cay but long xuong nghien muc da, cham rai dung day buoc lai ben khung cua so go cham long huong ve phia vuon hoa nha ho Vuong. Chang tu tua vao khung cua, doi mat ngong nhin ve phia buc tuong hoa ngap tran anh trang vang vat, mieng lam ram thi tham: 'Sau dong cang lac cang day / Ba thu don lai mot ngay dai ghe...'. Gio dem lam ta ao gam lam ngoc khe lay dong. 24fps dien anh 16:9, tuyet doi khong chen chu. Âm thanh: Giọng Kim Trọng thì thầm ngâm nga trầm ấm tha thiết: 'Sầu đong càng lắc càng đầy, Ba thu dọn lại một ngày dài ghê...', tiếng gió đêm lướt qua hàng trúc xào xạc. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rác.",
        "camera_motion": "Smooth tracking shot as Kim rises and walks to the moonlit lattice window",
        "duration_sec": 10,
        "audio_prompt": "Âm thanh: Giọng Kim Trọng thì thầm ngâm nga trầm ấm tha thiết: 'Sầu đong càng lắc càng đầy, Ba thu dọn lại một ngày dài ghê...', tiếng gió đêm lướt qua hàng trúc xào xạc. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rác."
    },
    "ep01_scene07_shot03": {
        "scene_title": "EP01 Scene 07 Shot 03: Ánh Trăng Soi Bức Tường Hoa Vườn Thúy Đêm Hè",
        "mode": "Image-to-Video",
        "reference_start_frame": "Frame cuối của ep01_scene07_shot02: Kim Trọng nhìn qua khung cửa sổ.",
        "motion_prompt": "Tao video 10 giay dien anh: May quay tu cua so phong Kim Trong tu tu pan va dolly forward huong ra khong gian dem vuon Thuy. Buc tuong hoa ngan cach hai nha ruc ro duoi anh trang thuong tuan huyen ao. Canh hoa dao vuon sang vuon ben khe rung rinh trong gio dem. Khong gian tinh mich, dem he ngap tran huong hoa da ly thoang thoang, tao nen cam giac khac khoai truoc ngay hoi ngo. 24fps dien anh 16:9, tuyet doi khong chen chu. Âm thanh: Tiếng côn trùng rả rích đêm hè từ xa, tiếng gió nhẹ thoảng qua cành đào lay động xào xạc, tiếng chuông gió xa xăm buông tiếng 'leng keng' thanh khiết. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rè.",
        "camera_motion": "Slow forward dolly through open window toward the moonlit garden wall and peach blossom branches",
        "duration_sec": 10,
        "audio_prompt": "Âm thanh: Tiếng côn trùng rả rích đêm hè từ xa, tiếng gió nhẹ thoảng qua cành đào lay động xào xạc, tiếng chuông gió xa xăm buông tiếng 'leng keng' thanh khiết. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rè."
    },
    "ep01_scene08_shot01": {
        "scene_title": "EP01 Scene 08 Shot 01: Cành Đào Vắt Chiếc Thoa Kim & Gió Hè Cuốn Bay",
        "mode": "Image-to-Video",
        "character_asset_ref": "04_Assets/characters/01_Main_Protagonists/thuy_kieu_maiden_16yo_720p.png",
        "reference_start_frame": "Master asset Thúy Kiều 16 tuổi bên cành hoa đào trưa hè.",
        "motion_prompt": "Tao video 10 giay dien anh co phong: Nang he ruc ro chieu qua buc tuong hoa co kinh. Mot con gio he bat ngo thoi bay chiec thoa kim tren mai toc Thuy Kieu, chiec thoa vang lap lanh roi vuong vat ngang tren canh hoa dao vuon sang vuon nha ben. Canh hoa dao rung roi la ta theo lan gio. May quay macro tracking nhe theo chiec thoa kim lap lanh giua nhung canh hoa hong ruc. 24fps dien anh 16:9, tuyet doi khong chen chu. Âm thanh: Tiếng gió hè thổi xào xạc cành lá, tiếng 'leng keng' thanh mảnh khi thoa vàng chạm vào cành gỗ, tiếng chim hót buổi trưa. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rác.",
        "camera_motion": "Slow macro glide tracking the golden hairpin caught on flowering peach branch",
        "duration_sec": 10,
        "audio_prompt": "Âm thanh: Tiếng gió hè thổi xào xạc cành lá, tiếng 'leng keng' thanh mảnh khi thoa vàng chạm vào cành gỗ, tiếng chim hót buổi trưa. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rác."
    },
    "ep01_scene08_shot02": {
        "scene_title": "EP01 Scene 08 Shot 02: Kim Trọng Nhặt Thoa Kim & Lời Ước Bên Tường Hoa",
        "mode": "Image-to-Video",
        "character_asset_ref": "04_Assets/characters/01_Main_Protagonists/kim_trong_18yo_720p.png",
        "reference_start_frame": "Frame cuối của ep01_scene08_shot01: Cành đào vắt chiếc thoa kim.",
        "motion_prompt": "Tao video 10 giay: Kim Trong buoc ra vuon hoa nhin thay chiec thoa kim lap lanh tren canh dao. Chang mung ro buoc lai, hai tay run run don lay ky vat quy gia nang niu truoc nguc. Dung luc ay tu ben kia buc tuong hoa vang len giong noi e ap trong treo cua Kieu: 'Dam hoi ben ay co ai trong thay chiec thoa kim chang?'. Mat Kim Trong bung sang hanh phuc. 24fps dien anh 16:9, tuyet doi khong chen chu. Âm thanh: Tiếng bước chân nhẹ trên lối sỏi, giọng Thúy Kiều e ấp vọng qua bờ tường: 'Dám hỏi bên ấy có ai trông thấy chiếc thoa kim chăng?', tiếng Kim Trọng thở phào vui sướng. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rác.",
        "camera_motion": "Medium tracking shot as Kim approaches peach tree and reaches up reverently",
        "duration_sec": 10,
        "audio_prompt": "Âm thanh: Tiếng bước chân nhẹ trên lối sỏi, giọng Thúy Kiều e ấp vọng qua bờ tường: 'Dám hỏi bên ấy có ai trông thấy chiếc thoa kim chăng?', tiếng Kim Trọng thở phào vui sướng. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rác."
    },
    "ep01_scene08_shot03": {
        "scene_title": "EP01 Scene 08 Shot 03: Trao Thoa Bên Tường Hoa - Chạm Tay Bàng Hoàng",
        "mode": "Image-to-Video",
        "reference_start_frame": "Frame cuối của ep01_scene08_shot02: Kim Trọng nâng chiếc thoa kim.",
        "motion_prompt": "Tao video 10 giay: Thuy Kieu khe he guong mat thanh tu tren bo tuong hoa ruc ro anh nang. Kim Trong ngap tran say dam voi va buoc len trao lai chiec thoa vang. Khoanh khac hai ngon tay vo tinh cham nhe vao nhau, ca hai deu giat minh e le, anh mat nhin nhau boi roi xao xuyen giua troi hoa he. 24fps dien anh 16:9, tuyet doi khong chen chu. Âm thanh: Tiếng thì thầm run rẩy của Kim Trọng: 'Tiểu sinh may mắn nhặt được...', tiếng thở nhẹ của Kiều, tiếng ve râm ran đầu cành báo hiệu mùa hè. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rác.",
        "camera_motion": "Tight two-shot profile across the flowering wall with rack focus on touching fingertips",
        "duration_sec": 10,
        "audio_prompt": "Âm thanh: Tiếng thì thầm run rẩy của Kim Trọng: 'Tiểu sinh may mắn nhặt được...', tiếng thở nhẹ của Kiều, tiếng ve râm ran đầu cành báo hiệu mùa hè. Tuyệt đối KHÔNG sinh nhạc nền, không tạp âm rác."
    }
}

for k, v in new_shots.items():
    motion_prompts[k] = v

with open(prompts_path, "w", encoding="utf-8") as f:
    json.dump(data, f, indent=2, ensure_ascii=False)

print(f"[✓] Đã cập nhật thành công {len(new_shots)} shots cho Scene 07 và Scene 08 vào {prompts_path}!")