add quality feedback

.gitignore

@@ -1 +1,2 @@
 .idea
+flagged/

app.py

@@ -12,6 +12,13 @@ MODEL_URL_MUL_MYV = 'slone/mbart-large-51-mul-myv-v1'
 MODEL_URL_LANGID = 'https://huggingface.co/slone/fastText-LID-323/resolve/main/lid.323.ftz'
 MODEL_PATH_LANGID = 'lid.323.ftz'
 
+HF_TOKEN = os.getenv('HF_TOKEN')
+hf_writer = gr.HuggingFaceDatasetSaver(
+    HF_TOKEN,
+    dataset_name="myv-translation-2022-demo-flags",
+    private=True,
+)
+
 
 lang_to_code = {
     'Эрзянь | Erzya': 'myv_XX',
@@ -136,6 +143,19 @@ def translate_wrapper(text, src, trg):
     return result
 
 
+article = """
+Те васенце автоматической ютавтыця ютавтыця эрзянь кельсэ. Тонакстнэ улить статьясо (курок сы).
+
+Это первый автоматический переводчик для эрзянского языка. Подробности – в статье (скоро выйдет). 
+Пожалуйста, оставляйте своё мнение о качестве переводов с помощью кнопок с эмодзи!
+
+This is the first automatic translator for the Erzya language. The details are in the paper (will be published soon).
+Please leave your feedback about the quality of translations using the buttons with emojis.
+
+The code and models for translation can be found in the repository: https://github.com/slone-nlp/myv-nmt. 
+"""
+
+
 interface = gr.Interface(
     translate_wrapper,
     [
@@ -144,6 +164,11 @@ interface = gr.Interface(
         gr.Dropdown(list(lang_to_code.keys()), type="value", label='target language / эряви келесь', value=list(lang_to_code.keys())[1]),
     ],
     "text",
+    allow_flagging="manual",
+    flagging_options=["good 🙂", "50/50 😐", "bad 🙁"],
+    flagging_callback=hf_writer,
+    title='Эрзянь ютавтыця | Эрзянский переводчик | Erzya translator',
+    article=article,
 )