import base64
import io
from PIL import Image

# 假設(shè)有一個(gè)Base64編碼的圖片字符串，這里我們使用一個(gè)簡(jiǎn)化的示例
base64_str = "data:image/jpeg;base64,/9j/4AAQSkZJRgABAQEAAAAAAAD/4gIoSUNDX1BST0ZJTEUAAQEAAAIYAAAAAAIQAABtbnRyUkdCIFhZJiouMjY7...（此處省略其他Base64編碼的內(nèi)容）"

# 去掉前綴（'data:image/jpeg;base64,'）
base64_data = base64_str.split(',')[1]

# 解碼Base64數(shù)據(jù)
image_bytes = base64.b64decode(base64_data)

# 將字節(jié)數(shù)據(jù)轉(zhuǎn)換為BytesIO對(duì)象
image_io = io.BytesIO(image_bytes)

# 使用PIL庫(kù)讀取圖片
image = Image.open(image_io)

# 保存圖片到文件
image.save('output.jpg')

# 獲取base64圖片數(shù)據(jù)
import requests

def query(payload, token='你自己的token令牌', appCode='你的appCode'):
    API_URL = f"https://{appCode}.aistudio-hub.baidu.com/image/generations"
    headers = {
        # 請(qǐng)前往 https://aistudio.baidu.com/index/accessToken 查看 訪問(wèn)令牌
        "Authorization": f"token {token}",
        "Content-Type": "application/json"
    }
    response = requests.post(API_URL, headers=headers, json=payload)
    return response.json()

def access_image(prompt):
    output = query({
        "prompt": f"{prompt}"
    })
    return output['data'][0]["b64_image"]

# base64數(shù)據(jù)轉(zhuǎn)圖片
import base64  
import io  
from PIL import Image  


def decode_b64_image(base64_data):
    # 解碼Base64數(shù)據(jù)  
    image_bytes = base64.b64decode(base64_data)  

    # 將字節(jié)數(shù)據(jù)轉(zhuǎn)換為BytesIO對(duì)象  
    image_io = io.BytesIO(image_bytes)  
    
    # 使用PIL庫(kù)讀取圖片  
    image = Image.open(image_io)  
    
    # 保存圖片到文件  
    # image.save('output.jpg')
    return image

# 調(diào)用函數(shù)
prompt = "杰作，高品質(zhì)，超精細(xì)，全細(xì)節(jié)，8k"
base64_image = access_image(prompt)
decode_b64_image(base64_image)