import gradio as gr
from simple_nlp_library import embeddings, preprocessing, metrics

vectors = embeddings.vectors()
stop_words = preprocessing.stop_words()


def semantic_similarity(first_sentence, second_sentence):
    return metrics.cosine_similarity(
        embeddings.tokens_vector(vectors, preprocessing.semantic_tokens(stop_words, first_sentence)),
        embeddings.tokens_vector(vectors, preprocessing.semantic_tokens(stop_words, second_sentence)),
    )


gr.Interface(
    semantic_similarity,
    inputs=["text", "text"],
    outputs=["number"],
).launch()