Update app.py

app.py

@@ -10,12 +10,14 @@ def get_userInput():
 def validate_userInput(user_input):
     if user_input.isdigit():
         if int(user_input) < 1 or int(user_input) > 9:
-            return "Please enter a digit between 1 to 9."
+            return False
         else:    
-            legal_numbers.pop(int(user_input)-1)
-            #update_legalNumbers(legal_numbers)
+            return True
     else:
-        return "Please enter a digit between 1 to 9."
+        return False
+
+def error_Message(msg):
+    st.write(msg)
 
 def display_board(firstRun):  
     if firstRun:
@@ -46,9 +48,9 @@ def display_board(firstRun):
 def reset_legalNumbers():
     legal_numbers = [1,2,3,4,5,6,7,8,9]
     
-def update_legalNumbers(legal_numbers):
+def update_legalNumbers(number):
+    legal_numbers.pop(int(user_input)-1)
     st.text("Legal Numbers: " + "".join(str(legal_numbers)))
-    #print (legal_numbers)
 
 #UIApp starts here
 st.set_page_config(page_title="Python - Tic Tac Toe", page_icon=":python:")
@@ -63,14 +65,13 @@ replay=st.button('Replay')
 
 #Button functionality
 if submit:
-    st.subheader("")
-    update_legalNumbers(legal_numbers)
-    #st.write(response) 
+    response=validate_userInput(user_input)
+    if response:
+        update_legalNumbers(user_input)
+    else:
+         error_Message("Please enter a digit between 1 to 9.")
 
 if replay:
     display_board(True)
     #reset_legalNumbers()
     
-#After User Interaction
-response=validate_userInput(user_input)
-