Update app.py

app.py

@@ -31,15 +31,12 @@ def setup_openai(api_key):
     openai.api_key = api_key
     return "API Key Set Successfully!"
 
-def ask_joe(api_key, chat_history):
+def ask_joe(api_key, message):
     # 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": message
-    })
+    conversation_history.append(["user", message])
     
     # Use the conversation history as the input to the model
     response = openai.ChatCompletion.create(
@@ -53,34 +50,25 @@ def ask_joe(api_key, chat_history):
     model_message = response.choices[0].message['content'].strip()
     
     # Add the model's message to the conversation history
-    conversation_history.append({
-        "role": "assistant",
-        "content": model_message
-    })
+    conversation_history.append(["assistant", model_message])
 
     # Write the conversation history to a file
-    with open('conversation_history.txt', 'a') as f:
-        f.write(f'User: {message}\n')
-        f.write(f'AI: {model_message}\n')
+    with open('conversation_history.txt', 'w') as f:
+        for role, content in conversation_history:
+            f.write(f'{role.capitalize()}: {content}\n')
 
     time.sleep(1)  # sleep for a bit for better UI experience
-    return chat_history + [[message, model_message]]
+    return model_message
 
 def reset_conversation(api_key):
     global conversation_history
     # Reset the conversation history with initial instructions
     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"))
-
-    chatbot.submit(ask_joe, inputs=[api_key, chatbot], outputs=chatbot)
+    return "Conversation reset"
 
-    reset_button = gr.Button("Reset Conversation")
-    reset_button.click(reset_conversation, inputs=api_key, outputs=chatbot, queue=False)
+api_key = gr.inputs.Textbox(label="OpenAI API Key")
+message = gr.inputs.Textbox(label="Your question")
+output_message = gr.outputs.Textbox(label="Joe's answer")
 
-demo.launch()
+reset_interface = gr.Interface(fn=reset_conversation, inputs=api_key, outputs=output_message)
+ask_joe_interface = gr.Interface(fn=ask_joe, inputs=[api