klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-06 13:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update clients to support rendering webp images inline

Browse Source 
This is for self-hosted scenarios where AWS S3 uploads is not enabled
This commit is contained in:
Debanjum Singh Solanky
2024-04-13 13:11:18 +05:30
parent d21f22ffa1
commit c6e8443631
3 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/interface/desktop/chat.html
View File

@@ -214,6 +214,8 @@
imageMarkdown = `![](data:image/png;base64,${message})`; imageMarkdown = `![](data:image/png;base64,${message})`;
} else if (intentType === "text-to-image2") { } else if (intentType === "text-to-image2") {
imageMarkdown = `![](${message})`; imageMarkdown = `![](${message})`;
} else if (intentType === "text-to-image-v3") {
imageMarkdown = `![](data:image/webp;base64,${message})`;
} }
const inferredQuery = inferredQueries?.[0]; const inferredQuery = inferredQueries?.[0];
@@ -288,6 +290,8 @@
imageMarkdown = `![](data:image/png;base64,${message})`; imageMarkdown = `![](data:image/png;base64,${message})`;
} else if (intentType === "text-to-image2") { } else if (intentType === "text-to-image2") {
imageMarkdown = `![](${message})`; imageMarkdown = `![](${message})`;
} else if (intentType === "text-to-image-v3") {
imageMarkdown = `![](data:image/webp;base64,${message})`;
} }
const inferredQuery = inferredQueries?.[0]; const inferredQuery = inferredQueries?.[0];
if (inferredQuery) { if (inferredQuery) {
@@ -509,6 +513,8 @@
rawResponse += `![${query}](data:image/png;base64,${responseAsJson.image})`; rawResponse += `![${query}](data:image/png;base64,${responseAsJson.image})`;
} else if (responseAsJson.intentType === "text-to-image2") { } else if (responseAsJson.intentType === "text-to-image2") {
rawResponse += `![${query}](${responseAsJson.image})`; rawResponse += `![${query}](${responseAsJson.image})`;
} else if (responseAsJson.intentType === "text-to-image-v3") {
rawResponse += `![${query}](data:image/webp;base64,${responseAsJson.image})`;
} }
const inferredQueries = responseAsJson.inferredQueries?.[0]; const inferredQueries = responseAsJson.inferredQueries?.[0];
if (inferredQueries) { if (inferredQueries) {

4
src/interface/obsidian/src/chat_modal.ts
View File

@@ -156,6 +156,8 @@ export class KhojChatModal extends Modal {
imageMarkdown = `![](data:image/png;base64,${message})`; imageMarkdown = `![](data:image/png;base64,${message})`;
} else if (intentType === "text-to-image2") { } else if (intentType === "text-to-image2") {
imageMarkdown = `![](${message})`; imageMarkdown = `![](${message})`;
} else if (intentType === "text-to-image-v3") {
imageMarkdown = `![](data:image/webp;base64,${message})`;
} }
if (inferredQueries) { if (inferredQueries) {
imageMarkdown += "\n\n**Inferred Query**:"; imageMarkdown += "\n\n**Inferred Query**:";
@@ -429,6 +431,8 @@ export class KhojChatModal extends Modal {
responseText += `![${query}](data:image/png;base64,${responseAsJson.image})`; responseText += `![${query}](data:image/png;base64,${responseAsJson.image})`;
} else if (responseAsJson.intentType === "text-to-image2") { } else if (responseAsJson.intentType === "text-to-image2") {
responseText += `![${query}](${responseAsJson.image})`; responseText += `![${query}](${responseAsJson.image})`;
} else if (responseAsJson.intentType === "text-to-image-v3") {
responseText += `![${query}](data:image/webp;base64,${responseAsJson.image})`;
} }
const inferredQuery = responseAsJson.inferredQueries?.[0]; const inferredQuery = responseAsJson.inferredQueries?.[0];
if (inferredQuery) { if (inferredQuery) {

6
src/khoj/interface/web/chat.html
View File

@@ -244,6 +244,8 @@ To get started, just start typing below. You can also type / to see a list of co
imageMarkdown = `![](data:image/png;base64,${message})`; imageMarkdown = `![](data:image/png;base64,${message})`;
} else if (intentType === "text-to-image2") { } else if (intentType === "text-to-image2") {
imageMarkdown = `![](${message})`; imageMarkdown = `![](${message})`;
} else if (intentType === "text-to-image-v3") {
imageMarkdown = `![](data:image/webp;base64,${message})`;
} }
const inferredQuery = inferredQueries?.[0]; const inferredQuery = inferredQueries?.[0];
if (inferredQuery) { if (inferredQuery) {
@@ -312,6 +314,8 @@ To get started, just start typing below. You can also type / to see a list of co
imageMarkdown = `![](data:image/png;base64,${message})`; imageMarkdown = `![](data:image/png;base64,${message})`;
} else if (intentType === "text-to-image2") { } else if (intentType === "text-to-image2") {
imageMarkdown = `![](${message})`; imageMarkdown = `![](${message})`;
} else if (intentType === "text-to-image-v3") {
imageMarkdown = `![](data:image/webp;base64,${message})`;
} }
const inferredQuery = inferredQueries?.[0]; const inferredQuery = inferredQueries?.[0];
if (inferredQuery) { if (inferredQuery) {
@@ -619,6 +623,8 @@ To get started, just start typing below. You can also type / to see a list of co
rawResponse += `![generated_image](data:image/png;base64,${imageJson.image})`; rawResponse += `![generated_image](data:image/png;base64,${imageJson.image})`;
} else if (imageJson.intentType === "text-to-image2") { } else if (imageJson.intentType === "text-to-image2") {
rawResponse += `![generated_image](${imageJson.image})`; rawResponse += `![generated_image](${imageJson.image})`;
} else if (imageJson.intentType === "text-to-image-v3") {
rawResponse = `![](data:image/webp;base64,${imageJson.image})`;
} }
if (inferredQuery) { if (inferredQuery) {
rawResponse += `\n\n**Inferred Query**:\n\n${inferredQuery}`; rawResponse += `\n\n**Inferred Query**:\n\n${inferredQuery}`;