Update app.py

app.py

@@ -1 +1,29 @@
-test
+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()