Update app.py

app.py

@@ -1,29 +1,48 @@
 import gradio as gr
 import openai
+import random
+import time
 
-def generate_code(api_key, prompt):
-    openai.api_key = api_key
+# Set up OpenAI API key
+openai.api_key = "insert your API key here"
 
-    response = openai.Completion.create(
-        engine="text-davinci-002",
-        prompt=prompt,
-        max_tokens=100,
-        n=1,
-        stop=None,
-        temperature=0.5,
-    )
+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]]
 
-    return response.choices[0].text
+    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
+
+    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
+
+    def init_history(messages_history):
+        messages_history = []
+        messages_history += [system_message]
+        return messages_history
+
+    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")
-    prompt_input = gr.inputs.Textbox(lines=5, label="Prompt")
-    output = gr.outputs.Textbox(label="Generated Code")
+    clear.click(lambda: None, None, chatbot, queue=False).success(init_history, [state], [state])
 
-    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()
+demo.launch()