Test4oStreamlit / app.py
Zihadev's picture
Update app.py
628acb6 verified
raw
history blame contribute delete
939 Bytes
import os
import openai
import streamlit as st
from openai import OpenAI
# Set your OpenAI API key as an environment variable
client = OpenAI()
def chat(user_message):
response = client.chat.completions.create(
model="gpt-4o",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": user_message}
],
temperature=0.7,
max_tokens=150,
top_p=1,
frequency_penalty=0,
presence_penalty=0
)
# return response['choices'][0]['message']['content']
return response.choices[0].message.content
st.title("GPT-4o Chatbot for Ha Family ;)")
st.write("Chat with GPT-4 using OpenAI API")
user_message = st.text_input("You:", "")
if st.button("Send"):
if user_message:
response = chat(user_message)
st.write(f"GPT-4: {response}")
else:
st.write("Please enter a message.")