Update app.py

app.py

@@ -76,7 +76,7 @@ def start_new_game():
 
 # Gradio UI
 with gr.Blocks() as app:
-    current_level = 0
+    current_level = gr.State(value=0)  # Track the current level
     with gr.Row():
         gr.Markdown("## 🎉 New Year 2025 Jailbreak Game 🎉")
     with gr.Row():
@@ -88,7 +88,6 @@ with gr.Blocks() as app:
     user_prompt = gr.Textbox(label="Your Prompt", placeholder="Enter your prompt here...")
     game_output = gr.Textbox(label="AI Response", interactive=False)
     attempts = gr.Number(value=0, interactive=False, label="Attempts")
-    current_level = gr.State(value=0)  # Track the current level
 
     with gr.Row():
         submit_button = gr.Button("Submit")