# app.py
import os
import openai
import gradio as gr

# Load API key from environment variable
openai.api_key = os.getenv("OPENAI_API_KEY")

def openai_chat(message, history):
    messages = [{"role": "system", "content": "You are a helpful assistant."}]
    for user, bot in history:
        messages.append({"role": "user", "content": user})
        messages.append({"role": "assistant", "content": bot})
    messages.append({"role": "user", "content": message})

    response = openai.ChatCompletion.create(
        model="gpt-4",  # Or "gpt-3.5-turbo"
        messages=messages,
        temperature=0.7,
    )
    return response.choices[0].message["content"]

demo = gr.ChatInterface(
    fn=openai_chat,
    title="OpenAI Chatbot",
    description="Chat with GPT-4 using the OpenAI API",
    theme=gr.themes.Soft()
)

if __name__ == "__main__":
    demo.launch()