Added support for translation direction

app.py

@@ -6,31 +6,39 @@ from translator import Translator
 import gradio as gr
 import os
 
-
 # Load the API key from an environment variable or a file
 fanar_api_key = os.getenv("FANAR_API_KEY")
-langpair = "en-ar" # also pass this as an option to the interface. The user should be able to choose the language pair
-
-translator = Translator(api_key=fanar_api_key, langpair=langpair, model="Fanar-Shaheen-MT-1")
+translator_en_to_ar = Translator(api_key=fanar_api_key, langpair="en-ar", model="Fanar-Shaheen-MT-1")
+translator_ar_to_en = Translator(api_key=fanar_api_key, langpair="ar-en", model="Fanar-Shaheen-MT-1")
 
-def translate_text(text): 
-    return translator.translate(text)
+def translate_text(text, direction): 
+    if direction == "English to Arabic":
+        return translator_en_to_ar.translate(text)
+    else:
+        return translator_ar_to_en.translate(text)
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Fanar Translator")
-    gr.Markdown("## Translate text into Arabic Using the State of the Art Fanar Shaheen MT Model") 
+    gr.Markdown("# 🌍 Fanar Translation Demo")
+    gr.Markdown("## Select the translation direction, enter your text, and get the translation !") 
+    
+    direction_dropdown = gr.Dropdown(
+        choices = ["English to Arabic", "Arabic to English"], 
+        label = "Translation Direction",
+        value = "English to Arabic",
+        interactive=True,
+    )
+    
+    input_box = gr.Textbox(label="Input Text", placeholder="Enter text to translate here...", lines=4)
+    output_box = gr.Textbox(label="Output Text", placeholder="Translated text will appear here...", lines=4, interactive=False)
     
-    with gr.Row(): 
-        input_box = gr.Textbox(label="Input Text", placeholder="Enter text to translate here...", lines=4)
-    with gr.Row(): 
-        output_box = gr.Textbox(label="Output Text", placeholder="Translated text will appear here...", lines=4, interactive=False)
-        
     translate_button = gr.Button("Translate")
-    translate_button.click(translate_text, inputs=input_box, outputs=output_box)
     
-
-
-
+    translate_button.click(
+        fn=translate_text, 
+        inputs=[input_box, direction_dropdown],
+        outputs=output_box,
+    )
+    
 demo.launch()