Merlintxu commited on
Commit
d8db9a1
·
1 Parent(s): 4af357b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -18
app.py CHANGED
@@ -8,37 +8,32 @@ text_generator = pipeline('text-generation', model='gpt2')
8
 
9
  pos_tags = ['ADJ', 'ADP', 'ADV', 'AUX', 'CONJ', 'DET', 'INTJ', 'NOUN', 'NUM', 'PART', 'PRON', 'PROPN', 'PUNCT', 'SCONJ', 'SYM', 'VERB', 'X']
10
 
11
- sentence_state = {'sentence': '', 'tagged_words': []}
12
-
13
  def generate_sentence():
14
  result = text_generator('')[0]
15
  sentence = result['generated_text']
16
  tagged_words = analyze_sentence(sentence)
17
- sentence_state['sentence'] = sentence
18
- sentence_state['tagged_words'] = tagged_words
19
  return sentence, [word for word, _ in tagged_words]
20
 
21
  def analyze_sentence(sentence):
22
  doc = nlp(sentence)
23
  return [(token.text, token.pos_) for token in doc]
24
 
25
- def check_answer(*args):
26
- correct_answer = [tag for word, tag in sentence_state['tagged_words']]
 
27
  user_answer = list(args)
28
  if user_answer == correct_answer:
29
  return 'Correcto!'
30
  else:
31
  return 'Incorrecto. La respuesta correcta es: ' + str(correct_answer)
32
 
33
- def game_flow():
34
- sentence, words = generate_sentence()
35
- answer = check_answer(*gr.inputs)
36
- return sentence, words, answer
37
-
38
- iface = gr.Interface(fn=game_flow,
39
- inputs=[gr.inputs.Action(label='Generate Sentence')] +
40
- [gr.inputs.Dropdown(choices=pos_tags, label=f'Word {i+1}') for i in range(len(sentence_state['tagged_words']))],
41
- outputs=[gr.outputs.Textbox(label='Sentence'),
42
- gr.outputs.Textbox(label='Words'),
43
- gr.outputs.Textbox(label='Result')])
44
- iface.launch()
 
8
 
9
  pos_tags = ['ADJ', 'ADP', 'ADV', 'AUX', 'CONJ', 'DET', 'INTJ', 'NOUN', 'NUM', 'PART', 'PRON', 'PROPN', 'PUNCT', 'SCONJ', 'SYM', 'VERB', 'X']
10
 
 
 
11
  def generate_sentence():
12
  result = text_generator('')[0]
13
  sentence = result['generated_text']
14
  tagged_words = analyze_sentence(sentence)
 
 
15
  return sentence, [word for word, _ in tagged_words]
16
 
17
  def analyze_sentence(sentence):
18
  doc = nlp(sentence)
19
  return [(token.text, token.pos_) for token in doc]
20
 
21
+ def check_answer(sentence, *args):
22
+ tagged_words = analyze_sentence(sentence)
23
+ correct_answer = [tag for word, tag in tagged_words]
24
  user_answer = list(args)
25
  if user_answer == correct_answer:
26
  return 'Correcto!'
27
  else:
28
  return 'Incorrecto. La respuesta correcta es: ' + str(correct_answer)
29
 
30
+ generate_iface = gr.Interface(fn=generate_sentence,
31
+ inputs='number',
32
+ outputs=['textbox', 'textbox'])
33
+
34
+ check_iface = gr.Interface(fn=check_answer,
35
+ inputs=['textbox'] + [gr.inputs.Dropdown(choices=pos_tags) for _ in range(10)],
36
+ outputs='textbox')
37
+
38
+ generate_iface.launch()
39
+ check_iface.launch()