small translation test working

.gitignore

@@ -1 +1,2 @@
 google_api_key.json
+google-cloud-sdk/

requirements.txt

@@ -1,2 +1,3 @@
 python-dotenv==0.19.1
 google-cloud-translate==2.0.1
+tqdm==4.62.3

squad_transform.py

@@ -1,7 +1,9 @@
 import json
 from dotenv import load_dotenv
+from tqdm import tqdm
 
 from squad_utils import print_squad
+from translate_utils import translate_text
 
 
 def sort_qas_by_answer_index(squad):
@@ -69,6 +71,13 @@ def add_special_chars_to_paragraph(paragraph):
             other['answer_end'] += len(special_char) +1
 
 
+def translate_paragraphs(squad):
+    for article in tqdm(squad["data"]):
+        for paragraph in article["paragraphs"]:
+            translated = translate_text(paragraph["context"])
+            paragraph['translatedContext'] = translated
+
+
 if __name__ == "__main__":
     load_dotenv()
 
@@ -77,7 +86,7 @@ if __name__ == "__main__":
 
     sort_qas_by_answer_index(squad)
     transform_squad(squad)
-    print_squad(squad)
+    translate_paragraphs(squad)
 
     with open("./data/squad-v2-dev-small-translated.json", "w") as f:
         json.dump(squad, f, indent=2)

translate_utils.py

@@ -21,7 +21,26 @@ def translate_text(text):
     # will return a sequence of results for each text.
     result = translate_client.translate(text, target_language="sk")
 
-    print(u"Text: {}".format(result["input"]))
+    # print(u"Text: {}".format(result["input"]))
-    print(u"Translation: {}".format(result["translatedText"]))
+    # print(u"Translation: {}".format(result["translatedText"]))
-    print(u"Detected source language: {}".format(result["detectedSourceLanguage"]))
+    # print(u"Detected source language: {}".format(result["detectedSourceLanguage"]))
+    return result['translatedText']
+
+
+def translate_test():
+    try:
+        translate_text("Hello")
+
+        return True
+    except Exception as e:
+        print(e)
+
+        return False
+
+
+if __name__ == "__main__":
+    load_dotenv()
+
+    if translate_test():
+        print("Translate keys work")