Chat with
@@ -451,16 +557,24 @@
document.getElementById('chat-input').value = '';
addMessageToChat('You: ' + message);
+ // Check if we're using hypernetwork and include context
+ let formData = new FormData();
+ formData.append('message', message);
+
+ // Get the context text if hypernetwork is being used
+ const contextInput = document.getElementById('context-input');
+ if (document.getElementById('chat-model-name').textContent.includes('Hypernetwork') && contextInput) {
+ formData.append('context', contextInput.value);
+ }
+
fetch('/chat', {
method: 'POST',
- headers: {
- 'Content-Type': 'application/x-www-form-urlencoded',
- },
- body: 'message=' + encodeURIComponent(message)
+ body: formData
})
.then(response => response.json())
.then(data => {
- addMessageToChat('{{ model_name }}: ' + data.response);
+ const modelName = document.getElementById('chat-model-name').textContent || '{{ model_name }}';
+ addMessageToChat(modelName + ': ' + data.response);
})
.catch(error => {
console.error("Error during chat fetch:", error);
@@ -482,6 +596,80 @@
});
}
+ // Handle hypernetwork checkpoint loading
+ document.addEventListener('DOMContentLoaded', function () {
+ const applyHypernetworkButton = document.getElementById('apply-hypernetwork-button');
+ if (applyHypernetworkButton) {
+ applyHypernetworkButton.addEventListener('click', function () {
+ const checkpoint = document.getElementById('checkpoint-select').value;
+ const context = document.getElementById('context-input').value;
+
+ if (!checkpoint) {
+ alert('Please select a checkpoint first.');
+ return;
+ }
+
+ if (!context.trim()) {
+ if (!confirm('The context field is empty. Are you sure you want to continue without context?')) {
+ return;
+ }
+ }
+
+ const statusEl = document.getElementById('hypernetwork-status');
+ statusEl.textContent = 'Loading checkpoint...';
+ statusEl.style.color = '#1565c0'; // Blue to indicate in progress
+
+ // Disable the button while loading
+ applyHypernetworkButton.disabled = true;
+ applyHypernetworkButton.textContent = 'Loading...';
+
+ fetch('/load_checkpoint', {
+ method: 'POST',
+ headers: {
+ 'Content-Type': 'application/x-www-form-urlencoded',
+ },
+ body: 'run=' + encodeURIComponent('{{ run }}') +
+ '&checkpoint=' + encodeURIComponent(checkpoint) +
+ '&context=' + encodeURIComponent(context)
+ })
+ .then(response => response.json())
+ .then(data => {
+ // Re-enable the button
+ applyHypernetworkButton.disabled = false;
+ applyHypernetworkButton.textContent = 'Apply Hypernetwork';
+
+ if (data.success) {
+ statusEl.textContent = 'Success: ' + data.message;
+ statusEl.style.color = 'green';
+
+ // Show the chat container if it's hidden
+ document.getElementById('chat-container').style.display = 'block';
+ document.getElementById('chat-input').disabled = false;
+ document.getElementById('chat-model-name').textContent =
+ 'Hypernetwork: ' + checkpoint;
+
+ // Clear any existing chat and add an instruction
+ document.getElementById('chat-output').innerHTML = '';
+ addMessageToChat('System: Hypernetwork applied successfully. Your messages will now be processed through the context-aware model.');
+
+ } else {
+ statusEl.textContent = 'Error: ' + data.error;
+ statusEl.style.color = 'red';
+ }
+ })
+ .catch(error => {
+ // Re-enable the button on error
+ applyHypernetworkButton.disabled = false;
+ applyHypernetworkButton.textContent = 'Apply Hypernetwork';
+
+ console.error('Error loading checkpoint:', error);
+ statusEl.textContent = 'Error: ' + error.message;
+ statusEl.style.color = 'red';
+ });
+ });
+ }
+ });
+
// Call initializeChat when chat is loaded
document.getElementById('load-chat-button').addEventListener('click', function () {
fetch('/load_model', {
@@ -508,8 +696,17 @@
document.getElementById('chat-container').style.display = 'block';
document.getElementById('chat-input').disabled = false;
+ // Show the hypernetwork container now that base model is loaded
+ const hypernetworkContainer = document.getElementById('hypernetwork-container');
+ if (hypernetworkContainer) {
+ hypernetworkContainer.style.display = 'block';
+ }
+
console.log("Chat interface loaded successfully.");
initializeChat();
+
+ // Add a message about UI flow
+ addMessageToChat('System: Base model loaded successfully. You can now chat directly or apply a hypernetwork with context information.');
}
})
.catch(error => {
@@ -542,6 +739,40 @@
}
});
+ // Add the reset chat functionality
+ document.addEventListener('DOMContentLoaded', function () {
+ const resetChatButton = document.getElementById('reset-chat-button');
+ if (resetChatButton) {
+ resetChatButton.addEventListener('click', function () {
+ // Get the current system message
+ const systemMsg = document.getElementById('system-msg').value;
+
+ // Clear the chat output display
+ document.getElementById('chat-output').innerHTML = '';
+
+ fetch('/reset_chat', {
+ method: 'POST',
+ headers: {
+ 'Content-Type': 'application/x-www-form-urlencoded',
+ },
+ body: 'system_msg=' + encodeURIComponent(systemMsg)
+ })
+ .then(response => response.json())
+ .then(data => {
+ if (data.success) {
+ console.log('Chat history reset successfully');
+ addMessageToChat('System: Chat history has been reset.');
+ } else {
+ console.error('Failed to reset chat history');
+ }
+ })
+ .catch(error => {
+ console.error('Error resetting chat history:', error);
+ });
+ });
+ }
+ });
+
{% endblock %}
\ No newline at end of file