Added translator

Added translator to translate the text to get accurate output

app.py

@@ -3,10 +3,21 @@ from transformers import pipeline
 import openpyxl
 import tempfile
 import pycountry
+from deep_translator import GoogleTranslator
+from langdetect import detect
 
 classifier = pipeline("zero-shot-classification",
                       model="LogicSpine/address-large-text-classifier")
 
+def translate_text(text: str):
+    text = text.strip()
+    translator = GoogleTranslator(source='auto', target='en')
+    detected_lang = detect(text)
+    if detected_lang == 'en':
+        return text
+    translated = translator.translate(text)
+    return translated
+
 def check_for_third(address: str) -> bool:
     countries = [country.name.lower() for country in pycountry.countries]
     old_country_names = [
@@ -26,7 +37,9 @@ def check_for_third(address: str) -> bool:
         "gold coast",
         "nyasaland",
         "korea",
-        "russia"
+        "russia",
+        "usa",
+        "uk"
     ]
     countries = countries + old_country_names
     if "," in address:
@@ -194,23 +207,23 @@ def swapper(i1, i2, i3):
     for data in inputs:
         original_data = data
 
-        if check_for_third(data):
+        if check_for_third(translate_text(data)):
             if third is None:
                 third = data
             else:
                 third, data = data, third
-                if check_for_first(data):
+                if check_for_first(translate_text(data)):
                     first_candidates.append(data)
-                elif check_for_second(data):
+                elif check_for_second(translate_text(data)):
                     if second is None:
                         second = data
                     else:
                         second, data = data, second
 
-        elif check_for_first(data):
+        elif check_for_first(translate_text(data)):
             first_candidates.append(data)
 
-        elif check_for_second(data):
+        elif check_for_second(translate_text(data)):
             if second is None:
                 second = data
             else: