Update app.py

app.py

@@ -196,12 +196,11 @@ stop = False
 st.title("Chatbot Demo")
 
 with st.container():
-    assistant_role_name = st.text_input("User-AI", value="Singapore Tourism Board")
-    user_role_name = st.text_input("Assistant-AI", value="Tourist that has never been to Singapore")
+    assistant_role_name = st.text_input("Assistant-AI", value="Singapore Tourism Board")
+    user_role_name = st.text_input("User-AI", value="Tourist that has never been to Singapore")
     task = st.text_input("Task", value="Discuss the best tourist attractions to see in Singapore")
     word_limit = st.slider("Word limit", min_value=0, max_value=50, value=15)
     chat_turn_limit = st.slider("Max. Messages", min_value=0, max_value=30, value=10)
-    specified_task_container = st.empty()
 
 with st.container():
     chat_container = st.empty()
@@ -216,7 +215,7 @@ with st.container():
 if gen_button:
     specified_task = get_specified_task(assistant_role_name, user_role_name, task, word_limit)
     with chat_container.container():
-        st.write(f"Task: {specified_task}")
+        task = st.text_input("Task", value=specified_task)
 
 if stop_button:
     stop = True
@@ -224,7 +223,7 @@ if stop_button:
 
 if init_button:
     # Initialize agents
-    assistant_sys_msg, user_sys_msg = get_sys_msgs(assistant_role_name, user_role_name, specified_task)
+    assistant_sys_msg, user_sys_msg = get_sys_msgs(assistant_role_name, user_role_name, task)
     assistant_agent = CAMELAgent(assistant_sys_msg, ChatOpenAI(temperature=0.2))
     user_agent = CAMELAgent(user_sys_msg, ChatOpenAI(temperature=0.2))