git: ensure the order of image urls (#179)

ensure the order of image urls
dsdanielpark · Sep 4, 2023 · f5d8fcc · f5d8fcc
2 parents 539a3c5 + 00575ae
commit f5d8fcc
Show file tree

Hide file tree

Showing 7 changed files with 19 additions and 19 deletions.
diff --git a/bardapi/core.py b/bardapi/core.py
@@ -154,7 +154,7 @@ def get_answer(self, input_text: str) -> dict:
                     "text_query": str,
                     "choices": list,
                     "links": list,
-                    "images": set,
+                    "images": list,
                     "program_lang": str,
                     "code": str,
                     "status_code": int
@@ -219,12 +219,12 @@ def get_answer(self, input_text: str) -> dict:
         resp_json = json.loads(resp_dict)
 
         # [Optional] gather image links
-        images = set()
+        images = list()
         try:
             if len(resp_json) >= 3:
                 nested_list = resp_json[4][0][4]
                 for img in nested_list:
-                    images.add(img[0][0][0])
+                    images.append(img[0][0][0])
         except (IndexError, TypeError, KeyError):
             pass
 
@@ -461,7 +461,7 @@ def ask_about_image(
                     "text_query": str,
                     "choices": list,
                     "links": list,
-                    "images": set,
+                    "images": list,
                     "program_lang": str,
                     "code": str,
                     "status_code": int

diff --git a/bardapi/core_async.py b/bardapi/core_async.py
@@ -143,7 +143,7 @@ async def get_answer(self, input_text: str) -> dict:
                     "text_query": str,
                     "choices": list,
                     "links": list
-                    "images": set,
+                    "images": list,
                     "program_lang": str,
                     "code": str,
                     "status_code": int
@@ -210,12 +210,12 @@ async def get_answer(self, input_text: str) -> dict:
 
         # [Optional] Gather image links
         try:
-            images = set()
+            images = list()
             if len(resp_json) >= 3:
                 if len(resp_json[4][0]) >= 4 and resp_json[4][0][4] is not None:
                     for img in resp_json[4][0][4]:
                         try:
-                            images.add(img[0][0][0])
+                            images.append(img[0][0][0])
                         except Exception as e:
                             # TODO:
                             #  handle exception using logging instead
@@ -584,7 +584,7 @@ async def ask_about_image(
                     "text_query": str,
                     "choices": list,
                     "links": list,
-                    "images": set,
+                    "images": list,
                     "program_lang": str,
                     "code": str,
                     "status_code": int

diff --git a/bardapi/core_cookies.py b/bardapi/core_cookies.py
@@ -138,7 +138,7 @@ def get_answer(self, input_text: str) -> dict:
                     "text_query": str,
                     "choices": list,
                     "links": list
-                    "images": set,
+                    "images": list,
                     "program_lang": str,
                     "code": str,
                     "status_code": int
@@ -202,7 +202,7 @@ def ask_about_image(
                     "text_query": str,
                     "choices": list,
                     "links": list,
-                    "images": set,
+                    "images": list,
                     "program_lang": str,
                     "code": str,
                     "status_code": int
@@ -398,7 +398,7 @@ async def get_answer(self, input_text: str) -> dict:
                     "text_query": str,
                     "choices": list,
                     "links": list
-                    "images": set,
+                    "images": list,
                     "program_lang": str,
                     "code": str,
                     "status_code": int
@@ -475,7 +475,7 @@ async def ask_about_image(
                     "text_query": str,
                     "choices": list,
                     "links": list,
-                    "images": set,
+                    "images": list,
                     "program_lang": str,
                     "code": str,
                     "status_code": int

diff --git a/documents/README_DEV.md b/documents/README_DEV.md
@@ -195,7 +195,7 @@ from bardapi import Bard
 bard = Bard(token='xxxxxxxx')
 res = bard.get_answer("Find me an image of the main entrance of Stanford University.")
 res['links'] # Get image links (list)
-res['images'] # Get images (set)
+res['images'] # Get images (list)
 ```
 
 <br>

diff --git a/func_test.ipynb b/func_test.ipynb
@@ -51,7 +51,7 @@
     "bard = Bard(token=token)\n",
     "res = bard.get_answer(\"Find me an image of the main entrance of Stanford University.\")\n",
     "res['links'] # Get image links (list)\n",
-    "res['images'] # Get images (set)"
+    "res['images'] # Get images (list)"
    ]
   },
   {
@@ -64,7 +64,7 @@
     "bard = Bard(token=token)\n",
     "res = bard.get_answer(\"Find me an image of the main entrance of Stanford University.\")\n",
     "res['links'] # Get image links (list)\n",
-    "res['images'] # Get images (set)"
+    "res['images'] # Get images (list)"
    ]
   },
   {

diff --git a/translate_to/asyncio_core.py b/translate_to/asyncio_core.py
@@ -153,11 +153,11 @@ async def get_answer(self, input_text: str) -> dict:
         if not resp_dict:
             return {"content": f"Response Error: {resp_text}."}
         resp_json = json.loads(resp_dict)
-        images = set()
+        images = list()
         if len(resp_json) >= 3:
             if len(resp_json[4][0]) >= 4 and resp_json[4][0][4] is not None:
                 for img in resp_json[4][0][4]:
-                    images.add(img[0][0][0])
+                    images.append(img[0][0][0])
         parsed_answer = json.loads(resp_dict)
         if self.language is not None and self.language not in ALLOWED_LANGUAGES:
             translator_to_lang = GoogleTranslator(source="auto", target=self.language)

diff --git a/translate_to/core_async.py b/translate_to/core_async.py
@@ -118,12 +118,12 @@ async def get_answer(self, input_text: str) -> dict:
         resp_json = json.loads(resp_dict)
 
         # Gather image links
-        images = set()
+        images = list()
         if len(resp_json) >= 3:
             if len(resp_json[4][0]) >= 4 and resp_json[4][0][4] is not None:
                 for img in resp_json[4][0][4]:
                     try:
-                        images.add(img[0][0][0])
+                        images.append(img[0][0][0])
                     except Exception as e:
                         # TODO:
                         #  handle exception using logging instead