import gradio as gr
import openai

def generate_code(api_key, prompt):
    openai.api_key = api_key

    response = openai.Completion.create(
        engine="text-davinci-002",
        prompt=prompt,
        max_tokens=100,
        n=1,
        stop=None,
        temperature=0.5,
    )

    return response.choices[0].text.strip()

if __name__ == "__main__":
    api_key_input = gr.inputs.Textbox(lines=1, label="OpenAI API Key", type="password")
    prompt_input = gr.inputs.Textbox(lines=5, label="Prompt")
    output = gr.outputs.Textbox(label="Generated Code")

    gr.Interface(
        fn=generate_code,
        inputs=[api_key_input, prompt_input],
        outputs=output,
        title="OpenAI Code Generator",
        description="Generate code using OpenAI's GPT-4 model. Enter your API key and a prompt, and let the AI write code for you!",
    ).launch()