Update produce_gif_script.py

produce_gif_script.py

@@ -49,6 +49,52 @@ pd.DataFrame(list(product(*[reference_video_list, prompt_list])))[[1, 0]].rename
 
 !python produce_gif_script.py xiangling_video_seed.csv "svjack/GenshinImpact_XL_Base" xiangling_gif_dir \
  --num_frames 16 --temp_folder temp_frames --seed 0 --controlnet_conditioning_scale 0.3
+
+import re
+def insert_content_in_string(insert_content, character_name, gender=None):
+    """
+    在原始字符串中特定位置插入内容。
+    :param insert_content: 要插入的内容
+    :param character_name: 角色名称
+    :param gender: 性别（可选，可以是 "1boy" 或 "1girl"）
+    :return: 修改后的字符串
+    """
+    # 根据 character_name 和 gender 生成 original_string
+    original_string = f"solo,{character_name}\(genshin impact\),{gender if gender else '1boy'},highres,"
+    original_string = f"{gender if gender else '1boy'},{character_name}, masterpiece, white lab coat, red tie"
+    # 根据 character_name 生成 target_pattern
+    target_pattern = re.escape(character_name)
+    # 插入内容
+    modified_string = re.sub(target_pattern, r'\g<0>' + insert_content, original_string)
+    return original_string ,modified_string
+    
+from datasets import load_dataset
+character_name = "Makise Kurisu"
+gender = "1girl"  # 可选参数
+prompt_list = load_dataset("svjack/daily-actions-en-zh")["train"].to_pandas()["en"].map(
+    lambda x: ", {}".format(x)
+).map(
+    lambda insert_content: insert_content_in_string(insert_content, character_name, gender)[-1]
+).dropna().drop_duplicates().values.tolist()
+print(len(prompt_list))
+
+import pandas as pd
+import pathlib
+reference_video_list = pd.Series(
+list(pathlib.Path("video-dataset-Lily-Bikini-rm-background-organized").rglob("*.mp4"))
+).map(str).values.tolist()
+print(len(reference_video_list))
+
+from itertools import product
+pd.DataFrame(list(product(*[reference_video_list, prompt_list])))[[1, 0]].rename(
+    columns = {
+        1: "prompt",
+        0: "input_video"
+    }
+).to_csv("Makise_Kurisu_video_seed.csv", index = False)
+
+!python produce_gif_script.py Makise_Kurisu_video_seed.csv "cagliostrolab/animagine-xl-3.1" Makise_Kurisu_gif_dir \
+ --num_frames 16 --temp_folder temp_frames --seed 0 --controlnet_conditioning_scale 0.3
 '''
 
 import sys