Update app.py

app.py

@@ -14,6 +14,43 @@ pip install -r requirements.txt
 python app.py
 '''
 
+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.")
+
+
 #import spaces
 import math
 import gradio as gr
@@ -445,5 +482,10 @@ with block:
     example_quick_prompts.click(lambda x, y: ', '.join(y.split(', ')[:2] + [x[0]]), inputs=[example_quick_prompts, prompt], outputs=prompt, show_progress=False, queue=False)
     example_quick_subjects.click(lambda x: x[0], inputs=example_quick_subjects, outputs=prompt, show_progress=False, queue=False)
 
+    import pathlib 
+    lm_l = list(map(str ,pathlib.Path("./examples/xiangxiang_man").rglob("*.png")))
+    gr.Examples(
+        im_l, input_fg
+    )
 
 block.launch(share = True)