improvement : Changed images to list so that it maintain order of ima… (

#49)
acheong08 · Jun 9, 2023 · 6aeace1 · 6aeace1
1 parent 77419ec
commit 6aeace1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Bard.py b/src/Bard.py
@@ -236,12 +236,12 @@ async def ask(self, message: str) -> dict:
         if not chat_data:
             return {"content": f"Google Bard encountered an error: {resp.content}."}
         json_chat_data = json.loads(chat_data)
-        images = set()
+        images = []
         if len(json_chat_data) >= 3:
             if len(json_chat_data[4][0]) >= 4:
                 if json_chat_data[4][0][4]:
                     for img in json_chat_data[4][0][4]:
-                        images.add(img[0][0][0])
+                        images.append(img[0][0][0])
         results = {
             "content": json_chat_data[0][0],
             "conversation_id": json_chat_data[1][0],