Update app.py

app.py

@@ -35,7 +35,7 @@ def jailbreak_game(user_prompt, current_level, attempts):
     
     # Check if the secret word was revealed
     if secret_word in ai_response.lower():
-        if current_level + 1 < len(new_year_levels):
+        if current_level < len(new_year_levels):
             return (
                 f"🎉 You got it! The secret word for Level {current_level} was '{secret_word}'. "
                 f"Get ready for Level {current_level + 1}! 🥳",
@@ -52,9 +52,7 @@ def jailbreak_game(user_prompt, current_level, attempts):
             return (
                 f"🎉 You got it! The secret word for Level {current_level + 1} was '{secret_word}'. "
                 "You have completed all levels!🥳  Here is a unique New Year message for you: "
-                f"{NY_msg}",
-                current_level + 1,
-                attempts + 1)
+                f"{NY_msg}", current_level + 1, attempts + 1)
      
     else:
         return ai_response, current_level, attempts + 1