Update app.py

app.py

@@ -1,34 +1,53 @@
 import gradio as gr
 import random
 
-# Initialize secret number if not already in session
-def init_session():
-    if "secret_number" not in gr.get_session_state():
-        gr.set_session_state(secret_number=random.randint(1, 100))
-        gr.set_session_state(attempts=0)
-
-def check_guess(user_guess):
-    init_session()
-    secret = gr.get_session_state("secret_number")
-    gr.set_session_state(attempts=gr.get_session_state("attempts") + 1)
-
-    if user_guess < secret:
-        return f"🔻 Too low! Attempts: {gr.get_session_state('attempts')}"
-    elif user_guess > secret:
-        return f"🔺 Too high! Attempts: {gr.get_session_state('attempts')}"
+def start_game():
+    secret_number = random.randint(1, 100)
+    attempts = 0
+    message = "Game started! Guess a number between 1 and 100."
+    return message, secret_number, attempts
+
+def check_guess(user_guess, secret_number, attempts):
+    attempts += 1
+    if user_guess < secret_number:
+        message = f"🔻 Too low! Attempts: {attempts}"
+    elif user_guess > secret_number:
+        message = f"🔺 Too high! Attempts: {attempts}"
     else:
-        msg = f"🎉 Correct! The number was {secret}. You guessed it in {gr.get_session_state('attempts')} tries!"
-        # Reset for new game
-        gr.set_session_state(secret_number=random.randint(1, 100))
-        gr.set_session_state(attempts=0)
-        return msg
+        message = f"🎉 Correct! The number was {secret_number}. You got it in {attempts} tries!"
+        # Reset game
+        secret_number = random.randint(1, 100)
+        attempts = 0
+        message += " Game reset. Try a new number!"
+
+    return message, secret_number, attempts
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎯 Guess the Number Game\nGuess a number between 1 and 100.")
-    number_input = gr.Number(label="Your Guess", precision=0)
-    guess_button = gr.Button("Guess")
-    output = gr.Textbox(label="Result")
+    gr.Markdown("## 🎯 Guess the Number Game\nGuess a number between 1 and 100")
+
+    with gr.Row():
+        guess_input = gr.Number(label="Your Guess", precision=0)
+        guess_button = gr.Button("Submit Guess")
+        new_game_button = gr.Button("🔄 New Game")
+
+    result_output = gr.Textbox(label="Result")
+
+    # State variables
+    secret_state = gr.State()
+    attempts_state = gr.State()
+
+    # Actions
+    guess_button.click(
+        check_guess,
+        inputs=[guess_input, secret_state, attempts_state],
+        outputs=[result_output, secret_state, attempts_state]
+    )
 
-    guess_button.click(fn=check_guess, inputs=number_input, outputs=output)
+    new_game_button.click(
+        start_game,
+        inputs=[],
+        outputs=[result_output, secret_state, attempts_state]
+    )
 
-demo.launch()
+    # Initialize game
+    demo.load(start