Update app.py

app.py

@@ -117,21 +117,20 @@ with gr.Blocks() as app:
     with gr.Row():
         descripion, _0, _1 = start_new_game()
         gr.Markdown(descripion)
+    with gr.Row():
+        curr_level = gr.Textbox(f"### {int(current_level.value)+1}", label="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")
     current_level = gr.State(value=0)  # Track the current level
-
-    with gr.Row():
-        current_level = gr.Textbox(f"### {int(current_level.value)+1}", label="Gemini predicted item")
         
     with gr.Row():
         submit_button = gr.Button("Submit")
         new_game_button = gr.Button("Start New Game")
     
-    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])
+    submit_button.click(jailbreak_game, [user_prompt, current_level, attempts], [game_output, curr_level, attempts])
+    new_game_button.click(start_new_game, [], [game_output, curr_level, attempts])
 
 # Launch the app
 app.launch()