File size: 3,458 Bytes
51f2dc1 |
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 |
// mapping of tab to its contents
const TABS = {
"chat-tab": "chat-tab-contents",
"explore-tab": "explore-tab-contents"
};
// switch to the specified tab
function switchTab(newTab) {
Object.keys(TABS).forEach(tabId => {
const tabElement = document.getElementById(tabId);
if (tabElement) {
tabElement.classList.remove("tab-active");
}
});
// Hide all tab contents
Object.values(TABS).forEach(contentId => {
const contentElement = document.getElementById(contentId);
if (contentElement) {
contentElement.classList.add("hidden");
}
});
// Activate the new tab if it exists in the mapping
if (newTab in TABS) {
const newTabElement = document.getElementById(newTab);
const newContentElement = document.getElementById(TABS[newTab]);
if (newTabElement) newTabElement.classList.add("tab-active");
if (newContentElement) newContentElement.classList.remove("hidden");
}
}
function addMessage(message, role = "user") {
const chatContainer = document.getElementById("chat-messages-container");
const chatDiv = document.createElement("div");
chatDiv.classList.add("chat");
if (role === "user")
chatDiv.classList.add("chat-end");
else if (role === "bot")
chatDiv.classList.add("chat-start");
else
chatDiv.classList.add("chat-start");
const bubbleDiv = document.createElement("div");
bubbleDiv.classList.add("chat-bubble");
if (role === "user")
bubbleDiv.classList.add("chat-bubble-primary");
else if (role === "bot")
bubbleDiv.classList.add("chat-bubble-secondary");
else
bubbleDiv.classList.add("chat-bubble-secondary");
bubbleDiv.textContent = message;
chatDiv.appendChild(bubbleDiv);
chatContainer.appendChild(chatDiv);
chatContainer.scrollTop = chatContainer.scrollHeight;
}
function escapeString(str) {
return str
.replace(/\\/g, '\\\\') // Escape backslashes
.replace(/"/g, '\\"') // Escape double quotes
.replace(/\n/g, '\\n') // Escape newlines
.replace(/\r/g, '\\r') // Escape carriage returns
.replace(/\t/g, '\\t'); // Escape tabs
}
// extract entities from the given input text
async function extractEntities(text) {
let escaped_text = escapeString(text);
try {
let graph_data_req = await fetch("/extract_entities", {
method: "POST",
headers: new Headers({
"Content-Type": "application/json"
}),
body: JSON.stringify({
content: escaped_text
})
})
let graph_data = await graph_data_req.json();
console.log(graph_data);
var options = {};
var network = new vis.Network(document.getElementById('grapha'), graph_data, options);
addMessage("I've created a knowledge graph based on what you've provided me. Check the explore tab !", "bot");
} catch (e) {
console.error("Error while trying to extract entities into a KG", e);
}
}
document.addEventListener('DOMContentLoaded', _ => {
document.getElementById('chat-send-input').addEventListener('click', (ev) => {
let textbox = document.getElementById('chat-input');
let text = textbox.value;
addMessage(text, "user");
extractEntities(text);
textbox.value = "";
});
}) |