Update app.py

app.py

@@ -65,26 +65,47 @@ def display_board(num):
     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:
-        st.text(st.session_state.gridNumbers[1])
-    with col3:
-        st.text(st.session_state.gridNumbers[2])
-    with col4:
-        st.text(st.session_state.gridNumbers[3])
-    with col5:
-        st.text(st.session_state.gridNumbers[4])
-    with col6:
-        st.text(st.session_state.gridNumbers[5])
-    with col7:
-        st.text(st.session_state.gridNumbers[6])
-    with col8:
-        st.text(st.session_state.gridNumbers[7])
-    with col9:
-        st.text(st.session_state.gridNumbers[8])
-    toggleValue()
+    
+    if num = 0:        
+        with col1:
+            st.text(st.session_state.gridNumbers[0])
+        with col2:
+            st.text(st.session_state.gridNumbers[1])
+        with col3:
+            st.text(st.session_state.gridNumbers[2])
+        with col4:
+            st.text(st.session_state.gridNumbers[3])
+        with col5:
+            st.text(st.session_state.gridNumbers[4])
+        with col6:
+            st.text(st.session_state.gridNumbers[5])
+        with col7:
+            st.text(st.session_state.gridNumbers[6])
+        with col8:
+            st.text(st.session_state.gridNumbers[7])
+        with col9:
+            st.text(st.session_state.gridNumbers[8])
+    else:
+        st.session_state.gridNumbers[int(num)-1]= st.session_state.Value    
+        with col1:
+            st.text(st.session_state.gridNumbers[0])
+        with col2:
+            st.text(st.session_state.gridNumbers[1])
+        with col3:
+            st.text(st.session_state.gridNumbers[2])
+        with col4:
+            st.text(st.session_state.gridNumbers[3])
+        with col5:
+            st.text(st.session_state.gridNumbers[4])
+        with col6:
+            st.text(st.session_state.gridNumbers[5])
+        with col7:
+            st.text(st.session_state.gridNumbers[6])
+        with col8:
+            st.text(st.session_state.gridNumbers[7])
+        with col9:
+            st.text(st.session_state.gridNumbers[8])
+        toggleValue()
 
 def reset_game():
     st.session_state.legal_numbers = [1,2,3,4,5,6,7,8,9]
@@ -101,7 +122,7 @@ st.header("Python - Tic Tac Toe")
 display_guide()
 
 user_input=get_userInput()
-display_board()
+display_board(0)
 
 #UI Buttons
 submit=st.button('Submit')