Update app.py

app.py

@@ -20,28 +20,26 @@ def print_results(results):
         result_string += pprint.pformat(hit, indent=4) + "\n"
     return result_string.strip()
 
-async def translate_bulk_parallel(bulk: list) -> list:
+async def translate_bulk(bulk: list) -> list:
     """
-    Translate the given text to English in parallel and return the translated text.
+    Translate the given text to English and return the translated text.
 
     Args:
-    - bulk (list): The list of texts to translate.
+    - text (str): The text to translate.
 
     Returns:
-    - list: The translated texts.
+    - str: The translated text.
     """
-    translated_bulk = []
-    with concurrent.futures.ThreadPoolExecutor() as executor:
-        # Translate each text in parallel
-        futures = [executor.submit(translator.translate, text, dest="en") for text in bulk]
-        for future, original_text  in zip(futures, bulk):
-            try:
-                translated_text = await future
-                translated_text = translated_text.text.lower().strip()
-                translated_bulk.append(translated_text)
-            except Exception as e:
-                print(f"Translation failed: {e}")
-                translated_bulk.append(original_text.lower().strip())  # Add placeholder for failed translation
+    try:
+        translated_bulk = await translator.translate(bulk, dest="en")
+        translated_bulk = [
+            translated_text.text.lower().strip() for translated_text in translated_bulk
+        ]
+    except Exception as e:
+        print(f"Bulk Translation failed: {e}")
+        translated_bulk = [
+            text.lower().strip() for text in bulk
+        ]  # Use original text if translation fails
     return translated_bulk
 
 async def encode_document(document: str):
@@ -103,7 +101,7 @@ async def predict(query):
 
     try:
         translation_start_time = time.time()
-        representation_list = await translate_bulk_parallel(tasks)
+        representation_list = await translate_bulk(tasks)
     except Exception as e:
         representation_list = tasks
         print(f"An error occurred while translating: {e}")