Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import openai
+import time
 
 # Initial instructions for the assistant
 initial_instructions = {
@@ -24,22 +25,20 @@ initial_instructions = {
 }
 
 # Initialize the conversation history with initial instructions
-conversation_history = [initial_instructions]
+conversation_history = [["system", initial_instructions["content"]]]
 
 def setup_openai(api_key):
     openai.api_key = api_key
     return "API Key Set Successfully!"
 
-def ask_joe(api_key, text):
-    global conversation_history
-
+def ask_joe(api_key, chat_history):
     # set up the api_key
     setup_openai(api_key)
-
+    message = chat_history[-1][0]  # get the last user message
     # Add the user's message to the conversation history
     conversation_history.append({
         "role": "user",
-        "content": text
+        "content": message
     })
     
     # Use the conversation history as the input to the model
@@ -61,25 +60,27 @@ def ask_joe(api_key, text):
 
     # Write the conversation history to a file
     with open('conversation_history.txt', 'a') as f:
-        f.write(f'User: {text}\n')
+        f.write(f'User: {message}\n')
         f.write(f'AI: {model_message}\n')
-    
-    return model_message
+
+    time.sleep(1)  # sleep for a bit for better UI experience
+    return chat_history + [[message, model_message]]
 
 def reset_conversation(api_key):
     global conversation_history
     # Reset the conversation history with initial instructions
-    conversation_history = [initial_instructions]
-    return "Conversation cleared."
+    conversation_history = [["system", initial_instructions["content"]]]
+    return []
+
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    api_key = gr.Textbox(default="your-openai-api-key", label="OpenAI API Key")
+
+    api_key.submit(setup_openai, outputs=gr.Text(label="API Key Status"))
 
-inputs_chat = gr.inputs.Textbox(lines=2, placeholder='Enter your question here')
-iface_chat = gr.Interface(fn=ask_joe, 
-                          inputs=[gr.inputs.Textbox(label="OpenAI API Key"), inputs_chat], 
-                          outputs=gr.outputs.Textbox(label="Joe's Response"))
+    chatbot.submit(ask_joe, inputs=[api_key, chatbot], outputs=chatbot)
 
-iface_reset = gr.Interface(fn=reset_conversation, 
-                           inputs=gr.inputs.Textbox(label="OpenAI API Key"), 
-                           outputs=gr.outputs.Textbox(label="Reset Status"))
+    reset_button = gr.Button("Reset Conversation")
+    reset_button.click(reset_conversation, inputs=api_key, outputs=chatbot, queue=False)
 
-iface_chat.launch()
-iface_reset.launch()
+demo.launch()