Update app.py

app.py

@@ -1,28 +1,34 @@
 # app.py (Translation Space)
 import gradio as gr
 from transformers import pipeline
+import asyncio
 
 # Initialize models
 try:
     translator_ar_en = pipeline("translation_ar_to_en", model="Helsinki-NLP/opus-mt-ar-en")
     translator_en_ar = pipeline("translation_en_to_ar", model="Helsinki-NLP/opus-mt-en-ar")
-    print("Translation model loaded successfully!")
+    print("Translation models loaded successfully!")
 except Exception as e:
-    print(f"Error loading translation model: {e}")
-    raise
+    print(f"Error loading translation models: {e}")
+    raise  # Stop execution if models fail to load
 
-def translate(input_text, source_lang, target_lang):
+# Translation function (Batched)
+async def translate(texts, source_lang, target_lang):
     try:
+        if not texts:
+            return []  # Handle empty input
+
         if source_lang == "ar" and target_lang == "en":
-            translation = translator_ar_en(input_text)[0]['translation_text']
+            translations = translator_ar_en(texts)
+            return [translation['translation_text'] for translation in translations]
         elif source_lang == "en" and target_lang == "ar":
-            translation = translator_en_ar(input_text)[0]['translation_text']
+            translations = translator_en_ar(texts)
+            return [translation['translation_text'] for translation in translations]
         else:
-            return "Invalid language combination"
-        return translation
+            return ["Invalid language combination"] * len(texts)  # Error for each input
     except Exception as e:
         print(f"Error in translation: {e}")
-        return "Translation Error"
+        return ["Translation Error"] * len(texts)  # Indicate error for each input
 
 # Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -32,13 +38,18 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         source_lang = gr.Dropdown(["ar", "en"], label="Source Language")
         target_lang = gr.Dropdown(["ar", "en"], label="Target Language")
 
-    text_input = gr.Textbox(label="Input Text", lines=5)
+    text_input = gr.Textbox(label="Input Text (Separate multiple texts with newlines)", lines=5)
 
-    translation_output = gr.Textbox(label="Translated Text")
+    translation_output = gr.Textbox(label="Translated Text", lines=5)
 
     submit = gr.Button("Translate", variant="primary")
 
-    submit.click(translate,
+    async def process_translation(text_input, source_lang, target_lang):
+        texts = text_input.split("\n")  # Split input into multiple texts
+        translations = await translate(texts, source_lang, target_lang)
+        return "\n".join(translations)  # Join translations with newlines
+
+    submit.click(process_translation,
                  inputs=[text_input, source_lang, target_lang],
                  outputs=[translation_output])