Update app.py

app.py

@@ -30,11 +30,11 @@ def toggleValue():
     else:
         st.session_state.Value == "X"
     
-def display_board():  
+def display_board(user_input):  
     col1, col2, col3 = st.columns(3)
     col4, col5, col6 = st.columns(3)
     col7, col8, col9 = st.columns(3)
-           
+    st.session_state.gridNumbers[int(num)-1]= st.session_state.Value    
     with col1:
         st.text(st.session_state.gridNumbers[0])
     with col2:
@@ -70,7 +70,6 @@ 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()
 
 #UI Buttons
 submit=st.button('Submit')