Merge pull request #13 from OpenGPTX/flags

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 
 import core as core
-from style import CSS, T_SYMBOLS, TITLE
+from style import CSS, T_SYMBOLS, TITLE, LANG_SYMBOLS
 
 demo = gr.Blocks(css=CSS)
 with demo:
@@ -38,7 +38,7 @@ with demo:
                     )
                 with gr.Row():
                     langs_bar = gr.CheckboxGroup(
-                        choices=core.languages_list,
+                        choices=[(LANG_SYMBOLS.get(l,l),l) for l in core.languages_list],
                         value=core.languages_list,
                         label="Select languages to average over",
                         elem_id="column-select",
@@ -58,7 +58,7 @@ with demo:
 
                         def update_bar():
                             langs_bar = gr.CheckboxGroup(
-                                choices=core.languages_list,
+                                choices=[(LANG_SYMBOLS.get(l,l),l) for l in core.languages_list],
                                 value=core.languages_list,
                                 label="Select languages to average over",
                                 elem_id="column-select",

style.py

@@ -10,7 +10,33 @@ CSS = """
       display: none !important;
 }
 """
+
 T_SYMBOLS = {
     "pretrained": "🟢",
     "chat": "💬"
 }
+
+LANG_SYMBOLS = {
+    "BG": "🇧🇬 BG",
+    "CS": "🇨🇿 CZ",
+    "DA": "🇩🇰 DK",
+    "EL": "🇬🇷 EL",
+    "ET": "🇪🇪 ET",
+    "ES": "🇪🇸 ES",
+    "EN": "🇬🇧 EN",
+    "FI": "🇫🇮 FI",
+    "DE": "🇩🇪 DE",
+    "FR": "🇫🇷 FR",
+    "HU": "🇭🇺 HU",
+    "IT": "🇮🇹 IT",
+    "LT": "🇱🇹 LT",
+    "LV": "🇱🇻 LV",
+    "NL": "🇳🇱 NL",
+    "PL": "🇵🇱 PL",
+    "PT-PT": "🇵🇹 PT",
+    "RO": "🇷🇴 RO",
+    "SK": "🇸🇰 SK",
+    "SL": "🇸🇮 SL",
+    "SV": "🇸🇪 SV"
+}
+