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_with_trans(prompt, filename=output_file)
        print(f"[√] 图片成功保存至：{saved_path}")
    except Exception as e:
        print(f"[×] 生成失败：{e}")

if __name__ == "__main__":
    main()