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()

def app(api_key, chatbot_prompt):
    generated_code = generate_code(api_key, chatbot_prompt)
    return f"Generated Code:\n{generated_code}"

if __name__ == "__main__":
    api_key_input = gr.inputs.Textbox(lines=1, label="OpenAI API Key", type="password")
    chat_interface = gr.Chat(
        inputs=[api_key_input],
        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!",
        examples=[],
        fn=app,
    )
    chat_interface.launch()