1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
| from auto_media_publisher.utils.utils_text2image_pollinations import Text2ImageGenerator
|
| def main():
| # 实例化工具类(可选修改默认参数)
| generator = Text2ImageGenerator()
|
| # 待生成的中文提示词(必填)
| prompt = "一只小狗在沙滩上晒太阳"
|
| # 可选文件名(不提供会自动用当前时间命名)
| output_file = "dog_on_beach.jpg"
|
| # 生成并保存图像
| try:
| saved_path = generator.generate_image(prompt, filename=output_file)
| print(f"[√] 图片成功保存至:{saved_path}")
| except Exception as e:
| print(f"[×] 生成失败:{e}")
|
| if __name__ == "__main__":
| main()
|
|