sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-20 06:55:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 13

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
Show stats 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}`;