Update app.py

app.py

@@ -12,8 +12,8 @@ def validate_userInput(user_input):
         if int(user_input) < 1 or int(user_input) > 9:
             return "Please enter a digit between 1 to 9."
         else:    
-            legal_numbers.pop(int(user_input)+1)
-            update_legalNumbers()
+            legal_numbers.pop(int(user_input)-1)
+            return legal_numbers
     else:
         return "Please enter a digit between 1 to 9."
 
@@ -52,21 +52,24 @@ st.set_page_config(page_title="Python - Tic Tac Toe", page_icon=":python:")
 st.header("Python - Tic Tac Toe")
 
 user_input=get_userInput()
-update_legalNumbers()
 display_board()
 
 #UI Buttons
 submit=st.button('Submit')
 replay=st.button('Replay')
 
-#After User Interaction
-response=validate_userInput(user_input)
-    
 #Button functionality
 if submit:
     st.subheader("")
-    st.write(response)
+    update_legalNumbers()
+    #st.write(response)
+else:
+   update_legalNumbers() 
 
 if replay:
     display_board()
-    reset_legalNumbers()
+    reset_legalNumbers()
+
+#After User Interaction
+response=validate_userInput(user_input)
+