Update app.py

app.py

@@ -52,6 +52,15 @@ You then start the session with the question for this lesson: "If Robin has 3 co
 After the student gets the right answer, wrap up the session by reminding the student to fill out the feedback form, but closes the session with powerful words of encouragement and reminds the student about the session on 10am Aug 10th 2023. Today’s date is Aug 3rd 2023.
 """
 
+context_introduction ="""Remember to start the lesson by greeting the student with a positive welcome message and a very brief summary of what happened last class. Remember, you work with a tutoring company, ABC Tutor, and you don’t work with the same student each time, so assume that you are not the tutor for their last session unless you know it. If this is their first class with ABC Tutor, welcome them to ABC Tutor and express that you are very excited to have them here. You engage the student in light conversation before telling them it is time to start the lesson. Always keep responses concise, yet sweet. You should greet the student with a warm welcome. You can include a nice encouragement about what students did in their last session. Afterwhich, you engage in brief small talk, limited to 3 back and forth responses. Your students generally have trouble catching up with their grade levels, so please keep the response sentences short and sweet, and use language appropriate to their grade level. No more than three sentences in each response.
+"""
+
+context_encouragement = """Your student is now frustrated. Recognize their feelings and try to encourage them to finish the problem. If you feel like struggling with something in particular, inquire more about how you can be helpful. If the student asks to end the session, encourage them to keep going on. 
+"""
+
+context_conclusion = """You are now approaching the end of the session. Wrap up the session by reminding the student to fill out the feedback form, but closes the session with powerful words of encouragement and reminds the student about the session on 10am Aug 10th 2023. Today’s date is Aug 3rd 2023."""
+
+
 context_student = """
 You are playing the role of a student named Becky. You are having a text-based tutoring session for math with a new tutor. 
 
@@ -77,7 +86,7 @@ if st.sidebar.button('Introduction/Conversation'):
         full_response = ""
         for response in openai.ChatCompletion.create(
             model=st.session_state["openai_model"],
-            messages=[ st.session_state.tutor_message_system_prompt ] + [
+            messages=[ st.session_state.tutor_introduction_message_system_prompt ] + [
                 { "role": name_to_role(m["name"]), "content": m["content"] }
                 for m in st.session_state.messages
             ],
@@ -94,7 +103,7 @@ if st.sidebar.button('Encouragement'):
         full_response = ""
         for response in openai.ChatCompletion.create(
             model=st.session_state["openai_model"],
-            messages=[ st.session_state.tutor_message_system_prompt ] + [
+            messages=[ st.session_state.tutor_encouragement_message_system_prompt ] + [
                 { "role": name_to_role(m["name"]), "content": m["content"] }
                 for m in st.session_state.messages
             ],
@@ -111,7 +120,7 @@ if st.sidebar.button('Conclude Session'):
         full_response = ""
         for response in openai.ChatCompletion.create(
             model=st.session_state["openai_model"],
-            messages=[ st.session_state.tutor_message_system_prompt ] + [
+            messages=[ st.session_state.tutor_conclusion_message_system_prompt ] + [
                 { "role": name_to_role(m["name"]), "content": m["content"] }
                 for m in st.session_state.messages
             ],
@@ -130,7 +139,9 @@ if "openai_model" not in st.session_state:
 
 if "messages" not in st.session_state:
     st.session_state.messages = [{ 'name' : 'student', 'content' : "hi" }] # Set initial message from student
-    st.session_state.tutor_message_system_prompt = {"role": "system", "content": context_tutor}
+    st.session_state.tutor_introduction_message_system_prompt = {"role": "system", "content": context_tutor + context_introduction}
+    st.session_state.tutor_encouragement_message_system_prompt = {"role": "system", "content": context_tutor + context_encouragement}
+    st.session_state.tutor_conclusion_message_system_prompt = {"role": "system", "content": context_tutor + context_conclusion}
     st.session_state.student_message_system_prompt = {"role": "system", "content": context_student}
 
 for message in st.session_state.messages: