Fix bugs in sidebar chat (#1)

- Fix bugs in sidebar chat (18ee183ef6fdd2492e3a163b6712b25e827b80dc)


Co-authored-by: Michael Wornow <[email protected]>

app.py

@@ -8,10 +8,11 @@ st.set_page_config(page_title="Tutor", page_icon=":heavy_plus_sign:")
 st.markdown("""
 <div>
 <h3 style='text-align: center;'> Tutor Session with Becky (Student) </h3>
-<p style='text-align: center;'> The bot is a 1st grade student Becky taking a Math class. """
+<p style='text-align: center;'> You are a tutor named Mary. The bot is a 1st grade student Becky taking a Math class. """
             , unsafe_allow_html=True)
 
 # Set org ID and API key
+os.environ['TOKEN'] = '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'
 openai.api_key = os.environ['TOKEN']
 API_KEY = os.environ['TOKEN']
 
@@ -64,95 +65,107 @@ You have a simple vocabulary, elementary grammar and sentence structure, and spe
 
 """
 
-def name_to_role(name):
+def name_to_role_gpt_as_tutor(name):
+    # have GPT-4 act as tutor
     return 'assistant' if name == 'tutor' else 'user'
-def reverse_name_to_role(name):
+def name_to_role_gpt_as_student(name):
+    # have GPT-4 act as student
     return 'assistant' if name == 'student' else 'user'
     
-st.sidebar.title("Sidebar")
-
-if st.sidebar.button('Introduction/Conversation'):
-    with st.chat_message(name_to_role('tutor')): # tutor
-        # assistant == tutor
-        message_placeholder = st.empty()
-        full_response = ""
-        for response in openai.ChatCompletion.create(
-            model=st.session_state["openai_model"],
-            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
-            ],
-            stream=True,
-            api_key=API_KEY
-        ):
-            full_response += response.choices[0].delta.get("content", "")
-        st.sidebar.success(full_response)
-
-if st.sidebar.button('Encouragement'):
-    with st.chat_message(name_to_role('tutor')): # tutor
-        # assistant == tutor
-        message_placeholder = st.empty()
-        full_response = ""
-        for response in openai.ChatCompletion.create(
-            model=st.session_state["openai_model"],
-            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
-            ],
-            stream=True,
-            api_key=API_KEY
-        ):
-            full_response += response.choices[0].delta.get("content", "")
-        st.sidebar.success(full_response)
-
-if st.sidebar.button('Conclude Session'):
-    with st.chat_message(name_to_role('tutor')): # tutor
-        # assistant == tutor
-        message_placeholder = st.empty()
-        full_response = ""
-        for response in openai.ChatCompletion.create(
-            model=st.session_state["openai_model"],
-            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
-            ],
-            stream=True,
-            api_key=API_KEY
-        ):
-            full_response += response.choices[0].delta.get("content", "")
-        st.sidebar.success(full_response)
-
-st.sidebar.text("""
-You are tutor Mary. Your student today is Becky, who is a first grade student here for math class (addition)
-""")
-
 if "openai_model" not in st.session_state:
     st.session_state["openai_model"] = "gpt-3.5-turbo"
 
 if "messages" not in st.session_state:
-    st.session_state.messages = [{ 'name' : 'student', 'content' : "hi" }] # Set initial message from student
+    st.session_state.messages = [{ 'name' : 'student', 'content' : "Hi, my name is Becky and I'm your student." }] # Set initial message from student
     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}
 
+################################
+#
+# Sidebar
+#
+################################
+
 for message in st.session_state.messages:
-    with st.chat_message(name_to_role(message['name'])):
+    with st.chat_message(name_to_role_gpt_as_student(message['name'])):
         st.markdown(message['content'])
 
+st.sidebar.title("Sidebar")
+
+if st.sidebar.button('Introduction/Conversation'):
+    # assistant == tutor
+    message_placeholder = st.empty()
+    full_response = ""
+    for response in openai.ChatCompletion.create(
+        model=st.session_state["openai_model"],
+        messages=[ st.session_state.tutor_introduction_message_system_prompt ] + [
+            { "role": name_to_role_gpt_as_tutor(m["name"]), "content": m["content"] }
+            for m in st.session_state.messages
+        ],
+        stream=True,
+        api_key=API_KEY
+    ):
+        full_response += response.choices[0].delta.get("content", "")
+    st.sidebar.success(full_response)
+
+if st.sidebar.button('Encouragement'):
+    # assistant == tutor
+    message_placeholder = st.empty()
+    full_response = ""
+    for response in openai.ChatCompletion.create(
+        model=st.session_state["openai_model"],
+        messages=[ st.session_state.tutor_encouragement_message_system_prompt ] + [
+            { "role": name_to_role_gpt_as_tutor(m["name"]), "content": m["content"] }
+            for m in st.session_state.messages
+        ],
+        stream=True,
+        api_key=API_KEY
+    ):
+        full_response += response.choices[0].delta.get("content", "")
+    st.sidebar.success(full_response)
+
+if st.sidebar.button('Conclude Session'):
+    # assistant == tutor
+    message_placeholder = st.empty()
+    full_response = ""
+    for response in openai.ChatCompletion.create(
+        model=st.session_state["openai_model"],
+        messages=[ st.session_state.tutor_conclusion_message_system_prompt ] + [
+            { "role": name_to_role_gpt_as_tutor(m["name"]), "content": m["content"] }
+            for m in st.session_state.messages
+        ],
+        stream=True,
+        api_key=API_KEY
+    ):
+        full_response += response.choices[0].delta.get("content", "")
+    st.sidebar.success(full_response)
+
+
+################################
+#
+# Main chat interface
+#
+################################
+
 if prompt := st.chat_input("Start talking with your student!"):
-    st.session_state.messages.append({"role": "user", "content": prompt})
-    with st.chat_message(name_to_role('tutor')):
+    st.session_state.messages.append({ "name": "tutor", "content": prompt })
+    with st.chat_message(name_to_role_gpt_as_student('tutor')):
         st.markdown(prompt)
-
-    with st.chat_message(name_to_role('student')): 
+    
+    with st.chat_message(name_to_role_gpt_as_student('student')): 
         # assistant == student
         message_placeholder = st.empty()
+        print([ st.session_state.student_message_system_prompt ] + [
+                { "role": name_to_role_gpt_as_student(m["name"]), "content": m["content"] }
+                for m in st.session_state.messages
+        ])
         full_response = ""
         for response in openai.ChatCompletion.create(
             model=st.session_state["openai_model"],
             messages=[ st.session_state.student_message_system_prompt ] + [
-                { "role": reverse_name_to_role(m["name"]), "content": m["content"] }
+                { "role": name_to_role_gpt_as_student(m["name"]), "content": m["content"] }
                 for m in st.session_state.messages
             ],
             stream=True,
@@ -161,5 +174,4 @@ if prompt := st.chat_input("Start talking with your student!"):
             full_response += response.choices[0].delta.get("content", "")
             message_placeholder.markdown(full_response + "▌")
         message_placeholder.markdown(full_response)
-    print("Append to student:", full_response)
-    st.session_state.messages.append({ "name": "student", "content": full_response })
+        st.session_state.messages.append({ "name": "student", "content": full_response })