fix: conditions in core

dsdanielpark · Dec 27, 2023 · 04496f6 · 04496f6
1 parent b98554c
commit 04496f6
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/bardapi/core.py b/bardapi/core.py
@@ -674,19 +674,21 @@ def ask_about_image(
             translator_to_eng = GoogleTranslator(source="auto", target="en")
 
         # [Optional] Set language
-        if (
+        if self.language is None and lang is None:
+            translated_input_text = input_text
+        elif (
             (self.language is not None or lang is not None)
             and self.language not in ALLOWED_LANGUAGES
             and self.google_translator_api_key is None
         ):
             translator_to_eng = GoogleTranslator(source="auto", target="en")
-            transl_text = translator_to_eng.translate(input_text)
+            translated_input_text = translator_to_eng.translate(input_text)
         elif (
             (self.language is not None or lang is not None)
             and self.language not in ALLOWED_LANGUAGES
             and self.google_translator_api_key is not None
         ):
-            transl_text = google_official_translator.translate(
+            translated_input_text = google_official_translator.translate(
                 input_text, target_language="en"
             )
         elif (
@@ -695,15 +697,16 @@ def ask_about_image(
             and self.google_translator_api_key is None
         ):
             translator_to_eng = GoogleTranslator(source="auto", target="en")
-            transl_text = translator_to_eng.translate(input_text)
+            translated_input_text = translator_to_eng.translate(input_text)
+
 
         # Supported format: jpeg, png, webp
         image_url = upload_image(image)
 
         input_data_struct = [
             None,
             [
-                [transl_text, 0, None, [[[image_url, 1], "uploaded_photo.jpg"]]],
+                [translated_input_text, 0, None, [[[image_url, 1], "uploaded_photo.jpg"]]],
                 [lang if lang is not None else self.language],
                 ["", "", ""],
                 "",  # Unknown random string value (1000 characters +)