Update app.py

app.py

@@ -12,33 +12,36 @@ 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)
+            legal_numbers = legal_numbers.pop(int(user_input)-1)
             return legal_numbers
     else:
         return "Please enter a digit between 1 to 9."
 
-def display_board():  
-    col1, col2, col3 = st.columns(3)
-    col4, col5, col6 = st.columns(3)
-    col7, col8, col9 = st.columns(3)
-    with col1:
-        st.text("X")
-    with col2:
-       st.text("O")
-    with col3:
-       st.text("...")
-    with col4:
-        st.text(" X ")
-    with col5:
-       st.text(" O ")
-    with col6:
-       st.text(" ... ")
-    with col7:
-        st.text(" X ")
-    with col8:
-       st.text(" O ")
-    with col9:
-       st.text(" ... ")
+def display_board(text):  
+    if len(text) == 0:
+        col1, col2, col3 = st.columns(3)
+        col4, col5, col6 = st.columns(3)
+        col7, col8, col9 = st.columns(3)
+        with col1:
+            st.text("X")
+        with col2:
+           st.text("O")
+        with col3:
+           st.text("...")
+        with col4:
+            st.text(" X ")
+        with col5:
+           st.text(" O ")
+        with col6:
+           st.text(" ... ")
+        with col7:
+            st.text(" X ")
+        with col8:
+           st.text(" O ")
+        with col9:
+           st.text(" ... ")
+    else:
+        
 
 def reset_legalNumbers():
     legal_numbers = [1,2,3,4,5,6,7,8,9]
@@ -54,10 +57,6 @@ st.header("Python - Tic Tac Toe")
 user_input=get_userInput()
 display_board()
 
-#UI Buttons
-submit=st.button('Submit')
-replay=st.button('Replay')
-
 #Button functionality
 if submit:
     st.subheader("")
@@ -70,6 +69,10 @@ if replay:
     display_board()
     reset_legalNumbers()
 
+#UI Buttons
+submit=st.button('Submit')
+replay=st.button('Replay')
+    
 #After User Interaction
 response=validate_userInput(user_input)