import gradio as gr
import spacy
from kommunicate import Kommunicate

# Load the Spanish model for spaCy
nlp = spacy.load('es_core_news_sm')

# Map the Spanish POS tags to their English equivalents
pos_map = {
    'sustantivo': 'NOUN',
    'verbo': 'VERB',
    'adjetivo': 'ADJ',
    'artículo': 'DET'
}

# Function to identify the POS tags in a sentence
def identify_pos(sentence):
    doc = nlp(sentence)
    pos_tags = [(token.text, token.pos_) for token in doc]
    return pos_tags

# Game logic
def game_logic(sentence, user_word, user_pos):
    correct_answers = identify_pos(sentence)
    user_pos = pos_map[user_pos.lower()]
    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.'

# Main function for the Gradio interface
def main(sentence, user_word, user_pos):
    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)
        return message
    else:
        return 'Por favor, introduce una frase, una palabra y selecciona una función gramatical válida (sustantivo, verbo, adjetivo, artículo).'

# Create the Gradio interface
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=['Selecciona una función gramatical...', 'sustantivo', 'verbo', 'adjetivo', 'artículo'])
                  ], 
                  outputs=gr.outputs.Textbox())
iface.launch()