Update app.py

app.py

@@ -18,8 +18,10 @@ def reset_game():
     # Ask LLM for the first emoji puzzle
     emoji_prompt = "Give me an emoji-based puzzle for a word or phrase guessing game."
     response = pipe([{"role": "user", "content": emoji_prompt}], max_new_tokens=30)
-    current_emoji = response[0]['generated_text'].strip()
     
+    # Ensure to get the generated text correctly
+    current_emoji = response[0]['generated_text'] if response and isinstance(response, list) and len(response) > 0 else "❓"
+
     # Return introduction and the emoji puzzle
     return introduction + f"Here's your first puzzle: {current_emoji}"
 
@@ -47,10 +49,12 @@ def emoji_game(user_input=""):
     # Check if the user made a guess; LLM evaluates the guess
     guess_prompt = f"User guessed '{user_input}'. Was the guess correct for this emoji puzzle: {current_emoji}?"
     response = pipe([{"role": "user", "content": guess_prompt}], max_new_tokens=30)
-    response_text = response[0]['generated_text'].strip().lower()
     
+    # Ensure to get the generated text correctly
+    response_text = response[0]['generated_text'] if response and isinstance(response, list) and len(response) > 0 else "❓"
+
     # Process response: if it's correct, increase points; otherwise, decrease points
-    if "correct" in response_text:  # Correct guess
+    if "correct" in response_text.lower():  # Correct guess
         points += 1
         if points >= 10:
             return f"🎉 Correct! You've reached 10 points! You win the game! 🏆"
@@ -58,7 +62,9 @@ def emoji_game(user_input=""):
             # Get next emoji puzzle for the user
             next_emoji_prompt = "Give me another emoji-based puzzle."
             response = pipe([{"role": "user", "content": next_emoji_prompt}], max_new_tokens=30)
-            current_emoji = response[0]['generated_text'].strip()
+            
+            # Ensure to get the generated text correctly
+            current_emoji = response[0]['generated_text'] if response and isinstance(response, list) and len(response) > 0 else "❓"
             return f"✅ Correct! Your current points: {points}. Here's your next puzzle: {current_emoji}"
     else:  # Incorrect guess
         points -= 1