Adapted demo to gr.Blocks()

app.py

@@ -8,6 +8,11 @@ LANGUAGES_LIST = list(LANGUAGES.keys())
 HF_TOKEN = os.environ.get('HF_TOKEN')
 
 def translate_wrapper(text, src_lang, tgt_lang, by_sentence=True, clean=True, num_beams=4):
+    if src_lang in ["", None, []] or tgt_lang in ["", None, []]:
+        return "Ընտրեցէք թարգմանութեան կողմերը | Please select source and target languages"
+
+    if src_lang == tgt_lang:
+        return "Ընտրուած լեզուները նոյնն են։ | Source and target languages are identical."
 
     src_lang = LANGUAGES.get(src_lang)
     tgt_lang = LANGUAGES.get(tgt_lang)
@@ -17,28 +22,61 @@ def translate_wrapper(text, src_lang, tgt_lang, by_sentence=True, clean=True, nu
 
 
 hf_writer = gr.HuggingFaceDatasetSaver(HF_TOKEN, "AriNubar/hyw-en-crowd-source")
-desc = "<p>Հոս կը ցուցադրուի առաջին արեւմտահայերէն-անգլերէն մեքենական թարգմանիչը, որ կարուցուած է Մեթայի (Ֆեյսպուքի) 'No Language Left Behind' տիպարի հիման վրայ։ Թարգմանութեան տիպարը կ'աշխատի CPU-ի մը մէջ, ուրեմն նախադասութեան մը թարգմանութիւնը կրնայ տեւել մօտաւորապէս 40-60 երկվայրկեան։ Ձեր գնահատութիւնները եւ քննադատութիւնները շատ կարեւոր են մեզի համար, որոնց հետ տիպարի թարգմանութեան որակը կրնայ բարելաւուիլ։ Շնորհակալութիւն Գալուստ Կիւլպէնկեան Հիմնադրամին, որուն աջակցութեամբ այս նախագիծը իրականացուեցաւ։</p><p></p><p>This is the demo of the first Western Armenian-English neural machine translation system which is based on Meta's 'No Language Left Behind' model. The model runs on a CPU, so it might take approximately 40-60 seconds to translate a single sentence. Your feedback and comments are very important for us to improve the quality of the translation. Thanks to the Calouste Gulbenkian Foundation for making this project possible with their support.</p>"
-interface = gr.Interface(
-    translate_wrapper,
-    [
-        
-        gr.Textbox(lines=5, label="Ներմուծման Գրութիւն | Input Text"),
-        gr.Dropdown(LANGUAGES_LIST, type="value", label="Թարգմանէ Այս Լեզուէ | Source Language"),
-        gr.Dropdown(LANGUAGES_LIST, type="value", label="Թարգմանէ Այս Լեզուի | Target Language"),
-        gr.Checkbox(label="Նախադասութիւններու Բաժնէ | Split into Sentences", value=True, info="Տուփը նշանագրեցէք եթէ կ'ուզէք ձեր գրութիւնը թարգմանուի նախադասութիւն առ նախադասութիւն։ Այս կերպով թարգմանուած նախադասութիւններուն որակը աւելի լաւ են։ | Check this box if you want to split your input text into sentences. This way the quality of the translation will be better."),
-        gr.Checkbox(label="Մշակէ | Preprocess", value=True, info="Տուփը նշանագրեցէք եթէ կ'ուզէք ձեր գրութիւնը կանոնաւորուի ծրագրի կողմէ թարգմանութենէ առաջ։ Կանոնաւորումը թարգմանութեան որակի բարելաւման համար օգտակար է։ | Check this box if you want to preprocess your input text before translation. This way the quality of the translation will be better."),
-        gr.Dropdown([1, 2, 3, 4, 5], type="value", label="Առնջութեան Շողեր | Number of Beams", value=4, info="Բարձր թիւը ընդհանրապէս կը պատճառէ աւելի բարձր որակի, բայց նոյնիսկ երկարատեւ թարգմանութեան։ | Higher beam size will result in better quality translation, but also longer translation time."),
-    ],  
-    "text",
-    title="Արեւմտահայերէն-Անգլերէն Մեքենական Թարգմանիչ | Western Armenian-English Machine Translation",
-    article= desc,
-    allow_flagging="manual",
-    flagging_callback=hf_writer,
-    flagging_options=["Լաւ թարգմանութիւն | Good Translation", "Միջակ թարգմանութիւն | Average Translation", "Վատ թարգմանութիւն | Bad Translation"]
+
+
+theme = gr.themes.Default().set(
+    block_info_text_size="*text_xxs" # for info text
 )
 
-if __name__ == "__main__":
-    translator = Translator()
-    interface.launch()
+with gr.Blocks(title="Արեւմտահայերէն-Անգլերէն Մեքենական Թարգմանիչ | Western Armenian-English Machine Translation", 
+               theme=theme,
+               ) as demo:
+    gr.Markdown(
+    """
+    # Արեւմտահայերէն-Անգլերէն Մեքենական Թարգմանիչ | Western Armenian-English Machine Translation
+
+    Հոս կը ցուցադրուի առաջին արեւմտահայերէն-անգլերէն մեքենական թարգմանիչը, որ կարուցուած է Մեթայի (Ֆեյսպուքի) 'No Language Left Behind' տիպարի հիման վրայ։ Թարգմանութեան տիպարը կ'աշխատի CPU-ի մը մէջ, ուրեմն նախադասութեան մը թարգմանութիւնը կրնայ տեւել մօտաւորապէս 40-60 երկվայրկեան։ Ձեր գնահատութիւնները եւ քննադատութիւնները շատ կարեւոր են տիպարի թարգմանութեան որակը բարելաւելու համար։ Շնորհակալութիւն Գալուստ Կիւլպէնկեան Հիմնադրամին, որուն աջակցութեամբ այս նախագիծը իրականացուեցաւ։    
+
+    This is the demo of the first Western Armenian-English neural machine translation system which is based on Meta's 'No Language Left Behind' model. The model runs on a CPU, so it might take approximately 40-60 seconds to translate a single sentence. Your feedback and comments are very important for us to improve the quality of the translation. Thanks to the Calouste Gulbenkian Foundation for making this project possible with their support.
+    """
+    )
 
+    with gr.Row():
+        with gr.Column():
+            text = gr.Textbox(lines=5, label="Ներմուծման Գրութիւն | Input Text")
+            with gr.Row():
+                src_lang = gr.Dropdown(LANGUAGES_LIST, type="value", label="Թարգմանէ Այս Լեզուէ | Source Language")
+                tgt_lang = gr.Dropdown(LANGUAGES_LIST, type="value", label="Թարգմանէ Այս Լեզուի | Target Language")
+        with gr.Column():
+            translated = gr.Textbox(lines=5, label="Ելքային Գրութիւն | Output Text", interactive=False)
+            translate_btn = gr.Button(value="Թարգմանէ | Translate", variant="primary")
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("""
+                    ### Թարգմանութեան Որակ | Translation Quality
+                    """)                
+                    flag_good_btn = gr.Button(value="😊 Լաւ | Good", size="sm")
+                    flag_average_btn = gr.Button(value="😐 Միջակ | Average", size="sm")
+                    flag_bad_btn = gr.Button(value="☹️ Վատ | Bad", size="sm")
+    with gr.Row():
+        gr.Markdown("""
+            ## Յարաչափեր | Parameters
+        """
+        )
+        by_sentence = gr.Checkbox(label="Նախադասութիւններու Բաժնէ | Split into Sentences", value=True, info="Տուփը նշանագրեցէք եթէ կ'ուզէք ձեր գրութիւնը թարգմանուի նախադասութիւն առ նախադասութիւն։ Այս կերպով թարգմանուած նախադասութիւններուն որակը աւելի լաւ են։ | Check this box if you want to split your input text into sentences. This way the quality of the translation will be better.")
+        clean = gr.Checkbox(label="Մշակէ | Preprocess", value=True, info="Տուփը նշանագրեցէք եթէ կ'ուզէք ձեր գրութիւնը կանոնաւորուի ծրագրի կողմէ թարգմանութենէ առաջ։ Կանոնաւորումը թարգմանութեան որակի բարելաւման համար օգտակար է։ | Check this box if you want to preprocess your input text before translation. This way the quality of the translation will be better.")
+        num_beams = gr.Dropdown([1, 2, 3, 4, 5], type="value", label="Առնջութեան Շողեր | Number of Beams", value=4, info="Աւելի բարձր թիւը ընդհանրապէս կը պատճառէ աւելի բարձր որակի, բայց նոյնիսկ երկարատեւ թարգմանութեան։ | Higher beam size will result in better quality translation, but also longer translation time.")
+
+    translate_btn.click(translate_wrapper, inputs=[text, src_lang, tgt_lang, by_sentence, clean, num_beams], outputs=translated)
+    
+    hf_writer.setup([src_lang, tgt_lang, text, translated, by_sentence, clean, num_beams, flag_bad_btn], "flagged_data_points")
     
+    flag_good_btn.click(lambda *args: hf_writer.flag(args), [src_lang, tgt_lang, text, translated, by_sentence, clean, num_beams, flag_good_btn], None, preprocess=False)
+    flag_average_btn.click(lambda *args: hf_writer.flag(args), [src_lang, tgt_lang, text, translated, by_sentence, clean, num_beams, flag_average_btn], None, preprocess=False)
+    flag_bad_btn.click(lambda *args: hf_writer.flag(args), [src_lang, tgt_lang, text, translated, by_sentence, clean, num_beams, flag_bad_btn], None, preprocess=False)
+
+
+
+if __name__ == "__main__":
+    translator = Translator()
+    demo.launch()