Fix wrapping base64 generated image for inline display

Resolves #1082
2026-03-02 13:18:18 +00:00 · 2025-01-15 18:37:54 +07:00
parent f8b887cabd
commit ed18c04576
1 changed files with 1 additions and 1 deletions
--- 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