92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
"""
|
|
This code is supported by the website: https://www.guanjihuan.com
|
|
The newest version of this code is on the web page: https://www.guanjihuan.com/archives/38502
|
|
"""
|
|
|
|
import streamlit as st
|
|
st.set_page_config(
|
|
page_title="Chat",
|
|
layout='wide'
|
|
)
|
|
|
|
try:
|
|
import zhipuai
|
|
except:
|
|
import os
|
|
os.system('pip install zhipuai')
|
|
import zhipuai
|
|
|
|
# 从官网获取 API_KEY
|
|
zhipuai.api_key = " "
|
|
|
|
with st.sidebar:
|
|
with st.expander('参数', expanded=True):
|
|
top_p = st.slider('top_p', 0.01, 1.0, value=0.7, step=0.01)
|
|
temperature = st.slider('temperature', 0.01, 1.0, value=0.95, step=0.01)
|
|
|
|
def chatglm_chat(prompt=[]):
|
|
response = zhipuai.model_api.sse_invoke(
|
|
model="chatglm_turbo",
|
|
prompt=prompt,
|
|
temperature=temperature,
|
|
top_p=top_p,
|
|
)
|
|
return response
|
|
|
|
def getlength(text):
|
|
length = 0
|
|
for content in text:
|
|
temp = content["content"]
|
|
leng = len(temp)
|
|
length += leng
|
|
return length
|
|
|
|
def checklen(text):
|
|
while (getlength(text) > 8000):
|
|
del text[0]
|
|
return text
|
|
|
|
def getText(role,content, text):
|
|
jsoncon = {}
|
|
jsoncon["role"] = role
|
|
jsoncon["content"] = content
|
|
text.append(jsoncon)
|
|
return text
|
|
|
|
answer = ""
|
|
if "text0" not in st.session_state:
|
|
st.session_state.text0 = []
|
|
if "messages0" not in st.session_state:
|
|
st.session_state.messages0 = []
|
|
def clear_all0():
|
|
st.session_state.messages0 = []
|
|
st.session_state.text0 = []
|
|
if st.session_state.messages0 == []:
|
|
with st.chat_message("user", avatar="user"):
|
|
input_placeholder = st.empty()
|
|
with st.chat_message("robot", avatar="assistant"):
|
|
message_placeholder = st.empty()
|
|
for message in st.session_state.messages0:
|
|
with st.chat_message(message["role"], avatar=message.get("avatar")):
|
|
st.markdown(message["content"])
|
|
prompt_text = st.chat_input("请在这里输入您的命令")
|
|
|
|
if prompt_text:
|
|
if st.session_state.messages0 != []:
|
|
with st.chat_message("user", avatar="user"):
|
|
input_placeholder = st.empty()
|
|
with st.chat_message("robot", avatar="assistant"):
|
|
message_placeholder = st.empty()
|
|
input_placeholder.markdown(prompt_text)
|
|
st.session_state.messages0.append({"role": "user", "content": prompt_text, "avatar": "user"})
|
|
st.session_state.text0 = getText("user", prompt_text, st.session_state.text0)
|
|
question = checklen(st.session_state.text0)
|
|
response = chatglm_chat(question)
|
|
for event in response.events():
|
|
answer += event.data
|
|
message_placeholder.markdown(answer)
|
|
st.session_state.text0 = getText("assistant", answer, st.session_state.text0)
|
|
st.session_state.messages0.append({"role": "robot", "content": answer, "avatar": "assistant"})
|
|
st.rerun()
|
|
button_clear = st.button("清空", on_click=clear_all0, key='clear0')
|