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.")