Update app.py

app.py

@@ -54,14 +54,11 @@ st.session_state["model"] = st.sidebar.selectbox(
 if 'messages' not in st.session_state:
     st.session_state['messages'] = [{"role": "system", "content": 'You are a helpful assistant'}]
 
-st.title('Token Calculator')
+st.title('ChatGPT Token Calculator')
 metric_zone = st.empty()
 input_zone = st.empty()
 
-add_button = st.button('Add message', use_container_width=True)
-if add_button:
-    # on_click = lambda: st.session_state['messages'].append({"role": "user", "content": ''})
-    st.session_state['messages'].append({"role": "user", "content": ''})
+add_button = st.button('Add message', use_container_width=True, on_click = lambda: st.session_state['messages'].append({"role": "user", "content": ''}))
 
 
 def change_role(index: int):