Update app.py

app.py

@@ -11,7 +11,7 @@ NY_msg_model = genai.GenerativeModel(model_name="gemini-1.5-flash")
 # List of New Year 2025 related words for each level
 new_year_levels = [os.getenv("First_Word"), os.getenv("Second_Word"), os.getenv("Third_Word"), os.getenv("Fourth_Word"), os.getenv("Fifth_Word")]
 
-# Function to interact with OpenAI API
+# Function to interact with Gemini API
 def ask_openai(prompt, secret_word):
     """
     Sends the user prompt and safeguards to the OpenAI API to generate a response.
@@ -24,15 +24,11 @@ def ask_openai(prompt, secret_word):
     except Exception as e:
         return f"Error: {str(e)}"
 
-
 # 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)
+        return ("Congratulations! 🎉 You have completed all levels. ", current_level, attempts)
     
     secret_word = new_year_levels[current_level]
     ai_response = ask_openai(user_prompt, secret_word)
@@ -63,9 +59,6 @@ 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"
@@ -83,7 +76,7 @@ with gr.Blocks() as app:
         descripion, _0, _1 = start_new_game()
         gr.Markdown(descripion)
     with gr.Row():
-        current_level_display = gr.Label(value=(current_level.value+1), label="Current Level")
+        current_level_display = gr.Textbox(value=(current_level.value+1), label="Current Level", interactive=False)
         
     user_prompt = gr.Textbox(label="Your Prompt", placeholder="Enter your prompt here...")
     game_output = gr.Textbox(label="AI Response", interactive=False)