import gradio as gr
import openai
import os
from openai import OpenAI

# ใส่ API Key
client = OpenAI(
    api_key=os.environ.get("OPENAI_API_KEY"),  # This is the default and can be omitted
)


def generate_response(prompt):
    try:
        response = client.chat.completions.create(
            messages=[
                {
                    "role": "user",
                    "content": "Say this is a test",
                }
            ],
            model="gpt-4o-mini",
        )
        return response.choices[0].message["content"]
    except Exception as e:
        return f"Error: {str(e)}"

# สร้างอินเทอร์เฟซ Gradio
with gr.Blocks() as demo:
    gr.Markdown("## Test OpenAI GPT API")
    user_input = gr.Textbox(label="Enter your prompt")
    output = gr.Textbox(label="GPT Response")
    submit = gr.Button("Generate Response")
    submit.click(generate_response, inputs=user_input, outputs=output)

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