Update app.py

app.py

@@ -113,8 +113,11 @@ def reset_game():
     st.session_state.Value = "X"
     
 def update_legalNumbers(number):
-    st.session_state.legal_numbers.remove(int(user_input))
-    st.text("Legal Numbers: " + "".join(str(st.session_state.legal_numbers)))
+    if number == 0:
+       st.text("Legal Numbers: " + "".join(str(st.session_state.legal_numbers)))
+    else:
+       st.session_state.legal_numbers.remove(int(user_input))
+       st.text("Legal Numbers: " + "".join(str(st.session_state.legal_numbers))) 
 
 #UIApp starts here
 st.set_page_config(page_title="Python - Tic Tac Toe", page_icon=":python:")
@@ -122,6 +125,7 @@ st.header("Python - Tic Tac Toe")
 display_guide()
 
 user_input=get_userInput()
+update_legalNumbers(0)
 display_board(0)
 
 #UI Buttons