diff options
author | Heiner Lohaus <hlohaus@users.noreply.github.com> | 2024-03-16 10:48:37 +0100 |
---|---|---|
committer | Heiner Lohaus <hlohaus@users.noreply.github.com> | 2024-03-16 10:48:37 +0100 |
commit | 905ced06bd27d4a252791c2316d0bb00b8326d35 (patch) | |
tree | 223c46b0c1902ec73ea55d401eebbf01078c082d /g4f/gui/client/static/js/chat.v1.js | |
parent | Move raise_for_status, Create FormData wrapper (diff) | |
download | gpt4free-905ced06bd27d4a252791c2316d0bb00b8326d35.tar gpt4free-905ced06bd27d4a252791c2316d0bb00b8326d35.tar.gz gpt4free-905ced06bd27d4a252791c2316d0bb00b8326d35.tar.bz2 gpt4free-905ced06bd27d4a252791c2316d0bb00b8326d35.tar.lz gpt4free-905ced06bd27d4a252791c2316d0bb00b8326d35.tar.xz gpt4free-905ced06bd27d4a252791c2316d0bb00b8326d35.tar.zst gpt4free-905ced06bd27d4a252791c2316d0bb00b8326d35.zip |
Diffstat (limited to 'g4f/gui/client/static/js/chat.v1.js')
-rw-r--r-- | g4f/gui/client/static/js/chat.v1.js | 1016 |
1 files changed, 1016 insertions, 0 deletions
diff --git a/g4f/gui/client/static/js/chat.v1.js b/g4f/gui/client/static/js/chat.v1.js new file mode 100644 index 00000000..bfe641d5 --- /dev/null +++ b/g4f/gui/client/static/js/chat.v1.js @@ -0,0 +1,1016 @@ +const colorThemes = document.querySelectorAll('[name="theme"]'); +const message_box = document.getElementById(`messages`); +const messageInput = document.getElementById(`message-input`); +const box_conversations = document.querySelector(`.top`); +const stop_generating = document.querySelector(`.stop_generating`); +const regenerate = document.querySelector(`.regenerate`); +const sidebar = document.querySelector(".conversations"); +const sidebar_button = document.querySelector(".mobile-sidebar"); +const sendButton = document.getElementById("send-button"); +const imageInput = document.getElementById("image"); +const cameraInput = document.getElementById("camera"); +const fileInput = document.getElementById("file"); +const inputCount = document.getElementById("input-count") +const providerSelect = document.getElementById("provider"); +const modelSelect = document.getElementById("model"); +const modelProvider = document.getElementById("model2"); +const systemPrompt = document.getElementById("systemPrompt") +const jailbreak = document.getElementById("jailbreak"); + +let prompt_lock = false; + +const options = ["switch", "model", "model2", "jailbreak", "patch", "provider", "history"]; + +messageInput.addEventListener("blur", () => { + window.scrollTo(0, 0); +}); + +messageInput.addEventListener("focus", () => { + document.documentElement.scrollTop = document.documentElement.scrollHeight; +}); + +appStorage = window.localStorage || { + setItem: (key, value) => self[key] = value, + getItem: (key) => self[key], + removeItem: (key) => delete self[key], + length: 0 +} + +const markdown = window.markdownit(); +const markdown_render = (content) => { + return markdown.render(content + .replaceAll(/<!-- generated images start -->[\s\S]+<!-- generated images end -->/gm, "") + .replaceAll(/<img data-prompt="[^>]+">/gm, "") + ) + .replaceAll("<a href=", '<a target="_blank" href=') + .replaceAll('<code>', '<code class="language-plaintext">') +} + +hljs.addPlugin(new CopyButtonPlugin()); +let typesetPromise = Promise.resolve(); +const highlight = (container) => { + container.querySelectorAll('code:not(.hljs').forEach((el) => { + if (el.className != "hljs") { + hljs.highlightElement(el); + } + }); + typesetPromise = typesetPromise.then( + () => MathJax.typesetPromise([container]) + ).catch( + (err) => console.log('Typeset failed: ' + err.message) + ); +} + +const register_remove_message = async () => { + document.querySelectorAll(".message .fa-xmark").forEach(async (el) => { + if (!("click" in el.dataset)) { + el.dataset.click = "true"; + el.addEventListener("click", async () => { + if (prompt_lock) { + return; + } + const message_el = el.parentElement.parentElement; + await remove_message(window.conversation_id, message_el.dataset.index); + await load_conversation(window.conversation_id, false); + }) + } + }); +} + +const delete_conversations = async () => { + for (let i = 0; i < appStorage.length; i++){ + let key = appStorage.key(i); + if (key.startsWith("conversation:")) { + appStorage.removeItem(key); + } + } + hide_sidebar(); + await new_conversation(); +}; + +const handle_ask = async () => { + messageInput.style.height = "82px"; + messageInput.focus(); + window.scrollTo(0, 0); + + message = messageInput.value + if (message.length <= 0) { + return; + } + messageInput.value = ""; + prompt_lock = true; + count_input() + await add_conversation(window.conversation_id, message); + + if ("text" in fileInput.dataset) { + message += '\n```' + fileInput.dataset.type + '\n'; + message += fileInput.dataset.text; + message += '\n```' + } + let message_index = await add_message(window.conversation_id, "user", message); + window.token = message_id(); + + if (imageInput.dataset.src) URL.revokeObjectURL(imageInput.dataset.src); + const input = imageInput && imageInput.files.length > 0 ? imageInput : cameraInput + if (input.files.length > 0) imageInput.dataset.src = URL.createObjectURL(input.files[0]); + else delete imageInput.dataset.src + + message_box.innerHTML += ` + <div class="message" data-index="${message_index}"> + <div class="user"> + ${user_image} + <i class="fa-solid fa-xmark"></i> + <i class="fa-regular fa-phone-arrow-up-right"></i> + </div> + <div class="content" id="user_${token}"> + <div class="content_inner"> + ${markdown_render(message)} + ${imageInput.dataset.src + ? '<img src="' + imageInput.dataset.src + '" alt="Image upload">' + : '' + } + </div> + <div class="count">${count_words_and_tokens(message, get_selected_model())}</div> + </div> + </div> + `; + highlight(message_box); + await ask_gpt(); +}; + +const remove_cancel_button = async () => { + stop_generating.classList.add(`stop_generating-hiding`); + + setTimeout(() => { + stop_generating.classList.remove(`stop_generating-hiding`); + stop_generating.classList.add(`stop_generating-hidden`); + }, 300); +}; + +const prepare_messages = (messages, filter_last_message=true) => { + // Removes none user messages at end + if (filter_last_message) { + let last_message; + while (last_message = messages.pop()) { + if (last_message["role"] == "user") { + messages.push(last_message); + break; + } + } + } + + // Remove history, if it's selected + if (document.getElementById('history')?.checked) { + if (filter_last_message) { + messages = [messages.pop()]; + } else { + messages = [messages.pop(), messages.pop()]; + } + } + + let new_messages = []; + if (messages) { + for (i in messages) { + new_message = messages[i]; + // Remove generated images from history + new_message.content = new_message.content.replaceAll( + /<!-- generated images start -->[\s\S]+<!-- generated images end -->/gm, + "" + ) + delete new_message["provider"]; + // Remove regenerated messages + if (!new_message.regenerate) { + new_messages.push(new_message) + } + } + } + + // Add system message + system_content = systemPrompt?.value; + if (system_content) { + new_messages.unshift({ + "role": "system", + "content": system_content + }); + } + + return new_messages; +} + +async function add_message_chunk(message) { + if (message.type == "provider") { + window.provider_result = message.provider; + window.content.querySelector('.provider').innerHTML = ` + <a href="${message.provider.url}" target="_blank"> + ${message.provider.name} + </a> + ${message.provider.model ? ' with ' + message.provider.model : ''} + ` + } else if (message.type == "message") { + console.error(messag.message) + return; + } else if (message.type == "error") { + console.error(message.error); + window.content_inner.innerHTML += `<p><strong>An error occured:</strong> ${message.error}</p>`; + } else if (message.type == "content") { + window.text += message.content; + html = markdown_render(window.text); + let lastElement, lastIndex = null; + for (element of ['</p>', '</code></pre>', '</p>\n</li>\n</ol>', '</li>\n</ol>', '</li>\n</ul>']) { + const index = html.lastIndexOf(element) + if (index - element.length > lastIndex) { + lastElement = element; + lastIndex = index; + } + } + if (lastIndex) { + html = html.substring(0, lastIndex) + '<span id="cursor"></span>' + lastElement; + } + window.content_inner.innerHTML = html; + window.content_count.innerText = count_words_and_tokens(text, window.provider_result?.model); + highlight(window.content_inner); + } + + window.scrollTo(0, 0); + if (message_box.scrollTop >= message_box.scrollHeight - message_box.clientHeight - 100) { + message_box.scrollTo({ top: message_box.scrollHeight, behavior: "auto" }); + } +} + +const ask_gpt = async () => { + regenerate.classList.add(`regenerate-hidden`); + messages = await get_messages(window.conversation_id); + total_messages = messages.length; + messages = prepare_messages(messages); + + stop_generating.classList.remove(`stop_generating-hidden`); + + message_box.scrollTop = message_box.scrollHeight; + window.scrollTo(0, 0); + + el = message_box.querySelector('.count_total'); + el ? el.parentElement.removeChild(el) : null; + + message_box.innerHTML += ` + <div class="message" data-index="${total_messages}"> + <div class="assistant"> + ${gpt_image} + <i class="fa-solid fa-xmark"></i> + <i class="fa-regular fa-phone-arrow-down-left"></i> + </div> + <div class="content" id="gpt_${window.token}"> + <div class="provider"></div> + <div class="content_inner"><span id="cursor"></span></div> + <div class="count"></div> + </div> + </div> + `; + + window.controller = new AbortController(); + window.text = ""; + window.error = null; + window.provider_result = null; + + window.content = document.getElementById(`gpt_${window.token}`); + window.content_inner = content.querySelector('.content_inner'); + window.content_count = content.querySelector('.count'); + + message_box.scrollTop = message_box.scrollHeight; + window.scrollTo(0, 0); + try { + const input = imageInput && imageInput.files.length > 0 ? imageInput : cameraInput; + const file = input && input.files.length > 0 ? input.files[0] : null; + await api("conversation", { + id: window.token, + conversation_id: window.conversation_id, + model: get_selected_model(), + jailbreak: jailbreak?.options[jailbreak.selectedIndex].value, + web_search: document.getElementById("switch").checked, + provider: providerSelect.options[providerSelect.selectedIndex].value, + patch_provider: document.getElementById("patch")?.checked, + messages: messages + }, file); + if (!error) { + html = markdown_render(text); + content_inner.innerHTML = html; + highlight(content_inner); + + if (imageInput) imageInput.value = ""; + if (cameraInput) cameraInput.value = ""; + if (fileInput) fileInput.value = ""; + } + } catch (e) { + console.error(e); + if (e.name != "AbortError") { + error = true; + text = "oops ! something went wrong, please try again / reload. [stacktrace in console]"; + content_inner.innerHTML = text; + } else { + content_inner.innerHTML += " [aborted]"; + if (text) text += " [aborted]"; + } + } + if (!error && text) { + await add_message(window.conversation_id, "assistant", text, provider_result); + await load_conversation(window.conversation_id); + } else { + let cursorDiv = document.getElementById("cursor"); + if (cursorDiv) cursorDiv.parentNode.removeChild(cursorDiv); + } + window.scrollTo(0, 0); + message_box.scrollTop = message_box.scrollHeight; + await remove_cancel_button(); + await register_remove_message(); + prompt_lock = false; + await load_conversations(); + regenerate.classList.remove("regenerate-hidden"); +}; + +const clear_conversations = async () => { + const elements = box_conversations.childNodes; + let index = elements.length; + + if (index > 0) { + while (index--) { + const element = elements[index]; + if ( + element.nodeType === Node.ELEMENT_NODE && + element.tagName.toLowerCase() !== `button` + ) { + box_conversations.removeChild(element); + } + } + } +}; + +const clear_conversation = async () => { + let messages = message_box.getElementsByTagName(`div`); + + while (messages.length > 0) { + message_box.removeChild(messages[0]); + } +}; + +const show_option = async (conversation_id) => { + const conv = document.getElementById(`conv-${conversation_id}`); + const choi = document.getElementById(`cho-${conversation_id}`); + + conv.style.display = "none"; + choi.style.display = "block"; +}; + +const hide_option = async (conversation_id) => { + const conv = document.getElementById(`conv-${conversation_id}`); + const choi = document.getElementById(`cho-${conversation_id}`); + + conv.style.display = "block"; + choi.style.display = "none"; +}; + +const delete_conversation = async (conversation_id) => { + appStorage.removeItem(`conversation:${conversation_id}`); + + const conversation = document.getElementById(`convo-${conversation_id}`); + conversation.remove(); + + if (window.conversation_id == conversation_id) { + await new_conversation(); + } + + await load_conversations(); +}; + +const set_conversation = async (conversation_id) => { + history.pushState({}, null, `/chat/${conversation_id}`); + window.conversation_id = conversation_id; + + await clear_conversation(); + await load_conversation(conversation_id); + load_conversations(); + hide_sidebar(); +}; + +const new_conversation = async () => { + history.pushState({}, null, `/chat/`); + window.conversation_id = uuid(); + + await clear_conversation(); + if (systemPrompt) { + systemPrompt.value = ""; + } + load_conversations(); + hide_sidebar(); + say_hello(); +}; + +const load_conversation = async (conversation_id, scroll=true) => { + let conversation = await get_conversation(conversation_id); + let messages = conversation?.items || []; + + if (systemPrompt) { + systemPrompt.value = conversation.system || ""; + } + + let elements = ""; + let last_model = null; + for (i in messages) { + let item = messages[i]; + last_model = item.provider?.model; + let next_i = parseInt(i) + 1; + let next_provider = item.provider ? item.provider : (messages.length > next_i ? messages[next_i].provider : null); + let provider_link = item.provider?.name ? `<a href="${item.provider.url}" target="_blank">${item.provider.name}</a>` : ""; + let provider = provider_link ? ` + <div class="provider"> + ${provider_link} + ${item.provider.model ? ' with ' + item.provider.model : ''} + </div> + ` : ""; + elements += ` + <div class="message${item.regenerate ? " regenerate": ""}" data-index="${i}"> + <div class="${item.role}"> + ${item.role == "assistant" ? gpt_image : user_image} + <i class="fa-solid fa-xmark"></i> + ${item.role == "assistant" + ? `<i class="fa-regular fa-phone-arrow-down-left"></i>` + : `<i class="fa-regular fa-phone-arrow-up-right"></i>` + } + </div> + <div class="content"> + ${provider} + <div class="content_inner">${markdown_render(item.content)}</div> + <div class="count">${count_words_and_tokens(item.content, next_provider?.model)}</div> + </div> + </div> + `; + } + + const filtered = prepare_messages(messages, false); + if (filtered.length > 0) { + last_model = last_model?.startsWith("gpt-4") ? "gpt-4" : "gpt-3.5-turbo" + let count_total = GPTTokenizer_cl100k_base?.encodeChat(filtered, last_model).length + if (count_total > 0) { + elements += `<div class="count_total">(${count_total} tokens used)</div>`; + } + } + + message_box.innerHTML = elements; + register_remove_message(); + highlight(message_box); + + if (scroll) { + message_box.scrollTo({ top: message_box.scrollHeight, behavior: "smooth" }); + + setTimeout(() => { + message_box.scrollTop = message_box.scrollHeight; + }, 500); + } +}; + +async function get_conversation(conversation_id) { + let conversation = await JSON.parse( + appStorage.getItem(`conversation:${conversation_id}`) + ); + return conversation; +} + +async function save_conversation(conversation_id, conversation) { + appStorage.setItem( + `conversation:${conversation_id}`, + JSON.stringify(conversation) + ); +} + +async function get_messages(conversation_id) { + let conversation = await get_conversation(conversation_id); + return conversation?.items || []; +} + +async function add_conversation(conversation_id, content) { + if (content.length > 17) { + title = content.substring(0, 17) + '...' + } else { + title = content + ' '.repeat(19 - content.length) + } + + if (appStorage.getItem(`conversation:${conversation_id}`) == null) { + await save_conversation(conversation_id, { + id: conversation_id, + title: title, + system: systemPrompt?.value, + items: [], + }); + } + + history.pushState({}, null, `/chat/${conversation_id}`); +} + +async function save_system_message() { + if (!window.conversation_id) { + return; + } + const conversation = await get_conversation(window.conversation_id); + conversation.system = systemPrompt?.value; + await save_conversation(window.conversation_id, conversation); +} + +const hide_last_message = async (conversation_id) => { + const conversation = await get_conversation(conversation_id) + const last_message = conversation.items.pop(); + if (last_message !== null) { + if (last_message["role"] == "assistant") { + last_message["regenerate"] = true; + } + conversation.items.push(last_message); + } + await save_conversation(conversation_id, conversation); +}; + +const remove_message = async (conversation_id, index) => { + const conversation = await get_conversation(conversation_id); + let new_items = []; + for (i in conversation.items) { + if (i == index - 1) { + if (!conversation.items[index]?.regenerate) { + delete conversation.items[i]["regenerate"]; + } + } + if (i != index) { + new_items.push(conversation.items[i]) + } + } + conversation.items = new_items; + await save_conversation(conversation_id, conversation); +}; + +const add_message = async (conversation_id, role, content, provider) => { + const conversation = await get_conversation(conversation_id); + conversation.items.push({ + role: role, + content: content, + provider: provider + }); + await save_conversation(conversation_id, conversation); + return conversation.items.length - 1; +}; + +const load_conversations = async () => { + let conversations = []; + for (let i = 0; i < appStorage.length; i++) { + if (appStorage.key(i).startsWith("conversation:")) { + let conversation = appStorage.getItem(appStorage.key(i)); + conversations.push(JSON.parse(conversation)); + } + } + + await clear_conversations(); + + for (conversation of conversations) { + box_conversations.innerHTML += ` + <div class="convo" id="convo-${conversation.id}"> + <div class="left" onclick="set_conversation('${conversation.id}')"> + <i class="fa-regular fa-comments"></i> + <span class="convo-title">${conversation.title}</span> + </div> + <i onclick="show_option('${conversation.id}')" class="fa-regular fa-trash" id="conv-${conversation.id}"></i> + <div id="cho-${conversation.id}" class="choise" style="display:none;"> + <i onclick="delete_conversation('${conversation.id}')" class="fa-regular fa-check"></i> + <i onclick="hide_option('${conversation.id}')" class="fa-regular fa-x"></i> + </div> + </div> + `; + } +}; + +document.getElementById(`cancelButton`).addEventListener(`click`, async () => { + window.controller.abort(); + console.log(`aborted ${window.conversation_id}`); +}); + +document.getElementById(`regenerateButton`).addEventListener(`click`, async () => { + prompt_lock = true; + await hide_last_message(window.conversation_id); + window.token = message_id(); + await ask_gpt(); +}); + +const uuid = () => { + return `xxxxxxxx-xxxx-4xxx-yxxx-${Date.now().toString(16)}`.replace( + /[xy]/g, + function (c) { + var r = (Math.random() * 16) | 0, + v = c == "x" ? r : (r & 0x3) | 0x8; + return v.toString(16); + } + ); +}; + +const message_id = () => { + random_bytes = (Math.floor(Math.random() * 1338377565) + 2956589730).toString( + 2 + ); + unix = Math.floor(Date.now() / 1000).toString(2); + + return BigInt(`0b${unix}${random_bytes}`).toString(); +}; + +async function hide_sidebar() { + sidebar.classList.remove("shown"); + sidebar_button.classList.remove("rotated"); +} + +sidebar_button.addEventListener("click", (event) => { + if (sidebar.classList.contains("shown")) { + hide_sidebar(); + } else { + sidebar.classList.add("shown"); + sidebar_button.classList.add("rotated"); + } + window.scrollTo(0, 0); +}); + +const register_settings_storage = async () => { + options.forEach((id) => { + element = document.getElementById(id); + if (!element) { + return; + } + element.addEventListener('change', async (event) => { + switch (event.target.type) { + case "checkbox": + appStorage.setItem(id, event.target.checked); + break; + case "select-one": + appStorage.setItem(id, event.target.selectedIndex); + break; + default: + console.warn("Unresolved element type"); + } + }); + }); +} + +const load_settings_storage = async () => { + options.forEach((id) => { + element = document.getElementById(id); + if (!element || !(value = appStorage.getItem(id))) { + return; + } + if (value) { + switch (element.type) { + case "checkbox": + element.checked = value === "true"; + break; + case "select-one": + element.selectedIndex = parseInt(value); + break; + default: + console.warn("Unresolved element type"); + } + } + }); +} + +const say_hello = async () => { + tokens = [`Hello`, `!`, ` How`,` can`, ` I`,` assist`,` you`,` today`,`?`] + + message_box.innerHTML += ` + <div class="message"> + <div class="assistant"> + ${gpt_image} + <i class="fa-regular fa-phone-arrow-down-left"></i> + </div> + <div class="content"> + <p class=" welcome-message"></p> + </div> + </div> + `; + + to_modify = document.querySelector(`.welcome-message`); + for (token of tokens) { + await new Promise(resolve => setTimeout(resolve, (Math.random() * (100 - 200) + 100))) + to_modify.textContent += token; + } +} + +// Theme storage for recurring viewers +const storeTheme = function (theme) { + appStorage.setItem("theme", theme); +}; + +// set theme when visitor returns +const setTheme = function () { + const activeTheme = appStorage.getItem("theme"); + colorThemes.forEach((themeOption) => { + if (themeOption.id === activeTheme) { + themeOption.checked = true; + } + }); + // fallback for no :has() support + document.documentElement.className = activeTheme; +}; + +colorThemes.forEach((themeOption) => { + themeOption.addEventListener("click", () => { + storeTheme(themeOption.id); + // fallback for no :has() support + document.documentElement.className = themeOption.id; + }); +}); + +function count_tokens(model, text) { + if (model) { + if (model.startsWith("llama2") || model.startsWith("codellama")) { + return llamaTokenizer?.encode(text).length; + } + if (model.startsWith("mistral") || model.startsWith("mixtral")) { + return mistralTokenizer?.encode(text).length; + } + } + return GPTTokenizer_cl100k_base?.encode(text).length; +} + +function count_words(text) { + return text.trim().match(/[\w\u4E00-\u9FA5]+/gu)?.length || 0; +} + +function count_chars(text) { + return text.match(/[^\s\p{P}]/gu)?.length || 0; +} + +function count_words_and_tokens(text, model) { + return `(${count_words(text)} words, ${count_chars(text)} chars, ${count_tokens(model, text)} tokens)`; +} + +let countFocus = messageInput; +let timeoutId; +const count_input = async () => { + if (timeoutId) clearTimeout(timeoutId); + timeoutId = setTimeout(() => { + if (countFocus.value) { + inputCount.innerText = count_words_and_tokens(countFocus.value, get_selected_model()); + } else { + inputCount.innerHTML = " " + } + }, 100); +}; +messageInput.addEventListener("keyup", count_input); +systemPrompt.addEventListener("keyup", count_input); +systemPrompt.addEventListener("focus", function() { + countFocus = systemPrompt; + count_input(); +}); +systemPrompt.addEventListener("blur", function() { + countFocus = messageInput; + count_input(); +}); + +window.addEventListener('load', async function() { + await on_load(); + if (window.conversation_id == "{{chat_id}}") { + window.conversation_id = uuid(); + } else { + await on_api(); + } +}); + +window.addEventListener('pywebviewready', async function() { + await on_api(); +}); + +async function on_load() { + setTheme(); + count_input(); + + if (/\/chat\/.+/.test(window.location.href)) { + load_conversation(window.conversation_id); + } else { + say_hello() + } + load_conversations(); +} + +async function on_api() { + messageInput.addEventListener("keydown", async (evt) => { + if (prompt_lock) return; + + if (evt.keyCode === 13 && !evt.shiftKey) { + evt.preventDefault(); + console.log("pressed enter"); + await handle_ask(); + } else { + messageInput.style.removeProperty("height"); + messageInput.style.height = messageInput.scrollHeight + "px"; + } + }); + sendButton.addEventListener(`click`, async () => { + console.log("clicked send"); + if (prompt_lock) return; + await handle_ask(); + }); + messageInput.focus(); + + register_settings_storage(); + + versions = await api('version') + document.title = 'g4f - ' + versions["version"]; + let text = "version ~ " + if (versions["version"] != versions["latest_version"]) { + let release_url = 'https://github.com/xtekky/gpt4free/releases/tag/' + versions["latest_version"]; + let title = `New version: ${versions["latest_version"]}`; + text += `<a href="${release_url}" target="_blank" title="${title}">${versions["version"]} 🆕</a>`; + } else { + text += versions["version"]; + } + document.getElementById("version_text").innerHTML = text + + models = await api("models"); + models.forEach((model) => { + let option = document.createElement('option'); + option.value = option.text = model; + modelSelect.appendChild(option); + }); + + providers = await api('providers') + providers.forEach((provider) => { + let option = document.createElement('option'); + option.value = option.text = provider; + providerSelect.appendChild(option); + }) + + load_provider_models(); + load_settings_storage() +} + +for (const el of [imageInput, cameraInput]) { + el.addEventListener('click', async () => { + el.value = ''; + if (imageInput.dataset.src) { + URL.revokeObjectURL(imageInput.dataset.src); + delete imageInput.dataset.src + } + }); +} + +fileInput.addEventListener('click', async (event) => { + fileInput.value = ''; + delete fileInput.dataset.text; +}); + +fileInput.addEventListener('change', async (event) => { + if (fileInput.files.length) { + type = fileInput.files[0].type; + if (type && type.indexOf('/')) { + type = type.split('/').pop().replace('x-', '') + type = type.replace('plain', 'plaintext') + .replace('shellscript', 'sh') + .replace('svg+xml', 'svg') + .replace('vnd.trolltech.linguist', 'ts') + } else { + type = fileInput.files[0].name.split('.').pop() + } + fileInput.dataset.type = type + const reader = new FileReader(); + reader.addEventListener('load', async (event) => { + fileInput.dataset.text = event.target.result; + if (type == "json") { + const data = JSON.parse(fileInput.dataset.text); + if ("g4f" in data.options) { + Object.keys(data).forEach(key => { + if (key != "options" && !localStorage.getItem(key)) { + appStorage.setItem(key, JSON.stringify(data[key])); + } + }); + delete fileInput.dataset.text; + await load_conversations(); + fileInput.value = ""; + } + } + }); + reader.readAsText(fileInput.files[0]); + } else { + delete fileInput.dataset.text; + } +}); + +systemPrompt?.addEventListener("blur", async () => { + await save_system_message(); +}); + +function get_selected_model() { + if (modelProvider.selectedIndex >= 0) { + return modelProvider.options[modelProvider.selectedIndex].value; + } else if (modelSelect.selectedIndex >= 0) { + return modelSelect.options[modelSelect.selectedIndex].value; + } +} + +async function api(ressource, args=null, file=null) { + if (window?.pywebview) { + if (args) { + if (ressource == "models") { + ressource = "provider_models"; + } + return pywebview.api["get_" + ressource](args); + } + return pywebview.api["get_" + ressource](); + } + if (ressource == "models" && args) { + ressource = `${ressource}/${args}`; + } + const url = `/backend-api/v2/${ressource}`; + if (ressource == "conversation") { + const body = JSON.stringify(args); + const headers = { + accept: 'text/event-stream' + } + if (file) { + const formData = new FormData(); + formData.append('file', file); + formData.append('json', body); + body = formData; + } else { + headers['content-type'] = 'application/json'; + } + response = await fetch(url, { + method: 'POST', + signal: window.controller.signal, + headers: headers, + body: body + }); + return read_response(response); + } + response = await fetch(url); + return await response.json(); +} + +async function read_response(response) { + const reader = response.body.pipeThrough(new TextDecoderStream()).getReader(); + let buffer = "" + while (true) { + const { value, done } = await reader.read(); + if (done) { + break; + } + for (const line of value.split("\n")) { + if (!line) { + continue; + } + try { + add_message_chunk(JSON.parse(buffer + line)) + buffer = ""; + } catch { + buffer += line + } + } + } +} + +async function load_provider_models() { + const provider = providerSelect.options[providerSelect.selectedIndex].value; + if (!provider) { + return; + } + const models = await api('models', provider); + modelProvider.innerHTML = ''; + if (models.length > 0) { + modelSelect.classList.add("hidden"); + modelProvider.classList.remove("hidden"); + models.forEach((model) => { + let option = document.createElement('option'); + option.value = option.text = model.model; + option.selected = model.default; + modelProvider.appendChild(option); + }); + } else { + modelProvider.classList.add("hidden"); + modelSelect.classList.remove("hidden"); + } +}; +providerSelect.addEventListener("change", load_provider_models) + +function save_storage() { + let filename = new Date().toLocaleString() + filename += ".json" + let data = {"options": {"g4f": ""}}; + for (let i = 0; i < appStorage.length; i++){ + let key = appStorage.key(i); + let item = appStorage.getItem(key); + if (key.startsWith("conversation:")) { + data[key] = JSON.parse(item); + } else { + data["options"][key] = item; + } + } + data = JSON.stringify(data, null, 4); + const blob = new Blob([data], {type: 'text/csv'}); + if(window.navigator.msSaveOrOpenBlob) { + window.navigator.msSaveBlob(blob, filename); + } else{ + const elem = window.document.createElement('a'); + elem.href = window.URL.createObjectURL(blob); + elem.download = filename; + document.body.appendChild(elem); + elem.click(); + document.body.removeChild(elem); + } +}
\ No newline at end of file |