ChatDev/chat_visualizer/static/chain_visualizer.html
2023-12-16 19:38:58 +05:30

165 lines
5.5 KiB
HTML

<!DOCTYPE html>
<html>
<head>
<title>ChatChain Visualizer</title>
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/materialize/1.0.0/css/materialize.min.css">
<style>
body {
display: flex;
justify-content: center;
align-items: center;
height: 100vh;
}
#visualization {
display: flex;
flex-wrap: nowrap;
overflow-x: visible;
overflow-y: visible;
max-width: 1800px;
max-height: 1600px;
margin: 20px;
}
.card {
margin-right: 10px;
display: inline-block;
min-width: 300px;
vertical-align: top;
font-size: 14px;
}
.simple-phase {
background-color: #E8ECEB; /* Light Blue for SimplePhase */
}
.composed-phase {
background-color: #A3B4C8; /* Light Red for ComposedPhase */
}
.nested-simple-phase {
background-color: #E3DCD2; /* Light Yellow for SimplePhase within ComposedPhase */
}
.card-content {
padding: 10px;
}
</style>
</head>
<body>
<div class="container">
<h2>ChatChain Visualizer</h2>
<p>Select your ChatChainConfig.json under CompanyConfig/ to visualize</p>
<input type="file" id="fileInput" accept=".json">
<button id="exportButton">Export as Image</button>
<div id="visualization"></div>
</div>
<script src="https://cdnjs.cloudflare.com/ajax/libs/html2canvas/0.5.0-beta4/html2canvas.min.js"></script>
<script>
document.getElementById('fileInput').addEventListener('change', handleFileSelect, false);
document.getElementById('exportButton').addEventListener('click', exportAsImage, false);
function handleFileSelect(event) {
const file = event.target.files[0];
if (!file) return;
const reader = new FileReader();
reader.onload = function(event) {
try {
const jsonContent = JSON.parse(event.target.result);
visualizeChain(jsonContent.chain);
} catch (error) {
alert('Error parsing JSON file.');
}
};
reader.readAsText(file);
}
function createCard(element) {
const card = document.createElement('div');
card.className = 'card';
const cardContent = document.createElement('div');
cardContent.className = 'card-content';
if (element.phaseType === "ComposedPhase") {
delete element.Composition;
}
const phase = document.createElement('span');
phase.innerHTML = `<strong>PhaseName: </strong>${element.phase || 'No PhaseName'}`;
const phaseType = document.createElement('p');
phaseType.innerHTML = `<strong>PhaseType: </strong>${element.phaseType || 'No phaseType'}`;
delete element.phase;
delete element.phaseType;
const jsonContent = document.createElement('pre');
jsonContent.innerText = JSON.stringify(element, null, 2);
cardContent.appendChild(phase);
cardContent.appendChild(phaseType);
cardContent.appendChild(jsonContent);
card.appendChild(cardContent);
return card;
}
function visualizeChain(chain) {
const visualization = document.getElementById('visualization');
visualization.innerHTML = '';
chain.forEach(element => {
if (element.phaseType === "ComposedPhase") {
const composition = element.Composition || [];
const card = createCard(element);
const nestedCards = composition.map(composedElement => {
return createCard(composedElement);
});
const nestedCardWrapper = document.createElement('div');
nestedCardWrapper.style.marginTop = '10px';
nestedCards.forEach(nestedCard => {
nestedCard.classList.add('nested-simple-phase');
nestedCardWrapper.appendChild(nestedCard);
});
card.classList.add('composed-phase');
card.appendChild(nestedCardWrapper);
visualization.appendChild(card);
} else {
const card = createCard(element);
card.classList.add('simple-phase');
visualization.appendChild(card);
}
});
}
function exportAsImage() {
const visualization = document.getElementById('visualization');
const totalWidth = visualization.scrollWidth;
const totalHeight = visualization.scrollHeight;
console.log(totalWidth, totalHeight)
const canvas = document.createElement('canvas');
canvas.width = totalWidth + 100;
canvas.height = totalHeight + 100;
html2canvas(visualization, { scrollX: 0, scrollY: 0, width: totalWidth, height: totalHeight, useCORS: true }).then(canvas => {
const link = document.createElement('a');
link.href = canvas.toDataURL();
link.download = 'ChatChain_Visualization.png';
link.click();
});
}
</script>
</body>
</html>