Update app.py

app.py

@@ -5,20 +5,21 @@ import azure.cognitiveservices.speech as speechsdk
 
 dialects = {"Palestinian/Jordanian": "P", "Syrian": "S", "Lebanese": "L", "Egyptian": "E"}
 
-translator = pipeline(task="translation", model="guymorlan/English2Dialect")
+translator_en2ar = pipeline(task="translation", model="guymorlan/English2Dialect")
+translator_ar2en = pipeline(task="translation", model="guymorlan/Shami2English")
 transliterator = pipeline(task="translation", model="guymorlan/DialectTransliterator")
 
 speech_config = speechsdk.SpeechConfig(subscription=os.environ.get('SPEECH_KEY'), region=os.environ.get('SPEECH_REGION'))
 
-def translate_text(input_text):
+def translate_english(input_text):
     inputs = [f"{val} {input_text}" for val in dialects.values()]
-    result = translator (inputs)
-    
-    audio_files = []
-
-        
+    result = translator_en2ar(inputs)
     return result[0]["translation_text"], result[1]["translation_text"], result[2]["translation_text"], result[3]["translation_text"]
 
+def translate_arabic(input_text):
+    result = translator_ar2en([input_text])
+    return result[0]["translation_text"]
+
 
 def get_audio(input_text):
     audio_config = speechsdk.audio.AudioOutputConfig(filename=f"{input_text}.wav")
@@ -37,26 +38,51 @@ css = """
 #trans textarea { direction: rtl; };
 """
 
-with gr.Blocks(title = "English to Levantine Arabic", css=css) as demo:
-    with gr.Row():
-        with gr.Column():
-            input_text = gr.inputs.Textbox(label="Input", placeholder="Enter English text", lines=1)
-            gr.Examples(["I wanted to go to the store yesterday, but it rained", "How are you feeling today?", "Let's go to your place"], input_text)
-            btn = gr.Button("Translate", label="Translate")
-            gr.Markdown("Built by [Guy Mor-Lan](mailto:guy.mor@mail.huji.ac.il). Pronunciation model is specifically tailored to urban Palestinian Arabic. Text-to-speech uses Microsoft Azure's API and may provide different result from the transliterated pronunciation.")
-
-        with gr.Column():
-            pal = gr.Textbox(lines=1, label="Palestinian", elem_id="trans")
-            pal_translit = gr.Textbox(lines=1, label="Palestinian Pronunciation", elem_id="liter")
-            sy = gr.Textbox(lines=1, label="Syrian", elem_id="trans")
-            lb = gr.Textbox(lines=1, label="Lebanese", elem_id="trans")
-            eg = gr.Textbox(lines=1, label="Egyptian", elem_id="trans")
-            with gr.Row():
-                audio = gr.Audio(label="Audio - Palestinian", interactive=False)
-                audio_button = gr.Button("Get Audio", label="Get Audio")
-                audio_button.click(get_audio, inputs=[pal], outputs=[audio])
-    btn.click(translate_text, inputs=input_text, outputs=[pal, sy, lb, eg])
-    input_text.submit(translate_text, inputs=input_text, outputs=[pal, sy, lb, eg])
-    pal.change(get_transliteration, inputs=[pal], outputs=[pal_translit])
+with gr.Blocks(title = "English to Levantine Arabic", css=css, theme="default") as demo:
+    gr.Markdown("# Levantine Arabic Translator")
+    with gr.Tab('From English to Levantine Arabic'):
+        with gr.Row():
+            with gr.Column():
+                input_text = gr.Textbox(label="Input", placeholder="Enter English text", lines=1)
+                gr.Examples(["I wanted to go to the store yesterday, but it rained", "How are you feeling today?", "Let's go to your place"], input_text)
+                btn = gr.Button("Translate", label="Translate")
+                gr.Markdown("Built by [Guy Mor-Lan](mailto:guy.mor@mail.huji.ac.il). Pronunciation model is specifically tailored to urban Palestinian Arabic. Text-to-speech uses Microsoft Azure's API and may provide different result from the transliterated pronunciation.")
+
+            with gr.Column():
+                pal = gr.Textbox(lines=1, label="Palestinian", elem_id="trans")
+                pal_translit = gr.Textbox(lines=1, label="Palestinian Pronunciation", elem_id="liter")
+                sy = gr.Textbox(lines=1, label="Syrian", elem_id="trans")
+                lb = gr.Textbox(lines=1, label="Lebanese", elem_id="trans")
+                eg = gr.Textbox(lines=1, label="Egyptian", elem_id="trans")
+                with gr.Row():
+                    audio = gr.Audio(label="Audio - Palestinian", interactive=False)
+                    audio_button = gr.Button("Get Audio", label="Click Here to Get Audio")
+                    audio_button.click(get_audio, inputs=[pal], outputs=[audio])
+        btn.click(translate_english,inputs=input_text, outputs=[pal, sy, lb, eg])
+        input_text.submit(translate_english, inputs=input_text, outputs=[pal, sy, lb, eg])
+        pal.change(get_transliteration, inputs=[pal], outputs=[pal_translit])
+        
+    with gr.Tab('From Levantine Arabic to English'):
+        with gr.Row():
+            with gr.Column():
+                input_text = gr.Textbox(label="Input", placeholder="Enter Levantine Arabic text", lines=1, elem_id="trans")
+                gr.Examples(["خلينا ندور على مطعم تاني", "قديش حق البندورة؟"], input_text)
+                btn = gr.Button("Translate", label="Translate")
+                gr.Markdown("Built by [Guy Mor-Lan](mailto:guy.mor@mail.huji.ac.il).")
+            with gr.Column():
+                eng = gr.Textbox(label="English", lines=1, elem_id="liter")
+        btn.click(translate_arabic,inputs=input_text, outputs=[eng])
+        
+    with gr.Tab("Transliterate"):
+        with gr.Row():
+            with gr.Column():
+                input_text = gr.Textbox(label="Input", placeholder="Enter Levantine Arabic text", lines=1)
+                gr.Examples(["خلينا ندور على مطعم تاني", "قديش حق البندورة؟"], input_text)
+                btn = gr.Button("Transliterate", label="Transliterate")
+                gr.Markdown("Built by [Guy Mor-Lan](mailto:guy.mor@mail.huji.ac.il)")
+            with gr.Column():
+                translit = gr.Textbox(label="Transliteration", lines=1, elem_id="liter")
+        btn.click(get_transliteration, inputs=input_text, outputs=[translit])
+
 
 demo.launch()