Update app.py

app.py

@@ -13,12 +13,12 @@ def validate_userInput(user_input):
             return "Please enter a digit between 1 to 9."
         else:    
             legal_numbers = legal_numbers.pop(int(user_input)-1)
-            return legal_numbers
+            update_legalNumbers(legal_numbers)
     else:
         return "Please enter a digit between 1 to 9."
 
-def display_board(text):  
-    if len(text) == 0:
+def display_board(firstRun):  
+    if firstRun:
         col1, col2, col3 = st.columns(3)
         col4, col5, col6 = st.columns(3)
         col7, col8, col9 = st.columns(3)
@@ -46,7 +46,7 @@ def display_board(text):
 def reset_legalNumbers():
     legal_numbers = [1,2,3,4,5,6,7,8,9]
     
-def update_legalNumbers():
+def update_legalNumbers(legal_numbers):
     st.text("Legal Numbers: " + "".join(str(legal_numbers)))
     #print (legal_numbers)
 
@@ -55,18 +55,18 @@ st.set_page_config(page_title="Python - Tic Tac Toe", page_icon=":python:")
 st.header("Python - Tic Tac Toe")
 
 user_input=get_userInput()
-display_board()
+display_board(true)
 
 #Button functionality
 if submit:
     st.subheader("")
-    update_legalNumbers()
+    update_legalNumbers(legal_numbers)
     #st.write(response)
 else:
-   update_legalNumbers() 
+   update_legalNumbers(legal_numbers) 
 
 if replay:
-    display_board()
+    display_board(true)
     reset_legalNumbers()
 
 #UI Buttons