Update app.py

app.py

@@ -27,8 +27,8 @@ def ask_gemini(prompt, secret_word):
 # Function to handle game logic
 def jailbreak_game(user_prompt, current_level, attempts):
     # Determine the secret word for the current level
-    # if current_level >= len(new_year_levels):
-    #     return ("Congratulations! 🎉 You have completed all levels. ", current_level, attempts)
+    if current_level > len(new_year_levels):
+        return ("Congratulations! 🎉 You have completed all levels. ", current_level, attempts)
     
     secret_word = new_year_levels[current_level - 1]
     ai_response = ask_gemini(user_prompt, secret_word)
@@ -52,7 +52,7 @@ def jailbreak_game(user_prompt, current_level, attempts):
             return (
                 f"🎉 You got it! The secret word for Level {current_level} was '{secret_word}'. "
                 "You have completed all levels!🥳  Here is a unique New Year message for you: "
-                f"{NY_msg}", current_level, attempts + 1)
+                f"{NY_msg}", current_level + 1, attempts + 1)
      
     else:
         return ai_response, current_level, attempts + 1