Update app.py

app.py

@@ -23,13 +23,14 @@ def identify_pos(sentence: str):
 def game_logic(sentence: str, user_word: str, user_pos: str):
     correct_answers = identify_pos(sentence)
     user_pos = pos_map.get(user_pos.lower(), '')
-    for word, pos in correct_answers:
-        if word == user_word:
-            if pos.lower() == user_pos.lower():
-                return True, f'¡Correcto! "{user_word}" es un {user_pos}.'
-            else:
-                return False, f'Incorrecto. "{user_word}" no es un {user_pos}, es un {pos}.'
-    return False, f'La palabra "{user_word}" no se encuentra en la frase.'
+    word_pos_list = [word_pos for word_pos in correct_answers if word_pos[0] == user_word]
+    if not word_pos_list:
+        return False, f'La palabra "{user_word}" no se encuentra en la frase.'
+    for word, pos in word_pos_list:
+        if pos.lower() == user_pos.lower():
+            return True, f'¡Correcto! "{user_word}" es un {user_pos}.'
+    return False, f'Incorrecto. "{user_word}" no es un {user_pos}, puede ser {", ".join(set(pos for word, pos in word_pos_list))}.'
+
 
 def generate_hint(sentence: str, user_word: str, user_pos: str):
     inputs = tokenizer.encode(f'Give a hint about the {user_pos} in the sentence "{sentence}"', return_tensors='pt')