add a chat history box

app.py

@@ -73,6 +73,8 @@ def start(modelName):
             st.session_state['g_t_id'] = 408203
     if 'stop' not in st.session_state:
         st.session_state['stop'] = 0
+    if 'chat_history' not in st.session_state or st.session_state['stop'] == 1:
+        st.session_state['chat_history'] = []
     if 'turn' not in st.session_state or st.session_state['stop'] == 1:
         st.session_state['turn'] = 0
         st.session_state['r_t'] = 0
@@ -103,6 +105,8 @@ def start(modelName):
         st.write('ranker ready')
         st.session_state['ranker'] = ranker_test
 
+    def stop():
+        st.session_state['stop'] = 1
 
     # st.write(st.session_state)
     st.write(st.session_state['model'] + ' loaded')
@@ -112,8 +116,6 @@ def start(modelName):
         st.session_state['turn'] += 1
         st.session_state['r_t'] = int(r_t)
         g_t_id = st.session_state['g_t_id']
-        # st.image(Image.open('./imgs/' + str(g_t_id) + '.jpg').resize((224, 224)), caption='g_t', use_column_width='auto')
-
 
         # st.write(st.session_state)
         def model(r_t, fg_t, fp_t):
@@ -178,7 +180,9 @@ def start(modelName):
         g_t_id_next = run_model(r_t, fg_t, fp_t)
         st.session_state['g_t_id'] = g_t_id_next
         # st.write(st.session_state)
-
+        st.session_state.chat_history.append(
+            {"user": "(positive)" + fp_t + ';(negative)' + fg_t,
+             "bot": str(st.session_state['g_t_id']), "reward": r_t})
 
     fp_t = st.text_input('Input text(positive)', value=' ')
     fg_t = st.text_input('Input text(negative)', value=' ')
@@ -188,10 +192,30 @@ def start(modelName):
         fg_t = ' '
     r_t = st.text_input('Input reward', value='0')
     st.button('Interactive', on_click=interactive, args=(r_t, fg_t, fp_t))
+    st.button('Stop', on_click=stop)
     g_t_id = st.session_state['g_t_id']
     st.image(Image.open('./imgs/' + str(g_t_id) + '.jpg'),
              caption='g_t_next:' + str(g_t_id), use_column_width='auto')
 
+    for i in range(len(st.session_state.chat_history)):
+        st.sidebar.write("User: " + st.session_state.chat_history[i]['user'])
+        st.sidebar.write("reward: " + st.session_state.chat_history[i]['reward'])
+        st.sidebar.write("turn: " + str(i))
+        st.sidebar.write('---------------------------------------------------')
+        st.sidebar.image(Image.open('./imgs/' + st.session_state.chat_history[i]['bot'] + '.jpg'),
+                         caption='g_t_next:' + str(g_t_id), width=224)
+        st.sidebar.write('---------------------------------------------------')
+
+
+
+
+
+
+
+
+
+
+
 
 
 def main():