Update app.py

app.py

@@ -1,31 +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 = "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"
 
-    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.strip()
+    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 app(api_key, chatbot_prompt):
-    generated_code = generate_code(api_key, chatbot_prompt)
-    return f"Generated Code:\n{generated_code}"
+    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
 
-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,
+    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]
     )
-    chat_interface.launch()
+
+    clear.click(lambda: None, None, chatbot, queue=False).success(init_history, [state], [state])
+
+demo.launch()