import os
from transformers import pipeline
import gradio as gr


hf_writer = gr.HuggingFaceDatasetSaver(os.getenv('HF_API_TOKEN'), "suicide-comment-es-space-human-feedback")
pipe = pipeline("text-classification", model='hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es')


def detect_suicide_comment(input):
    result = pipe(input)[0]
    # return {result['label']: result['score']}
    return result


iface = gr.Interface(
    fn=detect_suicide_comment,
    inputs="text",
    outputs="text",  # gr.Label(num_top_classes=1, type="confidences"),
    allow_flagging="manual",
    flagging_options=['Correct Classification', 'Incorrect Classification'],
    flagging_callback=hf_writer,
    examples=[
        "Quiero acabar con todo. No merece la pena vivir.",
        "El partido de fútbol fue igualado, todos disfrutamos mucho jugando juntos."
    ],
    article="""
## Motivación

* Cada año se suicidan casi 1.000.000 de personas en el mundo y por cada suicidio, se cometen unos 20 intentos.
* Cada año se suicidan unas 4000 personas en España, siendo la primera causa de muerte externa. Esto son unas 11 muertes al día.
* Entre jóvenes de 15 a 29 años de edad, el suicidio es la 2ª causa principal de muerte a nivel mundial.
* Cada suicidio es una tragedia. Los efectos sobre las familias, amigos y comunidades son terribles y de amplio alcance. Por cada suicidio consumado, al menos 6 seres queridos se ven directamente afectados.

## Demo

Introduce cualquier frase en español, un modelo de inteligencia artificial tratará de predecir si la frase contiene algún patrón indicativo de ideación o comportamiento suicida.

## Feedback

¡Ayúdanos a mejorar la aplicación marcando la respuesta como Correcta o Incorrecta según corresponda!

Nosotros usaremos este feedback para ampliar el dataset que se usará para fine-tunear el modelo en versiones posteriores.

¡Gracias!

## Dataset

https://huggingface.co/datasets/hackathon-somos-nlp-2023/suicide-comments-es

## Model

https://huggingface.co/hackathon-somos-nlp-2023/roberta-base-bne-finetuned-suicide-es

## Team

* [dariolopez](https://huggingface.co/dariolopez)
* [diegogd](https://huggingface.co/diegogd)

""",
    title="Detector de comentarios con ideación suicida"
)
iface.launch()