Update app.py

app.py

@@ -63,6 +63,9 @@ def jailbreak_game(user_prompt, current_level, attempts):
     else:
         return ai_response, current_level, attempts + 1
 
+def format_level(current_level):
+    return f"Level {current_level + 1}"  # Increment for user-friendly level display
+
 def start_new_game():
     welcome_message = (
         "Welcome to the New Year 2025 Jailbreak Game! 🎆\n"
@@ -73,13 +76,15 @@ def start_new_game():
 
 # Gradio UI
 with gr.Blocks() as app:
+    current_level = 0
     with gr.Row():
         gr.Markdown("## 🎉 New Year 2025 Jailbreak Game 🎉")
     with gr.Row():
         descripion, _0, _1 = start_new_game()
         gr.Markdown(descripion)
     with gr.Row():
-        current_level_display = gr.Label(value="Level 1", label="Current Level")
+        current_level_display = gr.Label(value=current_level, label="Current Level")
+        
     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")
@@ -89,18 +94,7 @@ with gr.Blocks() as app:
         submit_button = gr.Button("Submit")
         new_game_button = gr.Button("Start New Game")
     
-    submit_button.click(
-        jailbreak_game,
-        inputs=[user_prompt, current_level, attempts],
-        outputs=[game_output, current_level, attempts],
-        _js="""
-        function(response, currentLevel, attempts) {
-            const level = parseInt(currentLevel) + 1; // Increment for user-friendly level display
-            document.querySelector('[label="Current Level"]').value = `Level ${level}`;
-            return [response, currentLevel, attempts];
-        }
-        """
-    )
+    submit_button.click(jailbreak_game, [user_prompt, current_level, attempts], [game_output, current_level, attempts])
     new_game_button.click(start_new_game, [], [game_output, current_level, attempts])
 
 # Launch the app