File size: 4,619 Bytes
0bd62e5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 |
import type { FileData } from "@gradio/client";
import { uploadToHuggingFace } from "@gradio/utils";
import type {
TupleFormat,
ComponentMessage,
ComponentData,
TextMessage,
NormalisedMessage,
Message
} from "../types";
export const format_chat_for_sharing = async (
chat: [string | FileData | null, string | FileData | null][]
): Promise<string> => {
let messages = await Promise.all(
chat.map(async (message_pair) => {
return await Promise.all(
message_pair.map(async (message, i) => {
if (message === null) return "";
let speaker_emoji = i === 0 ? "😃" : "🤖";
let html_content = "";
if (typeof message === "string") {
const regexPatterns = {
audio: /<audio.*?src="(\/file=.*?)"/g,
video: /<video.*?src="(\/file=.*?)"/g,
image: /<img.*?src="(\/file=.*?)".*?\/>|!\[.*?\]\((\/file=.*?)\)/g
};
html_content = message;
for (let [_, regex] of Object.entries(regexPatterns)) {
let match;
while ((match = regex.exec(message)) !== null) {
const fileUrl = match[1] || match[2];
const newUrl = await uploadToHuggingFace(fileUrl, "url");
html_content = html_content.replace(fileUrl, newUrl);
}
}
} else {
if (!message?.url) return "";
const file_url = await uploadToHuggingFace(message.url, "url");
if (message.mime_type?.includes("audio")) {
html_content = `<audio controls src="${file_url}"></audio>`;
} else if (message.mime_type?.includes("video")) {
html_content = file_url;
} else if (message.mime_type?.includes("image")) {
html_content = `<img src="${file_url}" />`;
}
}
return `${speaker_emoji}: ${html_content}`;
})
);
})
);
return messages
.map((message_pair) =>
message_pair.join(
message_pair[0] !== "" && message_pair[1] !== "" ? "\n" : ""
)
)
.join("\n");
};
const redirect_src_url = (src: string, root: string): string =>
src.replace('src="/file', `src="${root}file`);
function get_component_for_mime_type(
mime_type: string | null | undefined
): string {
if (!mime_type) return "file";
if (mime_type.includes("audio")) return "audio";
if (mime_type.includes("video")) return "video";
if (mime_type.includes("image")) return "image";
return "file";
}
function convert_file_message_to_component_message(
message: any
): ComponentData {
const _file = Array.isArray(message.file) ? message.file[0] : message.file;
return {
component: get_component_for_mime_type(_file?.mime_type),
value: message.file,
alt_text: message.alt_text,
constructor_args: {},
props: {}
} as ComponentData;
}
export function normalise_messages(
messages: Message[] | null,
root: string
): NormalisedMessage[] | null {
if (messages === null) return messages;
return messages.map((message, i) => {
if (typeof message.content === "string") {
return {
role: message.role,
metadata: message.metadata,
content: redirect_src_url(message.content, root),
type: "text",
index: i
};
} else if ("file" in message.content) {
return {
content: convert_file_message_to_component_message(message.content),
metadata: message.metadata,
role: message.role,
type: "component",
index: i
};
}
return { type: "component", ...message } as ComponentMessage;
});
}
export function normalise_tuples(
messages: TupleFormat,
root: string
): NormalisedMessage[] | null {
if (messages === null) return messages;
const msg = messages.flatMap((message_pair, i) => {
return message_pair.map((message, index) => {
if (message == null) return null;
const role = index == 0 ? "user" : "assistant";
if (typeof message === "string") {
return {
role: role,
type: "text",
content: redirect_src_url(message, root),
metadata: { title: null },
index: [i, index]
} as TextMessage;
}
if ("file" in message) {
return {
content: convert_file_message_to_component_message(message),
role: role,
type: "component",
index: [i, index]
} as ComponentMessage;
}
return {
role: role,
content: message,
type: "component",
index: [i, index]
} as ComponentMessage;
});
});
return msg.filter((message) => message != null) as NormalisedMessage[];
}
export function is_component_message(
message: NormalisedMessage
): message is ComponentMessage {
return message.type === "component";
}
|