Update app.py

app.py

@@ -35,6 +35,42 @@ from face_utils import FaceAnalysis2, analyze_faces
 from style_template import styles
 from aspect_ratio_template import aspect_ratios
 
+import os
+from datasets import load_dataset
+import uuid
+
+# 定义保存路径
+save_path = "./examples/xiangxiang_man"
+
+# 清空目标路径（如果存在）
+if os.path.exists(save_path):
+    for file_name in os.listdir(save_path):
+        file_path = os.path.join(save_path, file_name)
+        if os.path.isfile(file_path):
+            os.remove(file_path)
+    print(f"Cleared existing files in {save_path}")
+else:
+    os.makedirs(save_path, exist_ok=True)
+    print(f"Created directory: {save_path}")
+
+# 加载数据集
+dataset = load_dataset("svjack/Prince_Xiang_iclight_v2")
+
+# 遍历数据集并保存图片
+for example in dataset["train"]:
+    # 获取图片数据
+    image = example["image"]
+    
+    # 生成唯一的文件名（使用 uuid）
+    file_name = f"{uuid.uuid4()}.png"
+    file_path = os.path.join(save_path, file_name)
+    
+    # 保存图片
+    image.save(file_path)
+    print(f"Saved {file_path}")
+
+print("All images have been saved.")
+
 # global variable
 base_model_path = 'SG161222/RealVisXL_V4.0'
 face_detector = FaceAnalysis2(providers=['CPUExecutionProvider', 'CUDAExecutionProvider'], allowed_modules=['detection', 'recognition'])
@@ -224,6 +260,12 @@ def get_image_path_list(folder_name):
 
 def get_example():
     case = [
+        [
+            get_image_path_list('./examples/xiangxiang_man'),
+            "detailed photo of a handsome man img. neon light, city. cars and people in l the background",
+            "(No style)",
+            "(asymmetry, worst quality, low quality, illustration, 3d, 2d, painting, cartoons, sketch), open mouth",
+        ],
         [
             get_image_path_list('./examples/scarletthead_woman'),
             "instagram photo, portrait photo of a woman img, colorful, perfect face, natural skin, hard shadows, film grain",