import gradio as gr
import openai

# ใส่ API Key
openai.api_key = os.getenv("OPENAI_API_KEY")


def generate_response(prompt):
    try:
        response = openai.ChatCompletion.create(
            model="gpt-4o-mini",  # หรือ gpt-4 หากต้องการ
            messages=[{"role": "user", "content": prompt}]
        )
        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()