SmallBot / app.py
hertogateis's picture
Update app.py
61e5bf4 verified
raw
history blame
1.79 kB
import streamlit as st
import requests
# Set up Streamlit page configuration
st.set_page_config(page_title="DeepSeek Chatbot", page_icon="🤖", layout="wide")
# API setup
url = "https://api.hyperbolic.xyz/v1/chat/completions"
headers = {
"Content-Type": "application/json",
"Authorization": "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiJtaXRyYWxlc3RhcmlwZXJzYWRhQGdtYWlsLmNvbSIsImlhdCI6MTczNjUwMzQxMX0.yuIoZsH1jouAlixx_h_eQ-bltZ1sg4alrJHMHr1axvA"
}
# Chat history container
if 'messages' not in st.session_state:
st.session_state.messages = []
# Function to send message and get response
def get_response(user_input):
data = {
"messages": [{"role": "user", "content": user_input}],
"model": "deepseek-ai/DeepSeek-V3",
"max_tokens": 512,
"temperature": 0.1,
"top_p": 0.9
}
response = requests.post(url, headers=headers, json=data)
return response.json()
# Streamlit chat UI
st.title("DeepSeek AI Chatbot")
# Display the chat history
for message in st.session_state.messages:
if message["role"] == "user":
st.chat_message("user").markdown(message["content"])
else:
st.chat_message("assistant").markdown(message["content"])
# Accept user input
user_input = st.text_input("You: ", "")
# Handle user input and update the chat
if user_input:
st.session_state.messages.append({"role": "user", "content": user_input})
response = get_response(user_input)
# Assuming the response is in the 'choices' field of the API response
bot_response = response.get('choices', [{}])[0].get('message', {}).get('content', 'Sorry, I did not understand that.')
st.session_state.messages.append({"role": "assistant", "content": bot_response})
st.experimental_rerun()