feat: show images from Bard's reply

F33RNI · Dec 9, 2023 · 60e0685 · 60e0685
1 parent 666c52d
commit 60e0685
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/BardModule.py b/BardModule.py
@@ -158,6 +158,8 @@ def process_request(self, request_response: RequestResponseContainer) -> None:
             logging.info("Response successfully processed for user {0} ({1})"
                          .format(request_response.user["user_name"], request_response.user["user_id"]))
             request_response.response = bard_response["content"]
+            if "images" in bard_response and len(bard_response["images"]) > 0:
+                request_response.response_images = bard_response["images"]
 
             # Save conversation
             logging.info("Saving conversation_id as {} and response_id as {} and choice_id as {}".