import gradio as gr
from openai import OpenAI
from utils import get_secret

# Initialize the OpenAI client
client = OpenAI(api_key=get_secret('OPENAI_API_KEY'))

def generate_response(prompt):
    try:
        response = client.chat.completions.create(
            model="gpt-3.5-turbo",
            messages=[
                {"role": "system", "content": "You are a helpful assistant."},
                {"role": "user", "content": prompt}
            ]
        )
        return response.choices[0].message.content
    except Exception as e:
        return str(e)

demo = gr.Interface(
    fn=generate_response,
    inputs="text",
    outputs="text",
    title="OpenAI GPT-3.5 Turbo Chatbot",
    description="Enter a prompt to get a response from OpenAI's GPT-3.5 Turbo model."
)

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