Update app.py

app.py

@@ -8,23 +8,22 @@ def identify_pos(sentence):
     pos_tags = [(token.text, token.pos_) for token in doc]
     return pos_tags
 
-def game_logic(sentence, user_input):
+def game_logic(sentence, user_word, user_pos):
     correct_answers = identify_pos(sentence)
-    words = [word for word, pos in correct_answers]
-    if user_input in words:
-        return True
-    else:
-        return False
-
+    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.'
 
-def main(sentence, user_input):
-    if sentence and user_input:
-        if game_logic(sentence, user_input):
-            return '¡Correcto!'
-        else:
-            return 'Incorrecto. Intenta de nuevo.'
+def main(sentence, user_word, user_pos):
+    if sentence and user_word and user_pos:
+        correct, message = game_logic(sentence, user_word, user_pos)
+        return message
     else:
-        return 'Por favor, introduce una frase y una palabra que creas que es un sustantivo, verbo, adjetivo o artículo.'
+        return 'Por favor, introduce una frase, una palabra y su función gramatical (sustantivo, verbo, adjetivo, artículo).'
 
-iface = gr.Interface(fn=main, inputs=["text", "text"], outputs="text")
+iface = gr.Interface(fn=main, inputs=["text", "text", "text"], outputs="text")
 iface.launch()