python+tkinter实现GUI界面调用即梦AI文生图片API接口_即梦ai的api调用
背景
目前字节跳动公司提供了即梦AI的接口免费试用，但是并发量只有1，不过足够我们使用了。我这里想做个使用python+tkinter实现的GUI可视化界面客户端，这样就不用每次都登录官方网站去进行文生图片，当然文生视频，或者图生视频也是如此的逻辑。
实现思路
1. 编写一个窗口
2. 窗口放置一个文本输入框，一个按钮和一个消息对话框就可以了
最后界面如下：
3. 给按钮绑定事件函数，当用户输入文字提示词后，点击按钮则生成对应的图片
4. 获取生成的图片链接地址，然后通过打开图片将图片写入对话框中
Tkinter的python实现代码
import timeimport tkinterfrom tkinter import scrolledtext, END, RAISEDfrom io import BytesIOimport requestsfrom tttt.ai_chat import AI_chatfrom PIL import Image, ImageTkfrom tttt.play_voice import play_audiofrom tttt.record_voice import record_audiofrom tttt.streaming_asr_demo import test_onefrom tttt.voice_generate import txt_to_voicefrom tttt.即梦AI文生图片 import txt_to_image# 创建根窗口root = tkinter.Tk()# 设置属性root.title(\'即梦AI文生图片python客户端\')# 设置窗口大小root.geometry(\"650x560\")# 文本历史记录txt_history = scrolledtext.ScrolledText(root, width=80, height=30)txt_history.tag_config(\'green\', foreground=\'#008B00\')txt_history.tag_config(\'red\', foreground=\'#FF0000\')txt_history.grid(column=0, row=0, columnspan=3, padx=10, pady=10)# 标签lbl = tkinter.Label(root, text=\"请输入文字:\")lbl.grid(column=0, row=1, padx=5, pady=5)# 文本输入框txt_input = tkinter.Entry(root, width=50)txt_input.grid(column=1, row=1, padx=5, pady=5)# 按钮样式button_style = { \"font\": (\"Arial\", 10, \"bold\"), # 字体样式 \"bg\": \"#4CAF50\", # 背景颜色 \"fg\": \"white\", # 文字颜色 \"activebackground\": \"#45a049\", # 按钮按下时的背景颜色 \"activeforeground\": \"white\", # 按钮按下时的文字颜色 \"relief\": RAISED, # 按钮边框样式 \"bd\": 2, # 按钮边框宽度 \"padx\": 5, # 按钮内部水平方向的填充 \"pady\": 5 # 按钮内部垂直方向的填充}# 从网页下载图片并加载def download_and_display_image(url): try: response = requests.get(url) response.raise_for_status() # 确保请求成功 image_data = BytesIO(response.content) image = Image.open(image_data) photo = ImageTk.PhotoImage(image) # 在历史消息对话框中插入图片 txt_history.image = photo # 保持对PhotoImage的引用，防止被垃圾回收 txt_history.insert(END, \"\\n\") txt_history.image_create(END, image=photo) txt_history.insert(END, \"\\n\") except requests.RequestException as e: txt_history.insert(END, f\"无法下载图片: {e}\\n\")# 发送文生图片def txt_generate_image(): # 获取输入框文本内容 user_message = txt_input.get() # 将用户输入的内容插入对话框中 txt_history.insert(END, \'即梦AI：\' + time.strftime(\"%Y-%m-%d %H:%M:%S\", time.localtime()) + \'\\n\', \'purple\') # 将输入的内容添加到文本历史记录中 txt_history.insert(END, \"正在请求处理中...\" + \"\\n\") try: # 文生图片接口 image_path = txt_to_image(user_message) if image_path: # 清空输入框 txt_input.delete(0, END) # 清空所有历史记录 txt_history.delete(\"1.0\", END) # 插入最新消息 txt_history.insert(END, \"成功生成图片如下\" + \"\\n\") # 显示图片到历史对话框中 download_and_display_image(image_path) except Exception as e: txt_history.insert(END, \"后台处理失败，原因如下:\" + \"\\n\") txt_history.insert(END, str(e) + \"\\n\")btn = tkinter.Button(root, text=\"文生图片\", command=txt_generate_image, **button_style)btn.grid(column=1, row=3, padx=5, pady=5)# 运行主窗口root.mainloop()火山引擎的即梦AI接口及改造
即梦AI文生图片接口文档
即梦AI-图像生成--即梦AI-火山引擎
http接口示例：
import jsonimport sysimport osimport base64import datetimeimport hashlibimport hmacimport requestsmethod = \'POST\'host = \'visual.volcengineapi.com\'region = \'cn-north-1\'endpoint = \'https://visual.volcengineapi.com\'service = \'cv\'def sign(key, msg): return hmac.new(key, msg.encode(\'utf-8\'), hashlib.sha256).digest()def getSignatureKey(key, dateStamp, regionName, serviceName): kDate = sign(key.encode(\'utf-8\'), dateStamp) kRegion = sign(kDate, regionName) kService = sign(kRegion, serviceName) kSigning = sign(kService, \'request\') return kSigningdef formatQuery(parameters): request_parameters_init = \'\' for key in sorted(parameters): request_parameters_init += key + \'=\' + parameters[key] + \'&\' request_parameters = request_parameters_init[:-1] return request_parametersdef signV4Request(access_key, secret_key, service, req_query, req_body): if access_key is None or secret_key is None: print(\'No access key is available.\') sys.exit() t = datetime.datetime.utcnow() current_date = t.strftime(\'%Y%m%dT%H%M%SZ\') # current_date = \'20210818T095729Z\' datestamp = t.strftime(\'%Y%m%d\') # Date w/o time, used in credential scope canonical_uri = \'/\' canonical_querystring = req_query signed_headers = \'content-type;host;x-content-sha256;x-date\' payload_hash = hashlib.sha256(req_body.encode(\'utf-8\')).hexdigest() content_type = \'application/json\' canonical_headers = \'content-type:\' + content_type + \'\\n\' + \'host:\' + host + \\ \'\\n\' + \'x-content-sha256:\' + payload_hash + \\ \'\\n\' + \'x-date:\' + current_date + \'\\n\' canonical_request = method + \'\\n\' + canonical_uri + \'\\n\' + canonical_querystring + \\ \'\\n\' + canonical_headers + \'\\n\' + signed_headers + \'\\n\' + payload_hash # print(canonical_request) algorithm = \'HMAC-SHA256\' credential_scope = datestamp + \'/\' + region + \'/\' + service + \'/\' + \'request\' string_to_sign = algorithm + \'\\n\' + current_date + \'\\n\' + credential_scope + \'\\n\' + hashlib.sha256( canonical_request.encode(\'utf-8\')).hexdigest() # print(string_to_sign) signing_key = getSignatureKey(secret_key, datestamp, region, service) # print(signing_key) signature = hmac.new(signing_key, (string_to_sign).encode( \'utf-8\'), hashlib.sha256).hexdigest() # print(signature) authorization_header = algorithm + \' \' + \'Credential=\' + access_key + \'/\' + \\ credential_scope + \', \' + \'SignedHeaders=\' + \\ signed_headers + \', \' + \'Signature=\' + signature # print(authorization_header) headers = {\'X-Date\': current_date, \'Authorization\': authorization_header, \'X-Content-Sha256\': payload_hash, \'Content-Type\': content_type } # print(headers) # ************* SEND THE REQUEST ************* request_url = endpoint + \'?\' + canonical_querystring print(\'\\nBEGIN REQUEST++++++++++++++++++++++++++++++++++++\') print(\'Request URL = \' + request_url) try: r = requests.post(request_url, headers=headers, data=req_body) except Exception as err: print(f\'error occurred: {err}\') raise else: print(\'\\nRESPONSE++++++++++++++++++++++++++++++++++++\') print(f\'Response code: {r.status_code}\\n\') # 使用 replace 方法将 \\u0026 替换为 & resp_str = r.text.replace(\"\\\\u0026\", \"&\") print(f\'Response body: {resp_str}\\n\')if __name__ == \"__main__\": # 请求凭证，从访问控制申请 access_key = \'AK*****\' secret_key = \'*****==\' # 请求Query，按照接口文档中填入即可 query_params = { \'Action\': \'CVProcess\', \'Version\': \'2022-08-31\', } formatted_query = formatQuery(query_params) # 请求Body，按照接口文档中填入即可 body_params = { \"req_key\": \"******\", # ...... } formatted_body = json.dumps(body_params) signV4Request(access_key, secret_key, service, formatted_query, formatted_body)这里要求我们开通服务，并创建对应的AK和SK
账号登录-火山引擎欢迎登录火山引擎，火山引擎是字节跳动旗下的云服务平台。
https://console.volcengine.com/ai/ability/info/104
账号登录-火山引擎
这里我已经开通了，自行选择开通试用就行，不需要点击正式调用，不然收费
接口改造
原来的接口只能查看是否调用成功，我们想获取图片地址，并将图片显示在对话框中，因此需要获取返回值，请求体里面加了\"return_url\":True，就是用来获取图片地址的。同时将认证的AK和SK也填充到自定义函数中，这样接口地址可以正常使用认证并返回我们要的结果。
import jsonimport sysimport osimport base64import datetimeimport hashlibimport hmacimport requestsmethod = \'POST\'host = \'visual.volcengineapi.com\'region = \'cn-north-1\'endpoint = \'https://visual.volcengineapi.com\'service = \'cv\'def sign(key, msg): return hmac.new(key, msg.encode(\'utf-8\'), hashlib.sha256).digest()def getSignatureKey(key, dateStamp, regionName, serviceName): kDate = sign(key.encode(\'utf-8\'), dateStamp) kRegion = sign(kDate, regionName) kService = sign(kRegion, serviceName) kSigning = sign(kService, \'request\') return kSigningdef formatQuery(parameters): request_parameters_init = \'\' for key in sorted(parameters): request_parameters_init += key + \'=\' + parameters[key] + \'&\' request_parameters = request_parameters_init[:-1] return request_parametersdef signV4Request(access_key, secret_key, service, req_query, req_body): if access_key is None or secret_key is None: print(\'No access key is available.\') sys.exit() t = datetime.datetime.utcnow() current_date = t.strftime(\'%Y%m%dT%H%M%SZ\') # current_date = \'20210818T095729Z\' datestamp = t.strftime(\'%Y%m%d\') # Date w/o time, used in credential scope canonical_uri = \'/\' canonical_querystring = req_query signed_headers = \'content-type;host;x-content-sha256;x-date\' payload_hash = hashlib.sha256(req_body.encode(\'utf-8\')).hexdigest() content_type = \'application/json\' canonical_headers = \'content-type:\' + content_type + \'\\n\' + \'host:\' + host + \\ \'\\n\' + \'x-content-sha256:\' + payload_hash + \\ \'\\n\' + \'x-date:\' + current_date + \'\\n\' canonical_request = method + \'\\n\' + canonical_uri + \'\\n\' + canonical_querystring + \\ \'\\n\' + canonical_headers + \'\\n\' + signed_headers + \'\\n\' + payload_hash # print(canonical_request) algorithm = \'HMAC-SHA256\' credential_scope = datestamp + \'/\' + region + \'/\' + service + \'/\' + \'request\' string_to_sign = algorithm + \'\\n\' + current_date + \'\\n\' + credential_scope + \'\\n\' + hashlib.sha256( canonical_request.encode(\'utf-8\')).hexdigest() # print(string_to_sign) signing_key = getSignatureKey(secret_key, datestamp, region, service) # print(signing_key) signature = hmac.new(signing_key, (string_to_sign).encode( \'utf-8\'), hashlib.sha256).hexdigest() # print(signature) authorization_header = algorithm + \' \' + \'Credential=\' + access_key + \'/\' + \\ credential_scope + \', \' + \'SignedHeaders=\' + \\ signed_headers + \', \' + \'Signature=\' + signature # print(authorization_header) headers = {\'X-Date\': current_date, \'Authorization\': authorization_header, \'X-Content-Sha256\': payload_hash, \'Content-Type\': content_type } # print(headers) # ************* SEND THE REQUEST ************* request_url = endpoint + \'?\' + canonical_querystring print(\'\\nBEGIN REQUEST++++++++++++++++++++++++++++++++++++\') # print(\'Request URL = \' + request_url) try: r = requests.post(request_url, headers=headers, data=req_body) except Exception as err: print(f\'error occurred: {err}\') raise else: # print(\'\\nRESPONSE++++++++++++++++++++++++++++++++++++\') print(f\'Response code: {r.status_code}\\n\') # 注释调原来的返回处理，我们不需要文本类型的，我们需要json格式的 # 使用 replace 方法将 \\u0026 替换为 & # resp_str = r.text.replace(\"\\\\u0026\", \"&\") # print(f\'Response body: {resp_str}\\n\') # 获取json类型的返回对象，可进行数据处理，获取返回的image_url print(r.json()[\'data\'][\'image_urls\'][0]) return r.json()[\'data\'][\'image_urls\'][0]def txt_to_image(prompt_text): # 请求凭证，从访问控制申请 access_key = \'\' secret_key = \'\' # 请求Query，按照接口文档中填入即可 query_params = { \'Action\': \'CVProcess\', \'Version\': \'2022-08-31\', } formatted_query = formatQuery(query_params) # 请求Body，按照接口文档中填入即可 body_params = { \"req_key\": \"jimeng_high_aes_general_v21_L\", \"prompt\": prompt_text, \"return_url\": True } formatted_body = json.dumps(body_params) image_url = signV4Request(access_key, secret_key, service, formatted_query, formatted_body) return image_urlif __name__ == \"__main__\": txt_to_image(\'一张海报,上面文字写着:\"新年快乐\"\')请求的body体两个参数必填，因此在body_params中必须放置，其中req_key取固定值，prompt就是我们的输入框中的文本提示词。
测试结果：
到这里就大功告成了！可以愉快的玩耍了
