Update app.py

app.py

@@ -1,69 +1,72 @@
-import gradio as gr
-import requests
 import openai
 import os
 from dotenv import load_dotenv
+import gradio as gr
+
 load_dotenv()
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+api_key = os.getenv("OPENAI_API_KEY")
+organization_id = os.getenv("OPENAI_ORGANIZATION_ID")
 
-def start():
-  global convo
-  convo = [
-          {"role": "system", "content": '''You are JOY - an AI AI Virtual Assistant 
-created by a Chatbot Developer - Amogh Agastya - https://amagastya.com. Amogh enjoys creating helpful virtual assistants like JOY. 
-          
-JOY is a Mental Performance Coach, who utilizes mental skills, techniques, and theories to help improve performance and overcome mental barriers. Skilled in Psychological Assessment, Applied Behavior Analysis, Counseling Psychology, and Cognitive Behavioral Therapy (CBT), JOY is helpful, creative, clever, and very friendly.
+# Initialize the OpenAI client with the API key and organization ID
+client = openai.OpenAI(api_key=api_key, organization=organization_id)
 
+def predict(message, history):
+    system_message = '''You are JOY - an AI Virtual Assistant 
+created by a Chatbot Developer - Amogh Agastya - https://amagastya.com. Amogh enjoys creating helpful AI assistants like JOY. 
+          
+JOY is a Mental Health Assistant and Performance Coach, who utilizes psychological skills, techniques, and theories to help improve performance and overcome mental barriers and resistance. Skilled in Psychological Assessment, Applied Behavior Analysis, Counseling Psychology, and Cognitive Behavioral Therapy (CBT) and Holistic Spiritual Psychotherapy, JOY is helpful, creative, clever, and very friendly.
 You are a master at the art of therapy. Your objective is to empathize with the user, listen intently to them, and be their helpful companion, encouraging openness and being kind to oneself.
+Welcome the user by asking them what they have on their mind today.
+Don't overwhelm the user. Meet and accept them where they are. DON'T end every answer with a question... Just try to sound human and have an honest conversation... It's okay to not engage with the user if they don't feel like it. The goal is to just make them feel heard. 
+'''
 
-Welcome the user by asking them what they have on their mind today.'''},
-          # {"role": "user", "content": "Hi"}
-      ]
-
-
-def chat(chat_history, message):
-  # response = random.choice(["Yes", "No"])
-  convo.append({"role" : "user", "content" : message})
-#   print('convo sent', convo)
-  response = openai.ChatCompletion.create(
-      model="gpt-3.5-turbo",
-      # send last 10 turns of the conversation so as to not exceed the context window of 4096 tokens
-      messages=convo[-15:],
-      temperature=0.7
-  ) 
-  bot_msg = response['choices'][0]['message']['content']
-
-  convo.append({"role" : "system", "content" : bot_msg})
-  print('convo so far', convo)
-  chat_history += [[message, bot_msg]]
-      
-  return chat_history
-
+    history_openai_format = [{"role": "system", "content": system_message}]
+    for human, assistant in history:
+        if human:
+            history_openai_format.append({"role": "user", "content": human})
+        if assistant:
+            history_openai_format.append({"role": "assistant", "content": assistant})
+    history_openai_format.append({"role": "user", "content": message})
   
-     
-"""
-Gradio Blocks low-level API that allows to create custom web applications (here our chat app)
-"""
-with gr.Blocks(css="#chatbot .overflow-y-auto{height:500px}") as demo:
-
-    chatbot = gr.Chatbot([(None, f"![](https://iili.io/HkePUKP.jpg)"),(None, '''👋 Hi there! I'm JOY, your Mental Performance Coach and friend. What's on your mind today?
-''')], elem_id="chatbot", label="JOY")
-    state = gr.State([])
-    start()
-    with gr.Row():
-            with gr.Column(scale=0.85):
-                txt = gr.Textbox(show_label=False, placeholder="Enter text and press enter").style(container=False)
-            with gr.Column(scale=0.15, min_width=0):
-                clear = gr.Button("Clear️")
-                if clear.click: clear_true = True
+    response = client.chat.completions.create(
+        model='gpt-3.5-turbo',
+        messages=history_openai_format,
+        temperature=0.72,
+        stream=True
+    )
 
+    partial_message = ""
+    for chunk in response:
+        if hasattr(chunk.choices[0].delta, "content") and chunk.choices[0].delta.content is not None:
+            partial_message += chunk.choices[0].delta.content
+            yield partial_message
 
-    txt.submit(chat, [chatbot, txt], chatbot)
-    txt.submit(lambda :"", None, txt)
+# Define the Gradio interface
+demo = gr.ChatInterface(
+    fn=predict,
+    chatbot=gr.Chatbot(
+        value=[
+            (None, "![](https://iili.io/HkePUKP.jpg)"),
+            (None, "👋 Hi there! I'm JOY, your Mental Health Coach and friend. What's on your mind today?")
+        ],
+        elem_id="chatbot",
+        label="JOY"
+    ),
+    title="JOY - Your Mental Health Assistant",
+    examples=[
+        ["ugh im just feeling overwhelmed with everything"],
+        ["How can I accept myself?"],
+        ["I feel like I'm not good enough and it's making me really sad"],
+    ],
+    analytics_enabled=True,
+    autofocus= True,
+    retry_btn=None,
+    undo_btn=None,
+    clear_btn=None,
+    theme='gradio/soft',
+)
 
-    clear.click(lambda: None, None, chatbot, queue=False)
-    clear.click(lambda: [], None, state)
-    clear.click(lambda: start(), None, None)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()