GramAPP / app.py
Merlintxu's picture
Update app.py
7828263
raw
history blame
1.48 kB
import gradio as gr
import spacy
nlp = spacy.load('es_core_news_sm')
def identify_pos(sentence):
doc = nlp(sentence)
pos_tags = [(token.text, token.pos_) for token in doc]
return pos_tags
def game_logic(sentence, user_word, user_pos):
correct_answers = identify_pos(sentence)
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_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, una palabra y su función gramatical (sustantivo, verbo, adjetivo, artículo).'
iface = gr.Interface(fn=main,
inputs=[
gr.inputs.Textbox(lines=2, placeholder='Introduce una frase aquí...'),
gr.inputs.Textbox(lines=1, placeholder='Introduce una palabra aquí...'),
gr.inputs.Dropdown(choices=['sustantivo', 'verbo', 'adjetivo', 'artículo'], placeholder='Selecciona una función gramatical...')
],
outputs="text")
iface.launch()