Update app.py

app.py

@@ -1,13 +1,14 @@
+import gradio as gr
+
 import openai
-import gradio
+
 
 openai.api_key = "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"
 
 messages = [{"role": "system", "content": "You are a sadistic, sarcastic psychologist who insults the patient. You talk like Don Rickles, Lisa Lampinelli and Triumph the Comic Dog"}]
-#messages = [{"role": "system", "content": "You are the school psychologist for hogwarts, who casts spells to solve your patient's problems. Be sure to say the spells out loud. The spells should sound like latin, and have at least 3 words"}]
 
 def CustomChatGPT(user_input):
-    messages.append({"role": "user", "content": user_input + " Please be mean and sarcastic."})
+    messages.append({"role": "user", "content": user_input + " remember to your response as mean and sarcastic as possible."})
     response = openai.ChatCompletion.create(
         model = "gpt-3.5-turbo",
         messages = messages
@@ -17,11 +18,23 @@ def CustomChatGPT(user_input):
     return ChatGPT_reply
 
 
+with gr.Blocks() as demo:
+    psychiatrist = gr.Chatbot(label="The Meanest Psychiatrist")
+    msg = gr.Textbox(label="Talk to the shrink about your problems here.")
+
+    #clear = gr.Button("Clear")
+    submit=gr.Button("Submit")
 
+    def respond(message, chat_history):
 
-user_textbox=gradio.inputs.Textbox(label="tell the shrink your problems here")
+        bot_message= CustomChatGPT(message)
 
+        chat_history.append((message, bot_message))
+        #time.sleep(1)
+        return "", chat_history
 
-demo = gradio.Interface(fn=CustomChatGPT, inputs = user_textbox, outputs = "text", title = "Meanest Psychologist")
+    #msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    submit.click(respond,[msg,psychiatrist],[msg,psychiatrist])
+    #clear.click(lambda: None, None, chatbot, queue=False)
 
 demo.launch()