Prathamesh Sable
fixed issue with source
64cadc5
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="utf-8" />
<meta content="width=device-width, initial-scale=1.0" name="viewport" />
<title>
Multi File Chat using RAG
</title>
<!-- <script src="https://cdn.tailwindcss.com"> -->
<script src="https://cdn.tailwindcss.com/3.4.15">
</script>
<link href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/5.15.3/css/all.min.css" rel="stylesheet" />
<link href="https://fonts.googleapis.com/css2?family=Roboto:wght@400;500;700&amp;display=swap" rel="stylesheet" />
<style>
body {
font-family: 'Roboto', sans-serif;
}
</style>
</head>
<body class="bg-gray-100 dark:bg-gray-900 text-gray-900 dark:text-gray-100">
<div id="loader" class="fixed top-0 left-0 w-full h-full"
style="background-color: rgba(50, 50, 50, 0.5);z-index: 1000;">
<div class="text-center" style="position: absolute;top: 50%;left: 50%;transform: translate(-50%, -50%);">
<div role="status">
<svg aria-hidden="true"
class="inline w-8 h-8 text-gray-200 animate-spin dark:text-gray-600 fill-blue-600"
viewBox="0 0 100 101" fill="none" xmlns="http://www.w3.org/2000/svg">
<path
d="M100 50.5908C100 78.2051 77.6142 100.591 50 100.591C22.3858 100.591 0 78.2051 0 50.5908C0 22.9766 22.3858 0.59082 50 0.59082C77.6142 0.59082 100 22.9766 100 50.5908ZM9.08144 50.5908C9.08144 73.1895 27.4013 91.5094 50 91.5094C72.5987 91.5094 90.9186 73.1895 90.9186 50.5908C90.9186 27.9921 72.5987 9.67226 50 9.67226C27.4013 9.67226 9.08144 27.9921 9.08144 50.5908Z"
fill="currentColor" />
<path
d="M93.9676 39.0409C96.393 38.4038 97.8624 35.9116 97.0079 33.5539C95.2932 28.8227 92.871 24.3692 89.8167 20.348C85.8452 15.1192 80.8826 10.7238 75.2124 7.41289C69.5422 4.10194 63.2754 1.94025 56.7698 1.05124C51.7666 0.367541 46.6976 0.446843 41.7345 1.27873C39.2613 1.69328 37.813 4.19778 38.4501 6.62326C39.0873 9.04874 41.5694 10.4717 44.0505 10.1071C47.8511 9.54855 51.7191 9.52689 55.5402 10.0491C60.8642 10.7766 65.9928 12.5457 70.6331 15.2552C75.2735 17.9648 79.3347 21.5619 82.5849 25.841C84.9175 28.9121 86.7997 32.2913 88.1811 35.8758C89.083 38.2158 91.5421 39.6781 93.9676 39.0409Z"
fill="currentFill" />
</svg>
<br>
<span id="loader_msg">Loading...</span>
</div>
</div>
</div>
<div class="flex flex-col h-screen">
<!-- Combined Header -->
<div class="bg-white dark:bg-gray-800 p-4 flex justify-between items-center w-full ">
<h2 class="text-xl font-bold">
Question Files
</h2>
<div class="flex items-center space-x-4">
<button class="bg-gray-200 dark:bg-gray-700 p-2 rounded-full hidden" id="theme-toggle">
<i class="fas fa-moon dark:hidden">
</i>
<i class="fas fa-sun hidden dark:block">
</i>
</button>
<button class="md:hidden bg-gray-200 dark:bg-gray-700 p-2 rounded-full" id="menu-toggle">
<i class="fas fa-bars">
</i>
</button>
</div>
</div>
<hr>
<div class="flex flex-1 overflow-hidden">
<!-- Sidebar -->
<div class="w-64 bg-white dark:bg-gray-800 p-4 flex flex-col absolute md:relative md:flex md:w-1/4 h-full z-10 transform -translate-x-full md:translate-x-0 transition-transform duration-300"
id="sidebar">
<div class="flex-1 overflow-y-auto">
<ul class="space-y-2" id="file-list">
<!-- <li class="flex justify-between items-center bg-gray-200 dark:bg-gray-700 p-2 rounded">
<span>
example-file.txt
</span>
<button class="text-red-500">
<i class="fas fa-times">
</i>
</button>
</li>
<li class="flex justify-between items-center bg-gray-200 dark:bg-gray-700 p-2 rounded">
<span>
another-file.pdf
</span>
<button class="text-red-500">
<i class="fas fa-times">
</i>
</button>
</li> -->
</ul>
<!-- no files added message -->
<p class="text-gray-500 text-center p-4 hidden" id="no-files-message">
No files Please Add Files to chat with them.
</p>
</div>
<div class="mt-4">
<input class="hidden" id="file-upload" type="file" accept=".pdf,.txt,.docx,.md,.html" multiple />
<label id="dropzone"
class="cursor-pointer border-2 border-dashed border-gray-300 dark:border-gray-600 p-4 rounded text-center block"
for="file-upload">
<p>
Drag &amp; Drop Files Here to Upload
</p>
<p class="text-sm text-gray-500 dark:text-gray-400">
Supported files: .pdf, .docx, .txt, .md, .html
</p>
<p class="mt-2 text-blue-500">
or click here to upload
</p>
</label>
</div>
</div>
<!-- Main Content -->
<div class="flex-1 flex flex-col h-full">
<!-- Chat Messages -->
<div class="flex-1 overflow-y-auto p-4 space-y-4" id="chat-messages">
<div class="flex items-start space-x-2">
<div class="bg-gray-200 dark:bg-gray-700 p-3 rounded-lg">
<p>
Upload your files, And you are ready to ask questions about them🫡.
</p>
</div>
</div>
<!-- <div class="flex items-start space-x-2">
<div class="bg-gray-200 dark:bg-gray-700 p-3 rounded-lg">
<p>
Hello! How can I assist you today?
</p>
<p class="text-sm text-gray-500 dark:text-gray-400 mt-2">
Sources: <a class="text-blue-500" href="#">source1.com</a>, <a class="text-blue-500"
href="#">source2.com</a>
</p>
</div>
</div>
<div class="flex items-start space-x-2 flex-row-reverse">
<div class="bg-blue-500 text-white p-3 rounded-lg">
<p>
I need help with my account.
</p>
</div>
</div> -->
</div>
<!-- Message Input -->
<form id="message-form">
<fieldset id="message-form-fields">
<div class="bg-white dark:bg-gray-800 p-4 flex items-center space-x-2">
<input
class="bg-white dark:bg-gray-800 flex-1 p-2 border border-gray-300 dark:border-gray-600 rounded text-black dark:text-white"
id="message-input" placeholder="Type your message..." type="text" />
<button type="submit"
class="bg-blue-500 text-white py-2 px-4 rounded hover:bg-blue-600 flex items-center">
<i class='fas fa-paper-plane mr-2'></i>
</button>
</div>
</fieldset>
</form>
</div>
</div>
</div>
<div id="infoPopup" class="fixed inset-0 flex items-center justify-center hidden bg-black bg-opacity-50">
<div class="bg-dark rounded-lg p-6 w-96 relative" style="background-color: #0f141b;">
<button id="closePopup" class="absolute top-2 right-2 text-gray-600 hover:text-gray-800">
<svg xmlns="http://www.w3.org/2000/svg" class="h-6 w-6" fill="none" viewBox="0 0 24 24"
stroke="currentColor">
<path stroke-linecap="round" stroke-linejoin="round" stroke-width="2" d="M6 18L18 6M6 6l12 12" />
</svg>
</button>
<h2 class="text-lg font-semibold mb-4">Source Information</h2>
<div id="popupContent" class="overflow-y-auto">
<p class="mb-2"><strong>Source:</strong> <span id="pop_up_source_name">Demonstrating Retrieval Augmented
Generation for enhanced
information retrieval.</span></p>
<p><strong>Content:</strong> <span id="pop_up_content">This project explores the implementation of RAG
to improve search accuracy
by combining pre-trained language models with external knowledge sources. It focuses on [mention
specific aspects of your project, e.g., efficiency, specific data sources, etc.].</span></p>
</div>
</div>
</div>
<script src="https://code.jquery.com/jquery-3.7.1.min.js"
integrity="sha256-/JqT3SQfawRcv/BIHPThkBvs0OEvtFFmqPF/lYI/Cxo=" crossorigin="anonymous"></script>
<script>
function set_loading(msg = "Loading...") {
$("#loader_msg").text(msg);
$("#loader").removeClass("hidden");
}
function remove_loading() {
$("#loader").addClass("hidden");
}
// File handling functionality
const dropZone = document.getElementById('dropzone'); // label of input
const fileInput = document.getElementById('file-upload'); // input element
const fileList = document.getElementById('file-list'); // div to display files uploaded
const chat = document.getElementById('chat-messages');
const infoPopup = document.getElementById('infoPopup');
const closePopup = document.getElementById('closePopup');
const popUpContent = document.getElementById('pop_up_content');
const popUpSourceName = document.getElementById('pop_up_source_name');
var file_list = [];
var file_count = 0;
// Check for saved user preference, if any, on load of the website
if (localStorage.getItem('theme') === 'dark') {
document.body.classList.add('dark');
} else {
document.body.classList.remove('dark');
}
document.getElementById('theme-toggle').addEventListener('click', function () {
if (document.body.classList.contains('dark')) {
document.body.classList.remove('dark');
localStorage.setItem('theme', 'light');
} else {
document.body.classList.add('dark');
localStorage.setItem('theme', 'dark');
}
});
// infoLink.addEventListener('click', (event) => {
// event.preventDefault(); // Prevent default anchor behavior
// infoPopup.classList.remove('hidden');
// });
function showPopUp(source, content) {
console.log("Showing pop up", source, content)
popUpSourceName.innerText = source;
popUpContent.innerText = content;
infoPopup.classList.remove('hidden');
}
closePopup.addEventListener('click', () => {
infoPopup.classList.add('hidden');
});
infoPopup.addEventListener('click', (event) => {
if (event.target === infoPopup) {
infoPopup.classList.add('hidden');
}
});
document.addEventListener('DOMContentLoaded', () => {
fileInput.addEventListener("change", () => {
const files = fileInput.files;
if (files.length > 0) {
handleFiles(files);
}
});
['dragenter', 'dragover', 'dragleave', 'drop'].forEach(eventName => {
dropZone.addEventListener(eventName, preventDefaults, false);
});
['dragenter', 'dragover'].forEach(eventName => {
dropZone.addEventListener(eventName, highlight, false);
});
['dragleave', 'drop'].forEach(eventName => {
dropZone.addEventListener(eventName, unhighlight, false);
});
['drop'].forEach(eventName => {
dropZone.addEventListener(eventName, handleDrop, false);
});
function preventDefaults(e) {
// console.log('prevent default');
e.preventDefault();
e.stopPropagation();
}
function highlight(e) {
// console.log('highlight');
dropZone.classList.add('bg-blue-500');
}
function unhighlight(e) {
// console.log('unhighlight');
dropZone.classList.remove('bg-blue-500');
}
function handleDrop(e) {
console.log("files drop");
const dt = e.dataTransfer;
const files = dt.files;
handleFiles(files);
}
function handleFiles(files) {
//console.log("Handle Files");
//console.log(files);
Array.from(files).forEach(file => {
file_list[file_count] = file;
const listItem = document.createElement('li');
listItem.id = `file_item_${file_count}`
listItem.className = 'flex justify-between items-center bg-gray-200 dark:bg-gray-700 p-2 rounded';
listItem.innerHTML = `
<span style="width: 90%;overflow: hidden;">${file.name}</span>
<button class="text-red-500 w-20" onclick="removeFile(${file_count})">
<i class="fas fa-times"></i>
</button>`;
fileList.appendChild(listItem);
set_loading("Uploading File " + file.name);
var formData = new FormData();
formData.append('file', file);
formData.append('file_count', file_count);
$.ajax({
url: '/upload-file',
type: 'POST',
contentType: false,
processData: false,
data: formData,
success: function (response) {
console.log(response);
},
error: function (xhr, status, error) {
console.log(error);
}
}).then(function () {
remove_loading();
})
file_count++;
});
manageEmptyFileList();
}
function add_user_message(msg) {
var item = document.createElement('div');
item.className = "flex items-start space-x-2 flex-row-reverse";
item.innerHTML = `<div class="bg-blue-500 text-white p-3 rounded-lg">
<p>
${msg}
</p>
</div>`;
chat.appendChild(item);
}
function add_ai_response(resp) {
var item = document.createElement('div');
item.className = "flex items-start space-x-2";
var responseDiv = document.createElement('div');
responseDiv.className = "bg-gray-200 dark:bg-gray-700 p-3 rounded-lg";
var answerPara = document.createElement('p');
answerPara.textContent = resp.answer;
responseDiv.appendChild(answerPara);
var sourcesPara = document.createElement('p');
sourcesPara.className = "text-sm text-gray-500 dark:text-gray-400 mt-2";
if (resp.is_relevant) {
sourcesPara.textContent = "Sources: ";
resp.sources.forEach(function (source) {
let sourceLink = document.createElement('button');
sourceLink.className = "text-blue-500 cursor-pointer hover:underline";
sourceLink.style.background = "#00000085";
sourceLink.style.borderRadius = "4px";
sourceLink.style.padding = "1px 3px";
sourceLink.style.margin = "2px";
sourceLink.textContent = source.metadata.file_name.substr(0, 20) + "...";
if (source.metadata.page !== undefined) {
sourceLink.textContent += ", Page:" + source.metadata.page;
}
// Store data in data attributes
sourceLink.dataset.sourceInfo = `${source.metadata.file_name.substr(0, 20)}... ${(source.metadata.page !== undefined) ? `(Page: ${source.metadata.page})` : ``}`;
sourceLink.dataset.pageContent = source.page_content.substr(0, 300) + "...";
console.log("Made source");
sourceLink.addEventListener('click', function () {
console.log("Clicked source");
showPopUp(this.dataset.sourceInfo, this.dataset.pageContent);
});
sourcesPara.appendChild(sourceLink);
sourcesPara.append(", "); // Add comma and space after each link
});
} else {
sourcesPara.textContent = "The Documents provided are not sufficient to answer the query provided, So response might not be accurate.";
}
responseDiv.appendChild(sourcesPara);
item.appendChild(responseDiv);
chat.appendChild(item);
}
// message send
document.getElementById('message-form').addEventListener('submit', function (e) {
e.preventDefault();
const messageInput = document.getElementById('message-input');
const message = messageInput.value.trim();
messageInput.value = "";
add_user_message(message)
if (message) {
$.ajax({
url: "/ask_query",
method: "POST",
data: {
query: message
},
success: function (response) {
resp = JSON.stringify(response)
resp = JSON.parse(resp)
console.log(resp);
add_ai_response(resp);
},
error: function (xhr, status, error) {
console.log(error);
}
})
}
});
});
function manageEmptyFileList() {
if (fileList.childElementCount == 0) {
fileList.classList.add('hidden');
document.getElementById("no-files-message").classList.remove('hidden');
// disable message form message-form
document.getElementById('message-form-fields').setAttribute('disabled', true);
} else {
document.getElementById("no-files-message").classList.add('hidden');
fileList.classList.remove('hidden');
document.getElementById('message-form-fields').removeAttribute('disabled');
}
}
function removeFile(index) {
console.log(index);
file_list[index] = null;
set_loading("Removing File");
// ajax call to remove file
$.ajax({
url: '/remove-file',
type: 'POST',
data: {
file_id: index
},
success: function (response) {
console.log(response);
},
error: function (xhr, status, error) {
console.log(error);
}
}).then(function () {
remove_loading();
});
document.getElementById(`file_item_${index}`).remove();
manageEmptyFileList();
}
function test(sec) {
console.log("Before Ajax");
set_loading();
// ajax call to remove file
$.ajax({
url: '/wait',
type: 'POST',
data: {
time: sec
},
success: function (response) {
console.log("Ajax Success");
console.log(response);
},
error: function (xhr, status, error) {
console.log(error);
}
}).then(function () {
console.log("Then Ajax");
remove_loading();
});
console.log("After Ajax")
}
document.addEventListener("DOMContentLoaded", function () {
set_loading("Loading Files");
fetch('/get-files')
.then(response => response.json())
.then(data => {
if (data.status == 'success') {
let max_file_cnt = 0;
Array.from(data.files).forEach(file => {
file_list[file.file_id] = { name: file.file_name };
const listItem = document.createElement('li');
listItem.id = `file_item_${file.file_id}`
listItem.className = 'flex justify-between items-center bg-gray-200 dark:bg-gray-700 p-2 rounded';
listItem.innerHTML = `
<span style="width: 90%;overflow: hidden;">${file.file_name}</span>
<button class="text-red-500 w-20" onclick="removeFile(${file.file_id})">
<i class="fas fa-times"></i>
</button>`;
fileList.appendChild(listItem);
max_file_cnt = max_file_cnt < file.file_id ? file.file_id : max_file_cnt;
});
file_count = max_file_cnt + 1;
manageEmptyFileList();
}
}).then(() => {
remove_loading();
})
.catch(error => console.error("Error fetching files:", error));
});
// Toggle sidebar for mobile view
document.getElementById('menu-toggle').addEventListener('click', function () {
const sidebar = document.getElementById('sidebar');
sidebar.classList.toggle('-translate-x-full');
});
window.onload = function () {
manageEmptyFileList();
remove_loading();
}
</script>
</body>
</html>