30 lines
1.5 KiB
Python
30 lines
1.5 KiB
Python
from zhipuai import ZhipuAI
|
||
from image_downloader import download_image
|
||
import os
|
||
client = ZhipuAI(api_key="8a795f242987fb1388b61f7c1e104b0d.9goBSAn25glQZGLP") # 请填写您自己的APIKey
|
||
response = client.images.generations(
|
||
model="cogview-3", #填写需要调用的模型名称
|
||
# prompt=" 一个城市在水晶瓶中欢快生活的场景，水彩画风格，展现出微观与珠宝般的美丽。")
|
||
# prompt="一个中国女性，35岁左右，长发，身材纤细，细腰，梨形身材，身材曲线有致,,穿白色薄纱连衣短裙和肉色连裤袜，躺姿，卧于床上，真实摄影效果，富士XT30相机56mm长焦")
|
||
prompt="""
|
||
1. 人物形象：
|
||
- 两人之间存在一种秘密的亲密关系。
|
||
- 他们可能希望避免引起外界过多的注意。
|
||
|
||
2. 职业和场景：
|
||
- 提到“在课堂上”，暗示至少一方可能是学生或教育工作者。
|
||
- 可以构建的场景可能是一个学习环境，如学校或图书馆。
|
||
|
||
3. 提词构建：
|
||
- “两人坐在安静的角落，低声讨论着只有他们俩才知道的秘密计划。”
|
||
- “阳光透过图书馆的窗户，洒在认真阅读的两个人身上，他们偶尔交换一个意味深长的眼神。”"""
|
||
)
|
||
print(response.data[0].url)
|
||
filename = "avatar6.png"
|
||
# check if file exist, if so, make name surfix num + 1, for example avatar6.png exist, then avatar7.png
|
||
if os.path.exists(filename):
|
||
i = 1
|
||
while os.path.exists(filename):
|
||
filename = "avatar" + str(i) + ".png"
|
||
i += 1
|
||
download_image(response.data[0].url, filename) |