Update run.py

run.py

@@ -28,8 +28,6 @@ def respond(state, chat_history, visible_elemIDs):
   global internal_state
   previous_state = internal_state
   internal_state = state
-  visible_elemIDs= [e for e in visible_elemIDs 
-    if ((str(e) != str(internal_state))|(str(e)=="1"))] # delete some buttons
   if(str(state) in [str(e) for e in visible_elemIDs]):  # react to Buttons
     bot_message = f"I was in state \"{previous_state}\" and am now in state \"{internal_state}\". please click another button."
     print("respond to button!")
@@ -40,6 +38,8 @@ def respond(state, chat_history, visible_elemIDs):
     state="0"
     print("respond to text!")
     chat_history.append((state, bot_message))
+  visible_elemIDs= [e for e in visible_elemIDs 
+    if ((str(e) != str(internal_state))|(str(e)=="1")|(str(e)=="0"))] # delete some buttons
   if(len(visible_elemIDs)==0): # if visible_elemIDs has zero length, reset
     print("restore visibility to "+str(defaultElemIDs)) 
     visible_elemIDs=defaultElemIDs