基于星火API接口制作一款AI交互软件

一、制作UI界面

通过python内置库tkinter制作一个简易的UI界面。

import tkinter as tk
from tkinter import ttk
root = tk.Tk()
root.title("人机交互")

def on_submit():
    pass

def sub_word():
    pass
frame = ttk.Frame(root, padding="10")
frame.grid(row=0, column=0)
entry = ttk.Entry(frame, width=50)
entry.grid(row=1, column=0, padx=5, pady=5)
submit_button = ttk.Button(frame, text="提交", command=on_submit)
submit_button.grid(row=1, column=1, padx=5, pady=5)
submit_button1 = ttk.Button(frame, text="清空", command=sub_word)
submit_button1.grid(row=1, column=2, padx=5, pady=5)

text_widget = tk.Text(frame, wrap='word', width=80, height=20)
text_widget.grid(row=0, column=0, columnspan=3, padx=5, pady=5)

root.mainloop()

二、星火API获取

讯飞星火大模型-AI大语言模型-星火大模型-科大讯飞 (xfyun.cn)

 领取一个api,创建应用。

右侧有APPID、APISecret、APIKey的值

三、调试代码获取

点击开放文档

在开放文档中选择python调用示例选择下载

将APPID、APISecret、APIKey填写完整,可以直接使用测试的代码。

四、代码附带UI界面设计(不包含ID)

# coding: utf-8
import _thread as thread
import os
import time
import base64

import base64
import datetime
import hashlib
import hmac
import json
from urllib.parse import urlparse
import ssl
from datetime import datetime
from time import mktime
from urllib.parse import urlencode
from wsgiref.handlers import format_date_time

import websocket
import openpyxl
from concurrent.futures import ThreadPoolExecutor, as_completed
import os

import tkinter as tk
from tkinter import ttk

def on_submit():
    query = entry.get()
    if query:
        main(
            appid="",
            api_secret="",
            api_key="",
            gpt_url="wss://spark-api.xf-yun.com/v3.5/chat",
            domain="generalv3.5",
            query=query
        )

class Ws_Param(object):
    # 初始化
    def __init__(self, APPID, APIKey, APISecret, gpt_url):
        self.APPID = APPID
        self.APIKey = APIKey
        self.APISecret = APISecret
        self.host = urlparse(gpt_url).netloc
        self.path = urlparse(gpt_url).path
        self.gpt_url = gpt_url

    # 生成url
    def create_url(self):
        # 生成RFC1123格式的时间戳
        now = datetime.now()
        date = format_date_time(mktime(now.timetuple()))

        # 拼接字符串
        signature_origin = "host: " + self.host + "\n"
        signature_origin += "date: " + date + "\n"
        signature_origin += "GET " + self.path + " HTTP/1.1"

        # 进行hmac-sha256进行加密
        signature_sha = hmac.new(self.APISecret.encode('utf-8'), signature_origin.encode('utf-8'),
                                 digestmod=hashlib.sha256).digest()

        signature_sha_base64 = base64.b64encode(signature_sha).decode(encoding='utf-8')

        authorization_origin = f'api_key="{self.APIKey}", algorithm="hmac-sha256", headers="host date request-line", signature="{signature_sha_base64}"'

        authorization = base64.b64encode(authorization_origin.encode('utf-8')).decode(encoding='utf-8')

        # 将请求的鉴权参数组合为字典
        v = {
            "authorization": authorization,
            "date": date,
            "host": self.host
        }
        # 拼接鉴权参数,生成url
        url = self.gpt_url + '?' + urlencode(v)
        # 此处打印出建立连接时候的url,参考本demo的时候可取消上方打印的注释,比对相同参数时生成的url与自己代码生成的url是否一致
        return url


# 收到websocket错误的处理
def on_error(ws, error):
    print("### error:", error)


# 收到websocket关闭的处理
def on_close(ws):
    print("### closed ###")


# 收到websocket连接建立的处理
def on_open(ws):
    thread.start_new_thread(run, (ws,))


def run(ws, *args):
    data = json.dumps(gen_params(appid=ws.appid, query=ws.query, domain=ws.domain))
    ws.send(data)


# 收到websocket消息的处理
def on_message(ws, message):
    # print(message)
    data = json.loads(message)
    code = data['header']['code']
    if code != 0:
        print(f'请求错误: {code}, {data}')
        ws.close()
    else:
        choices = data["payload"]["choices"]
        status = choices["status"]
        content = choices["text"][0]["content"]
        text_widget.insert(tk.END, content)
        #print(content,end='')
        if status == 2:
            print("#### 关闭会话")
            ws.close()


def gen_params(appid, query, domain):
    """
    通过appid和用户的提问来生成请参数
    """

    data = {
        "header": {
            "app_id": appid,
            "uid": "1234",           
            # "patch_id": []    #接入微调模型,对应服务发布后的resourceid          
        },
        "parameter": {
            "chat": {
                "domain": domain,
                "temperature": 0.5,
                "max_tokens": 4096,
                "auditing": "default",
            }
        },
        "payload": {
            "message": {
                "text": [{"role": "user", "content": query}]
            }
        }
    }
    return data


def main(appid, api_secret, api_key, gpt_url, domain, query):
    wsParam = Ws_Param(appid, api_key, api_secret, gpt_url)
    websocket.enableTrace(False)
    wsUrl = wsParam.create_url()

    ws = websocket.WebSocketApp(wsUrl, on_message=on_message, on_error=on_error, on_close=on_close, on_open=on_open)
    ws.appid = appid
    ws.query = query
    ws.domain = domain
    ws.run_forever(sslopt={"cert_reqs": ssl.CERT_NONE})

def sub_word():
    text_widget.delete(1.0,tk.END)
    entry.delete(0,tk.END)

root = tk.Tk()
root.title("人机交互")

frame = ttk.Frame(root, padding="10")
frame.grid(row=0, column=0, sticky=(tk.W, tk.E, tk.N, tk.S))

entry = ttk.Entry(frame, width=50)
entry.grid(row=1, column=0, padx=5, pady=5)

submit_button = ttk.Button(frame, text="提交", command=on_submit)
submit_button.grid(row=1, column=1, padx=5, pady=5)
submit_button1 = ttk.Button(frame, text="清空", command=sub_word)
submit_button1.grid(row=1, column=2, padx=5, pady=5)

text_widget = tk.Text(frame, wrap='word', width=80, height=20)
text_widget.grid(row=0, column=0, columnspan=3, padx=5, pady=5)

root.mainloop()

  • 14
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值