File size: 1,240 Bytes
9ccd396 | 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 | import os
import streamlit as st
from openai import OpenAI
from dotenv import load_dotenv
# Load environment variables
load_dotenv()
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
# Set the OpenAI API key
# Settings for the OpenAI model
settings = {
"model": "gpt-4",
"temperature": 0,
# ... more settings if needed
}
def get_openai_response(user_message):
response = client.chat.completions.create(model=settings["model"],
messages = [
{
"role": "system",
"content": "You are a helpful bot, you always provide programming solutions in python in English",
},
{
"role": "user",
"content": user_message,
},
],
temperature=settings["temperature"])
return response.choices[0].message.content
st.title("Chat with TechMalooBot")
user_message = st.text_input("Enter your message:")
if st.button("Send"):
if user_message:
with st.spinner("Generating response..."):
try:
response = get_openai_response(user_message)
st.write(response)
except Exception as e:
st.write(f"Error: {e}")
else:
st.write("Please enter a message.")
|