fix missing images in acheong08#78

itsharex · Jul 26, 2023 · c239143 · c239143
1 parent 1ad3f1d
commit c239143
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Bard.py b/src/Bard.py
@@ -319,7 +319,7 @@ async def ask(self, message: str) -> dict:
         MESSAGE = " ".join(sys.argv[1:])
         response = chatbot.ask(MESSAGE)
         console.print(Markdown(response["content"]))
-        console.print(response["images"] if response["images"] else "")
+        console.print(response["images"] if response.get("images") else "")
         sys.exit(0)
     parser = argparse.ArgumentParser()
     parser.add_argument(
@@ -355,7 +355,7 @@ async def ask(self, message: str) -> dict:
             print("Google Bard:")
             response = chatbot.ask(user_prompt)
             console.print(Markdown(response["content"]))
-            console.print(response["images"] if response["images"] else "")
+            console.print(response["images"] if response.get("images") else "")
             print()
     except KeyboardInterrupt:
         print("Exiting...")