Update app.py

app.py

@@ -2,46 +2,42 @@
 import gradio as gr
 import spacy
 from transformers import pipeline
-from gradio import Interface
-from gradio.components import Textbox, Dropdown
 
 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']
 
+# Store the sentence state globally
 sentence_state = {'sentence': '', 'tagged_words': []}
 
-def generate_sentence():
-    result = text_generator('', max_length=10)[0]  # Limiting max_length for simplicity
+# Generate a sentence and analyze it
+def generate_and_analyze():
+    result = text_generator('', max_length=10, do_sample=True, language='es')[0]
     sentence = result['generated_text']
-    tagged_words = analyze_sentence(sentence)
+    doc = nlp(sentence)
+    tagged_words = [(token.text, token.pos_) for token in doc]
     sentence_state['sentence'] = sentence
     sentence_state['tagged_words'] = tagged_words
-    return sentence, [word for word, _ in tagged_words]
-
-def analyze_sentence(sentence):
-    doc = nlp(sentence)
-    return [(token.text, token.pos_) for token in doc]
+    return sentence, tagged_words
 
-def check_answer(*args):
-    correct_answer = [tag for word, tag in sentence_state['tagged_words']]
-    user_answer = list(args)
-    if user_answer == correct_answer:
-        return 'Correcto!'
+# The game logic
+def game_flow(start_game, *args):
+    if start_game == 'Empezar':
+        sentence, _ = generate_and_analyze()
+        return sentence, '', ''
     else:
-        return 'Incorrecto. La respuesta correcta es: ' + str(correct_answer)
-
-def game_flow(start_game):
-    if start_game == 'Start':
-        sentence, words = generate_sentence()
-        answer = check_answer(*gr.inputs)
-        return sentence, words, answer
-
-iface = Interface(fn=game_flow, 
-                     inputs=[Textbox(label='Type "Start" to generate sentence')] + 
-                            [Dropdown(choices=pos_tags, label=f'Word {i+1}') for i in range(10)],  # Assumes sentences of 10 words
-                     outputs=[Textbox(label='Sentence'),
-                              Textbox(label='Words'),
-                              Textbox(label='Result')])
-iface.launch()
+        correct_answer = [tag for _, tag in sentence_state['tagged_words']]
+        user_answer = list(args)
+        if user_answer == correct_answer:
+            return sentence_state['sentence'], ' '.join(sentence_state['tagged_words']), '¡Correcto!'
+        else:
+            return sentence_state['sentence'], ' '.join(sentence_state['tagged_words']), 'Incorrecto. La respuesta correcta es: ' + ' '.join(correct_answer)
+
+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')])
+iface.launch()