Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
-from gradio import Interface
-from gradio.components import Textbox, Dropdown
+from gradio import Interface, inputs, outputs
 
 import spacy
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
@@ -16,14 +15,14 @@ pos_map = {
     'artículo': 'DET'
 }
 
-def identify_pos(sentence):
+def identify_pos(sentence: str):
     doc = nlp(sentence)
     pos_tags = [(token.text, token.pos_) for token in doc]
     return pos_tags
 
-def game_logic(sentence, user_word, user_pos):
+def game_logic(sentence: str, user_word: str, user_pos: str):
     correct_answers = identify_pos(sentence)
-    user_pos = pos_map[user_pos.lower()]
+    user_pos = pos_map.get(user_pos.lower(), '')
     for word, pos in correct_answers:
         if word == user_word:
             if pos.lower() == user_pos.lower():
@@ -32,13 +31,13 @@ def game_logic(sentence, user_word, user_pos):
                 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 generate_hint(sentence, user_word, user_pos):
+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')
     outputs = model.generate(inputs, max_length=150, num_return_sequences=1, no_repeat_ngram_size=2)
     hint = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return hint
 
-def main(sentence, user_word, user_pos):
+def main(sentence: str, user_word: str, user_pos: str):
     if sentence and user_word and user_pos and user_pos != 'Selecciona una función gramatical...':
         correct, message = game_logic(sentence, user_word, user_pos)
         hint = generate_hint(sentence, user_word, user_pos)
@@ -48,9 +47,9 @@ def main(sentence, user_word, user_pos):
 
 iface = Interface(fn=main, 
                   inputs=[
-                      Textbox(lines=2, placeholder='Introduce una frase aquí...'),
-                      Textbox(lines=1, placeholder='Introduce una palabra aquí...'),
-                      Dropdown(choices=['Selecciona una función gramatical...', 'sustantivo', 'verbo', 'adjetivo', 'artículo'])
+                      inputs.Textbox(lines=2, placeholder='Introduce una frase aquí...'),
+                      inputs.Textbox(lines=1, placeholder='Introduce una palabra aquí...'),
+                      inputs.Dropdown(choices=['Selecciona una función gramatical...', 'sustantivo', 'verbo', 'adjetivo', 'artículo'])
                   ], 
-                  outputs=[Textbox(), Textbox()])
+                  outputs=[outputs.Textbox(), outputs.Textbox()])
 iface.launch()