Update app.py

app.py

@@ -7,9 +7,10 @@ nlp = spacy.load('es_core_news_sm')
 text_generator = pipeline('text-generation', model='gpt2')
 
 pos_tags = ['ADJ', 'ADP', 'ADV', 'AUX', 'CONJ', 'DET', 'INTJ', 'NOUN', 'NUM', 'PART', 'PRON', 'PROPN', 'PUNCT', 'SCONJ', 'SYM', 'VERB', 'X']
+pos_tags_es = ['ADJ', 'PREP', 'ADV', 'AUX', 'CONJ', 'DET', 'INTJ', 'SUST', 'NUM', 'PART', 'PRON', 'PROPN', 'PUNCT', 'SCONJ', 'SYM', 'VERBO', 'X']
 
 # Store the sentence state globally
-sentence_state = {'sentence': '', 'tagged_words': []}
+sentence_state = {'sentence': '', 'tagged_words': [], 'tagged_words_es': []}
 
 # Generate a sentence and analyze it
 def generate_and_analyze():
@@ -17,27 +18,31 @@ def generate_and_analyze():
     sentence = result['generated_text']
     doc = nlp(sentence)
     tagged_words = [(token.text, token.pos_) for token in doc]
+    tagged_words_es = [(token.text, pos_tags_es[pos_tags.index(token.pos_)]) for token in doc]
     sentence_state['sentence'] = sentence
     sentence_state['tagged_words'] = tagged_words
-    return sentence, tagged_words
+    sentence_state['tagged_words_es'] = tagged_words_es
+    return sentence, tagged_words, tagged_words_es
 
 # The game logic
 def game_flow(start_game, *args):
     if start_game == 'Empezar':
-        sentence, _ = generate_and_analyze()
-        return sentence, '', ''
+        sentence, _, _ = generate_and_analyze()
+        return sentence, '', '', ''
     else:
-        correct_answer = [tag for _, tag in sentence_state['tagged_words']]
+        correct_answer = [tag for _, tag in sentence_state['tagged_words_es']]
         user_answer = list(args)
         if user_answer == correct_answer:
-            return sentence_state['sentence'], ' '.join(sentence_state['tagged_words']), '¡Correcto!'
+            return sentence_state['sentence'], ' '.join(sentence_state['tagged_words_es']), '¡Correcto!', ''
         else:
-            return sentence_state['sentence'], ' '.join(sentence_state['tagged_words']), 'Incorrecto. La respuesta correcta es: ' + ' '.join(correct_answer)
+            correction = ' '.join([f'{word} es {tag}' for word, tag in sentence_state['tagged_words_es']])
+            return sentence_state['sentence'], ' '.join(sentence_state['tagged_words_es']), 'Incorrecto.', 'La respuesta correcta es: ' + correction
 
 iface = gr.Interface(fn=game_flow, 
-                     inputs=[gr.inputs.Textbox(label='Escribe "Empezar" para generar una frase')] + 
-                            [gr.inputs.Dropdown(choices=pos_tags, label=f'Palabra {i+1}') for i in range(5)], 
-                     outputs=[gr.outputs.Textbox(label='Frase'),
-                              gr.outputs.Textbox(label='Palabras'),
-                              gr.outputs.Textbox(label='Resultado')])
+                     inputs=[gr.inputs.Textbox(label='Escribe "Empezar" para iniciar el juego')] + 
+                            [gr.inputs.Dropdown(choices=pos_tags_es, label=f'Función de la palabra {i+1}') for i in range(5)], 
+                     outputs=[gr.outputs.Textbox(label='Frase generada'),
+                              gr.outputs.Textbox(label='Funciones de las palabras'),
+                              gr.outputs.Textbox(label='Evaluación'),
+                              gr.outputs.Textbox(label='Corrección')])
 iface.launch()