clean up code

app.py

@@ -108,40 +108,7 @@ def get_related_words(word, num=5):
       
       return ", ".join(beam_preds[:num])
 
-if 'messages' not in st.session_state:
-  st.session_state.messages = []
-  
-if 'results' not in st.session_state:
-  st.session_state.results = {'results': False, 'results_print': False}
-  
-if 'actions' not in st.session_state:
-  st.session_state.actions = [""]
-  
-if 'counters' not in st.session_state:
-  st.session_state.counters = {"letter_count": 1, "word_count": 0}
-  
-if 'is_helpful' not in st.session_state:
-  st.session_state.is_helpful = {'ask':False}
-  
-if 'descriptions' not in st.session_state:
-  st.session_state.descriptions = []
-  
-st.title("You name it! 🗣")
-
-with st.chat_message('user', avatar='simon.jpg'):
-  st.write("Hey assistant!")
-  
-bot = st.chat_message('assistant')
-bot.write("Hello human! Wanna practice naming some words?")
-
-#for showing history of messages
-for message in st.session_state.messages:
-  if message['role'] == 'user':
-    with st.chat_message(message['role'], avatar='simon.jpg'):
-      st.markdown(message['content'])
-  else:
-    with st.chat_message(message['role']):
-      st.markdown(message['content'])
+#if 'messages' not in st.session_state:
 
 def get_text():
   input_text = st.chat_input()
@@ -239,6 +206,41 @@ def get_available_cues(target):
 # JS: moved the cue generation further down
 #def cue_generation():
  # if st.session_state.actions[-1] == 'cue':
+
+if 'messages' not in st.session_state:
+  st.session_state.messages = []
+  
+if 'results' not in st.session_state:
+  st.session_state.results = {'results': False, 'results_print': False}
+  
+if 'actions' not in st.session_state:
+  st.session_state.actions = [""]
+  
+if 'counters' not in st.session_state:
+  st.session_state.counters = {"letter_count": 1, "word_count": 0}
+  
+if 'is_helpful' not in st.session_state:
+  st.session_state.is_helpful = {'ask':False}
+  
+if 'descriptions' not in st.session_state:
+  st.session_state.descriptions = []
+  
+st.title("You name it! 🗣")
+
+with st.chat_message('user', avatar='simon.jpg'):
+  st.write("Hey assistant!")
+  
+bot = st.chat_message('assistant')
+bot.write("Hello human! Wanna practice naming some words?")
+
+#for showing history of messages
+for message in st.session_state.messages:
+  if message['role'] == 'user':
+    with st.chat_message(message['role'], avatar='simon.jpg'):
+      st.markdown(message['content'])
+  else:
+    with st.chat_message(message['role']):
+      st.markdown(message['content'])
   
 #display user message in chat message container
 prompt = get_text()