Update app.py

app.py

@@ -11,6 +11,10 @@ retrieval_model_name = 'output/sentence-transformer-finetuned/'
 
 openai.api_key = os.environ["OPENAI_API_KEY"]
 
+system_message = "You are a chess chatbot specialized in providing information on chess rules, strategies, and terminology."
+# Initial system message to set the behavior of the assistant
+messages = [{"role": "system", "content": system_message}]
+
 # Attempt to load the necessary models and provide feedback on success or failure
 try:
     retrieval_model = SentenceTransformer(retrieval_model_name)
@@ -63,12 +67,12 @@ def generate_response(user_query, relevant_segment):
     Generate a response emphasizing the bot's capability in providing chess information.
     """
     try:
-        system_message = "You are a chess chatbot specialized in providing information on chess rules, strategies, and terminology."
+        
         user_message = f"Here's the information on chess: {relevant_segment}"
-        messages = [
-            {"role": "system", "content": system_message},
-            {"role": "user", "content": user_message}
-        ]
+
+        # Append user's message to messages list
+        messages.append({"role": "user", "content": user_message})
+
         response = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
             messages=messages,
@@ -78,7 +82,15 @@ def generate_response(user_query, relevant_segment):
             frequency_penalty=0,
             presence_penalty=0
         )
-        return response['choices'][0]['message']['content'].strip()
+        
+        # Extract the response text
+        output_text = response['choices'][0]['message']['content'].strip()
+        
+        # Append assistant's message to messages list for context
+        messages.append({"role": "assistant", "content": output_text})
+        
+        return output_text
+        
     except Exception as e:
         print(f"Error in generating response: {e}")
         return f"Error in generating response: {e}"