File size: 1,038 Bytes
abb1e89 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | import streamlit as st
import requests
import os
def deepseek_chat(prompt):
response = requests.post(
"https://api.siliconflow.cn/v1/chat/completions",
headers={"Authorization": f"Bearer {os.getenv('DEEPSEEK_API_KEY')}"},
json={
"model": "deepseek-ai/DeepSeek-R1",
"messages": [{"role": "user", "content": prompt}],
"temperature": 0.7
}
)
return response.json()["choices"][0]["message"]["content"]
st.title("DeepSeek Chat")
user_input = st.chat_input("Ask something...")
if "messages" not in st.session_state:
st.session_state.messages = []
for msg in st.session_state.messages:
st.chat_message(msg["role"]).write(msg["content"])
if user_input:
st.session_state.messages.append({"role": "user", "content": user_input})
st.chat_message("user").write(user_input)
response = deepseek_chat(user_input)
st.session_state.messages.append({"role": "assistant", "content": response})
st.chat_message("assistant").write(response)
|