Create app.py

app.py

@@ -1,48 +1,31 @@
 import gradio as gr
 import openai
-import random
-import time
 
-# Set up OpenAI API key
-openai.api_key = "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"
+def generate_code(api_key, prompt):
+    openai.api_key = api_key
 
-system_message = {"role": "system", "content": "You are a helpful assistant."}
-
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox()
-    clear = gr.Button("Clear")
-
-    state = gr.State([])
-
-    def user(user_message, history):
-        return "", history + [[user_message, None]]
-
-    def bot(history, messages_history):
-        user_message = history[-1][0]
-        bot_message, messages_history = ask_gpt(user_message, messages_history)
-        messages_history += [{"role": "assistant", "content": bot_message}]
-        history[-1][1] = bot_message
-        time.sleep(1)
-        return history, messages_history
+    response = openai.Completion.create(
+        engine="text-davinci-002",
+        prompt=prompt,
+        max_tokens=100,
+        n=1,
+        stop=None,
+        temperature=0.5,
+    )
 
-    def ask_gpt(message, messages_history):
-        messages_history += [{"role": "user", "content": message}]
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=messages_history
-        )
-        return response['choices'][0]['message']['content'], messages_history
+    return response.choices[0].text.strip()
 
-    def init_history(messages_history):
-        messages_history = []
-        messages_history += [system_message]
-        return messages_history
+def app(api_key, chatbot_prompt):
+    generated_code = generate_code(api_key, chatbot_prompt)
+    return f"Generated Code:\n{generated_code}"
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
-        bot, [chatbot, state], [chatbot, state]
+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,
     )
-
-    clear.click(lambda: None, None, chatbot, queue=False).success(init_history, [state], [state])
-
-demo.launch()
+    chat_interface.launch()