Update app.py

app.py

@@ -1,12 +1,11 @@
 import os
 import gradio as gr
-import openai
-from langchain.llms import OpenAI
+from langchain.chat_models import ChatOpenAI
 from langchain.chains import LLMChain
-from langchain.prompts import PromptTemplate
+from langchain.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessagePromptTemplate
 
 # Set up OpenAI API key
-openai.api_key = os.environ.get("OPENAI_API_KEY")
+os.environ["OPENAI_API_KEY"] = os.environ.get("OPENAI_API_KEY")
 
 # Define mental health resources
 mental_health_resources = """
@@ -38,17 +37,13 @@ Remember, seeking professional help is important if symptoms persist or interfer
 """
 
 # Define prompt template
-template = """
-You are an empathetic and supportive mental health chatbot. A user has shared the following:
-
-{user_input}
-
-Based on this, provide a response that:
-1. Acknowledges their feelings
-2. Offers a comforting perspective
-3. Suggests a coping strategy
-4. Asks a follow-up question to encourage further discussion
-5. Reminds them that professional help is available if needed
+system_template = """
+You are an empathetic and supportive mental health chatbot. Provide responses that:
+1. Acknowledge the user's feelings
+2. Offer a comforting perspective
+3. Suggest a coping strategy
+4. Ask a follow-up question to encourage further discussion
+5. Remind them that professional help is available if needed
 
 Keep the tone warm and supportive throughout.
 
@@ -57,19 +52,19 @@ Consider the following mental health information when responding:
 
 Chat History:
 {chat_history}
-
-Response:
 """
 
-prompt = PromptTemplate(
-    template=template,
-    input_variables=["user_input", "mental_health_info", "chat_history"]
-)
+human_template = "{user_input}"
+
+system_message_prompt = SystemMessagePromptTemplate.from_template(system_template)
+human_message_prompt = HumanMessagePromptTemplate.from_template(human_template)
+
+chat_prompt = ChatPromptTemplate.from_messages([system_message_prompt, human_message_prompt])
 
 # Set up main chatbot chain
 def setup_chatbot_chain():
-    llm = OpenAI(temperature=0.7)
-    return LLMChain(llm=llm, prompt=prompt)
+    llm = ChatOpenAI(temperature=0.7, model_name="gpt-3.5-turbo")
+    return LLMChain(llm=llm, prompt=chat_prompt)
 
 chatbot_chain = setup_chatbot_chain()