Update app.py

app.py

@@ -73,41 +73,42 @@ for message in st.session_state.messages:
     with st.chat_message(name_to_role(message['name'])):
         st.markdown(message['content'])
 
-for i in range(LEN_CONVERSATION):
-    with st.chat_message(name_to_role('student')): 
-        # assistant == student
-        message_placeholder = st.empty()
-        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"] }
-                for m in st.session_state.messages
-            ],
-            stream=True,
-            api_key=API_KEY
-        ):
-            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 })
-if st.button('Small Talk'):
-    with st.chat_message(name_to_role('tutor')): # tutor
+
+if prompt := st.chat_input("Start talking with your tutor!"):
+    st.session_state.messages.append({"role": "user", "content": prompt})
+    with st.chat_message(name_to_role('tutor')):
+        st.markdown(prompt)
+        
+        if st.button('Small Talk'):
         # 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_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", "")
+                message_placeholder.markdown(full_response + "▌")
+            message_placeholder.markdown(full_response)
+            st.success(full_response)
+            
+    with st.chat_message(name_to_role('student')):
         message_placeholder = st.empty()
         full_response = ""
         for response in openai.ChatCompletion.create(
             model=st.session_state["openai_model"],
-            messages=[ st.session_state.tutor_message_system_prompt ] + [
-                { "role": name_to_role(m["name"]), "content": m["content"] }
+            messages=[
+                {"role": m["role"], "content": m["content"]}
                 for m in st.session_state.messages
             ],
             stream=True,
-            api_key=API_KEY
         ):
             full_response += response.choices[0].delta.get("content", "")
             message_placeholder.markdown(full_response + "▌")
         message_placeholder.markdown(full_response)
-    print("Append to tutor:", full_response)
-    st.success(full_response)
+    st.session_state.messages.append({"role": "assistant", "content": full_response})