From ed18c04576bdafda87452f51c847cffba829f84e Mon Sep 17 00:00:00 2001 From: Debanjum Date: Wed, 15 Jan 2025 18:37:54 +0700 Subject: [PATCH] Fix wrapping base64 generated image for inline display Resolves #1082 --- src/khoj/processor/image/generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/khoj/processor/image/generate.py b/src/khoj/processor/image/generate.py index daf0dbf9..252e61eb 100644 --- a/src/khoj/processor/image/generate.py +++ b/src/khoj/processor/image/generate.py @@ -111,7 +111,7 @@ async def text_to_image( image_url = upload_image(webp_image_bytes, user.uuid) if not image_url: - image = base64.b64encode(webp_image_bytes).decode("utf-8") + image = f"data:image/webp;base64,{base64.b64encode(webp_image_bytes).decode('utf-8')}" yield image_url or image, status_code, image_prompt