prompt count feature

convosim.py

@@ -10,8 +10,11 @@ from app_config import ISSUES, SOURCES, source2label
 
 logger = get_logger(__name__)
 openai_api_key = os.environ['OPENAI_API_KEY'] 
-memories = {'memory':{"issue": ISSUES[0], "source": SOURCES[0]}}
 
+if "sent_messages" not in st.session_state:
+    st.session_state['sent_messages'] = 0
+if "issue" not in st.session_state:
+    st.session_state['issue'] = ISSUES[0]
 if 'previous_source' not in st.session_state:
     st.session_state['previous_source'] = SOURCES[0]
 if 'db_client' not in st.session_state:
@@ -22,6 +25,8 @@ if 'texter_name' not in st.session_state:
     st.session_state["texter_name"] = get_random_name(names_df=DEFAULT_NAMES_DF)
     logger.info(f"texter name is {st.session_state['texter_name']}")
 
+memories = {'memory':{"issue": st.session_state['issue'], "source": st.session_state['previous_source']}}
+
 with st.sidebar:
     username = st.text_input("Username", value='ivnban-ctl', max_chars=30)
     temperature = st.slider("Temperature", 0., 1., value=0.8, step=0.1)
@@ -37,8 +42,8 @@ with st.sidebar:
     source = st.selectbox("Select a source Model A", SOURCES, index=0,
                           format_func=source2label, 
                         )
+    st.markdown(f"### Previous Prompt Count: :red[**{st.session_state['sent_messages']}**]")
 
-memories = {'memory':{"issue":issue, "source":source}}
 changed_source = any([
     st.session_state['previous_source'] != source,
     st.session_state['issue'] != issue
@@ -66,6 +71,7 @@ for msg in memoryA.buffer_as_messages:
     st.chat_message(role).write(msg.content)
 
 if prompt := st.chat_input():
+    st.session_state['sent_messages'] += 1
     if 'convo_id' not in st.session_state:
         push_convo2db(memories, username, language)
 

pages/comparisor.py

@@ -133,6 +133,8 @@ with st.sidebar:
                                format_func=source2label
                             )
         
+    st.markdown(f"### Previous Prompt Count: :red[**{st.session_state['sent_messages']}**]")
+        
     sbcol1, sbcol2 = st.columns(2)
     beta = sbcol1.button("A is better", on_click=replaceB)
     betb = sbcol2.button("B is better", on_click=replaceA)