OpenAI 以文生图 python代码

流光追不及

已于 2023-10-19 18:07:58 修改

阅读量314

点赞数

文章标签： python 开发语言

于 2023-10-19 18:06:54 首次发布

本文链接：https://blog.csdn.net/zwdcdut/article/details/133932024

版权

def getImage(prompt,n):
    # 设置代理服务器地址和端口号
    proxies = {
        "https": "http://127.0.0.1:7890",
        "http": "http://127.0.0.1:7890"
    }
    # 设置API密钥和API请求
    # API_KEY = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"

    url = "https://api.openai.com/v1/images/generations"

    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {API_KEY}"
    }
    data = {
        "prompt": prompt,
        "n": n,
        "size": "1024x1024"
    }

    # 发送API请求，通过代理服务器发送
    response = requests.post(url, headers=headers, json=data, proxies=proxies)

    # 解析API响应
    res=[]
    if response.status_code == 200:
        result = response.json()
        res=result['data']
    else:
        print("Error:", response.status_code, response.text)
    #[{'url':},{'url':}]
    return res