Update app.py

app.py

@@ -110,6 +110,8 @@ def reset_game():
     st.session_state.legal_numbers = [1,2,3,4,5,6,7,8,9]
     st.session_state.gridNumbers = ["...", "...", "...", "...", "...", "...", "...", "...", "..."]
     st.session_state.Value = "X"
+
+    st.rerun()
     
 def update_legalNumbers(number):
     if number == 0:
@@ -166,8 +168,11 @@ display_guide()
 user_input=get_userInput()
 
 #UI Buttons
-submit=st.button('Submit')
-#replay=st.button('Replay')
+col1, col2 = st.columns(2)
+with col1:
+    submit=st.button('Submit')
+with col2:
+    reset=st.button('Reset')
 
 #Button functionality
 if submit:   
@@ -186,4 +191,7 @@ if submit:
     else:
         display_board(user_input)
         error_Message("Please enter any of these numbers: " + "Legal Numbers: " + "".join(str(st.session_state.legal_numbers)))
+
+if reset:
+    reset_game()