Spaces:
Sleeping
Sleeping
File size: 1,730 Bytes
595a71e e626f8a 595a71e 6cc4ab6 595a71e e626f8a 595a71e e626f8a |
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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
import streamlit as st
import openai
import os
from dotenv import load_dotenv
load_dotenv()
# Set up OpenAI API key
openai.api_key = os.getenv("OPENAI_API_KEY") # Replace with your OpenAI API key if not using an environment variable
# Function to query OpenAI
def query_openai(prompt, model="gpt-4o-mini"):
try:
response = openai.ChatCompletion.create(
model=model,
messages=prompt,
max_tokens=200,
temperature=0.7,
)
return response["choices"][0]["message"]["content"]
except Exception as e:
return f"Error: {e}"
# Streamlit app
st.title("Simple OpenAI Chatbot")
# Initialize the chat history
if "messages" not in st.session_state:
st.session_state.messages = [{"role": "system", "content": "You are a helpful assistant. Respond in 20 words"}]
# Display chat history
for message in st.session_state.messages:
if message["role"] == "user":
st.markdown(f"**You:** {message['content']}")
elif message["role"] == "assistant":
st.markdown(f"**Bot:** {message['content']}")
# User input
user_input = st.text_input("Type your message:")
# Handle user input
if st.button("Send"):
if user_input.strip():
# Add user input to chat history
st.session_state.messages.append({"role": "user", "content": user_input})
# Get bot response
bot_response = query_openai(st.session_state.messages)
# Add bot response to chat history
st.session_state.messages.append({"role": "assistant", "content": bot_response})
# Clear chat
if st.button("Clear Chat"):
st.session_state.messages = [{"role": "system", "content": "You are a helpful assistant. Respond in 20 words"}]
|