datamule 1.0.2__py3-none-any.whl → 1.0.6__py3-none-any.whl

datamule/mulebot/mulebot_server/static/scripts/artifacts.js

@@ -1,68 +0,0 @@
-// artifacts.js
-import { renderTableArtifact } from './tableArtifacts.js';
-import { renderListArtifact } from './listArtifacts.js';
-import { renderFilingArtifact } from './filingArtifacts.js';
-
-export let artifactContent = null;
-export let artifactContainer = null;
-export let suggestionContainer = null;
-export let toggleArtifactsBtn = null;
-export let allArtifacts = [];
-
-export function initializeArtifacts() {
-    artifactContent = document.getElementById('artifact-content');
-    artifactContainer = document.getElementById('artifact-container');
-    suggestionContainer = document.querySelector('.suggestion-box');
-    toggleArtifactsBtn = document.getElementById('toggle-artifacts');
-
-    if (toggleArtifactsBtn) {
-        toggleArtifactsBtn.addEventListener('click', toggleArtifacts);
-    }
-
-    // Initially hide artifacts
-    hideArtifacts();
-}
-
-export function renderArtifact(artifactData, artifactType) {
-    if (artifactType === 'artifact-table') {
-        const newArtifacts = Array.isArray(artifactData) ? artifactData : [artifactData];
-        newArtifacts.forEach(artifact => {
-            artifact.type = artifactType;
-        });
-        allArtifacts = [...allArtifacts, ...newArtifacts];
-        renderTableArtifact(artifactData[0]);
-    }
-    else if (artifactType === 'artifact-list') {
-        renderListArtifact(artifactData);
-    }
-    else if (artifactType === 'artifact-filing') {
-        renderFilingArtifact(artifactData);
-    }
-    else {
-        console.log('Unsupported artifact type:', artifactType);
-    }
-}
-
-export function showArtifacts() {
-    if (artifactContainer && suggestionContainer) {
-        artifactContainer.style.display = 'block';
-        suggestionContainer.style.display = 'none';
-        if (toggleArtifactsBtn) toggleArtifactsBtn.textContent = 'Hide Artifacts';
-    }
-}
-
-export function hideArtifacts() {
-    if (artifactContainer && suggestionContainer) {
-        artifactContainer.style.display = 'none';
-        suggestionContainer.style.display = 'block';
-        if (toggleArtifactsBtn) toggleArtifactsBtn.textContent = 'Show Artifacts';
-    }
-}
-
-function toggleArtifacts() {
-    if (artifactContainer.style.display === 'none') {
-        showArtifacts();
-    } else {
-        hideArtifacts();
-    }
-}

datamule/mulebot/mulebot_server/static/scripts/chat.js

@@ -1,92 +0,0 @@
-// chat.js
-import { renderArtifact, showArtifacts } from './artifacts.js';
-
-let thinkingIndicator = null;
-let isHandlingResponse = false;
-
-export function appendMessage(sender, message) {
-    console.log(`Appending message from ${sender}: ${message}`);
-    const chatContainer = document.getElementById('chat-container');
-    const messageElement = document.createElement('div');
-    messageElement.innerHTML = `<strong>${sender}:</strong> ${message}`;
-    chatContainer.appendChild(messageElement);
-    scrollChatToBottom();
-}
-
-function scrollChatToBottom() {
-    const chatContainer = document.getElementById('chat-container');
-    chatContainer.scrollTop = chatContainer.scrollHeight;
-}
-
-export function showThinkingIndicator() {
-    if (!thinkingIndicator) {
-        thinkingIndicator = document.createElement('div');
-        thinkingIndicator.className = 'thinking-indicator';
-        thinkingIndicator.innerHTML = '<span>Bot is thinking</span><span class="dot-animation">...</span>';
-        document.getElementById('chat-container').appendChild(thinkingIndicator);
-    }
-    thinkingIndicator.style.display = 'block';
-    scrollChatToBottom();
-}
-
-export function hideThinkingIndicator() {
-    if (thinkingIndicator) {
-        thinkingIndicator.style.display = 'none';
-    }
-}
-
-export async function sendMessage(message) {
-    console.log(`Sending message: ${message}`);
-    showThinkingIndicator();
-    try {
-        const response = await fetch('/chat', {
-            method: 'POST',
-            headers: {
-                'Content-Type': 'application/json',
-            },
-            body: JSON.stringify({ message }),
-        });
-        const data = await response.json();
-        console.log('Received response:', data);
-        return data;
-    } finally {
-        hideThinkingIndicator();
-    }
-}
-
-export function handleResponse(response) {
-    if (isHandlingResponse) {
-        console.log('Already handling a response, skipping.');
-        return;
-    }
-    isHandlingResponse = true;
-    console.log('Handling response:', response);
-
-    try {
-        if (response.response.type === 'text') {
-            appendMessage('Bot', response.response.content);
-        } else if (response.response.type === 'artifact') {
-            let artifactType = response.response.artifact_type;
-            let artifactContent = response.response.content;
-
-            if (artifactType === 'artifact-filing') {
-                artifactContent = {
-                    content: response.response.content,
-                    data: response.response.data,
-                    section_id: response.response.section_id
-                };
-            }
-
-            appendMessage('Bot', `I have prepared an ${artifactType} for you. Please check the artifact view.`);
-            renderArtifact(artifactContent, artifactType);
-            showArtifacts();
-        } else {
-            appendMessage('Bot', 'I have received a response, but it is not a supported type.');
-        }
-    } finally {
-        isHandlingResponse = false;
-    }
-}
-
-// Make sendMessage globally accessible
-window.sendMessage = sendMessage;

datamule/mulebot/mulebot_server/static/scripts/filingArtifacts.js

@@ -1,56 +0,0 @@
-// filingArtifacts.js
-export function renderFilingArtifact(artifactData) {
-    const { content: html, data, section_id } = artifactData;
-
-    const container = document.createElement('div');
-    container.className = 'filing-container';
-    container.style.width = '100%';
-    container.style.height = '600px';
-
-    const iframe = document.createElement('iframe');
-    iframe.srcdoc = html;
-    iframe.style.width = '100%';
-    iframe.style.height = '100%';
-    iframe.style.border = 'none';
-
-    container.appendChild(iframe);
-
-    const artifactContent = document.getElementById('artifact-content');
-    if (artifactContent) {
-        artifactContent.innerHTML = '';
-        artifactContent.appendChild(container);
-    }
-
-    iframe.onload = () => {
-        if (section_id) {
-            const iframeDocument = iframe.contentDocument || iframe.contentWindow.document;
-            const targetElement = iframeDocument.getElementById(section_id);
-            if (targetElement) {
-                targetElement.scrollIntoView({ behavior: 'smooth', block: 'start' });
-            }
-        }
-    };
-
-    const buttonContainer = document.createElement('div');
-    buttonContainer.className = 'mt-3';
-
-    const downloadHtmlBtn = document.createElement('button');
-    downloadHtmlBtn.className = 'btn btn-secondary me-2';
-    downloadHtmlBtn.textContent = 'Download Filing HTML';
-    downloadHtmlBtn.onclick = () => {
-        const blob = new Blob([html], { type: 'text/html' });
-        saveAs(blob, 'filing.html');
-    };
-    buttonContainer.appendChild(downloadHtmlBtn);
-
-    const downloadJsonBtn = document.createElement('button');
-    downloadJsonBtn.className = 'btn btn-secondary';
-    downloadJsonBtn.textContent = 'Download Filing Data (JSON)';
-    downloadJsonBtn.onclick = () => {
-        const blob = new Blob([JSON.stringify(data, null, 2)], { type: 'application/json' });
-        saveAs(blob, 'filing_data.json');
-    };
-    buttonContainer.appendChild(downloadJsonBtn);
-
-    artifactContent.appendChild(buttonContainer);
-}

datamule/mulebot/mulebot_server/static/scripts/listArtifacts.js

@@ -1,15 +0,0 @@
-// listArtifacts.js
-export function renderListArtifact(listData) {
-    let html = `<h4>URLs</h4><ul class="list-group">`;
-
-    listData.forEach(url => {
-        html += `<li class="list-group-item"><a href="${url}" target="_blank">${url}</a></li>`;
-    });
-
-    html += '</ul>';
-
-    const artifactContent = document.getElementById('artifact-content');
-    if (artifactContent) {
-        artifactContent.innerHTML = html;
-    }
-}

datamule/mulebot/mulebot_server/static/scripts/main.js

@@ -1,57 +0,0 @@
-// main.js
-import { appendMessage, sendMessage, handleResponse, showThinkingIndicator, hideThinkingIndicator } from './chat.js';
-import { initializeArtifacts } from './artifacts.js';
-import { handleDocumentClick } from './tableArtifacts.js';
-import { initializeSuggestions } from './suggestions.js';
-import { handlePrefilledPrompt } from './prefilledPrompt.js';
-
-let chatInitialized = false;
-
-function initializeChat() {
-    if (chatInitialized) return;
-    chatInitialized = true;
-
-    console.log('Initializing chat');
-
-    initializeArtifacts();
-    initializeSuggestions();
-
-    const chatForm = document.getElementById('chat-form');
-    const userInput = document.getElementById('user-input');
-
-    if (chatForm) {
-        chatForm.addEventListener('submit', async (e) => {
-            e.preventDefault();
-            console.log('Form submitted');
-            const message = userInput.value.trim();
-            if (message) {
-                appendMessage('You', message);
-                userInput.value = '';
-                showThinkingIndicator();
-                try {
-                    const response = await sendMessage(message);
-                    handleResponse(response);
-                } catch (error) {
-                    console.error('Error processing message:', error);
-                } finally {
-                    hideThinkingIndicator();
-                }
-            }
-        });
-    }
-
-    document.addEventListener('click', handleDocumentClick);
-}
-
-// Wait for the DOM to be fully loaded before initializing
-if (document.readyState === 'loading') {
-    document.addEventListener('DOMContentLoaded', initializeChat);
-} else {
-    initializeChat();
-}
-
-// Add this new event listener for the window load event
-window.addEventListener('load', () => {
-    console.log('Window fully loaded, handling prefilled prompt');
-    handlePrefilledPrompt();
-});

datamule/mulebot/mulebot_server/static/scripts/prefilledPrompt.js

@@ -1,27 +0,0 @@
-// prefilledPrompt.js
-import { sendMessage, handleResponse, appendMessage } from './chat.js';
-
-export function handlePrefilledPrompt() {
-    const urlParams = new URLSearchParams(window.location.search);
-    const prefilled_prompt = urlParams.get('prompt');
-    if (prefilled_prompt) {
-        const userInput = document.getElementById('user-input');
-        if (userInput) {
-            userInput.value = prefilled_prompt;
-            appendMessage('You', prefilled_prompt); // Show the user's message in the chat
-        }
-        if (typeof sendMessage === 'function') {
-            sendMessage(prefilled_prompt).then(response => {
-                handleResponse(response);
-                if (userInput) {
-                    userInput.value = ''; // Clear the input field after sending
-                }
-            }).catch(error => {
-                console.error('Error processing prefilled prompt:', error);
-                if (userInput) {
-                    userInput.value = ''; // Clear the input field even if there's an error
-                }
-            });
-        }
-    }
-}

datamule/mulebot/mulebot_server/static/scripts/suggestions.js

@@ -1,47 +0,0 @@
-// suggestions.js
-import { appendMessage, sendMessage, handleResponse, showThinkingIndicator, hideThinkingIndicator } from './chat.js';
-import { renderArtifact, showArtifacts } from './artifacts.js';
-
-export function initializeSuggestions() {
-    const userInput = document.getElementById('user-input');
-    const chatForm = document.getElementById('chat-form');
-    const suggestionItems = document.querySelectorAll('.suggestion-item');
-
-    // Function to handle suggestion click
-    function handleSuggestionClick(event) {
-        const suggestionText = event.target.textContent;
-        userInput.value = suggestionText;
-        executeChatRequest(suggestionText);
-        userInput.value = ''; // Clear the input after executing the request
-    }
-
-    // Add click event listeners to all suggestion items
-    suggestionItems.forEach(item => {
-        item.addEventListener('click', handleSuggestionClick);
-    });
-
-    // Function to execute chat request
-    async function executeChatRequest(message) {
-        try {
-            appendMessage('You', message);
-            showThinkingIndicator();
-            const response = await sendMessage(message);
-            handleResponse(response);
-        } catch (error) {
-            console.error('Error executing chat request:', error);
-            appendMessage('Bot', 'Sorry, there was an error processing your request.');
-        } finally {
-            hideThinkingIndicator();
-        }
-    }
-
-    // Prevent default form submission
-    chatForm.addEventListener('submit', function (event) {
-        event.preventDefault();
-        const message = userInput.value.trim();
-        if (message) {
-            executeChatRequest(message);
-            userInput.value = ''; // Clear the input after executing the request
-        }
-    });
-}

datamule/mulebot/mulebot_server/static/scripts/tableArtifacts.js

@@ -1,129 +0,0 @@
-// tableArtifacts.js
-import { renderMetadata, downloadCSV } from './utils.js';
-import { allArtifacts } from './artifacts.js';
-
-export function renderTableArtifact(tableData) {
-    let html = `
-        <div class="mb-3 select-wrapper">
-            <input type="text" id="artifact-select" class="form-control" placeholder="Select a table..." value="${tableData.fact}">
-            <div id="autocomplete-list" class="autocomplete-items"></div>
-        </div>
-        <div class="mb-3">
-            <button id="download-csv" class="btn btn-secondary me-2">Download Selected Table (CSV)</button>
-            <button id="download-all-zip" class="btn btn-secondary">Download All Tables (ZIP)</button>
-        </div>
-        <div id="metadata-content"></div>
-    `;
-
-    if (tableData.table && tableData.table.length > 0) {
-        html += '<table class="table table-striped mt-3"><thead><tr>';
-        Object.keys(tableData.table[0]).forEach(header => {
-            html += `<th>${header}</th>`;
-        });
-        html += '</tr></thead><tbody>';
-        tableData.table.forEach(row => {
-            html += '<tr>';
-            Object.values(row).forEach(cell => {
-                html += `<td>${cell}</td>`;
-            });
-            html += '</tr>';
-        });
-        html += '</tbody></table>';
-    } else {
-        html += '<p>No table data available.</p>';
-    }
-
-    const artifactContent = document.getElementById('artifact-content');
-    if (artifactContent) {
-        artifactContent.innerHTML = html;
-    }
-    renderMetadata(tableData);
-    setupTableEventListeners(tableData);
-}
-
-function setupTableEventListeners(tableData) {
-    const artifactSelect = document.getElementById('artifact-select');
-    const downloadCsvBtn = document.getElementById('download-csv');
-    const downloadAllZipBtn = document.getElementById('download-all-zip');
-
-    if (artifactSelect) {
-        artifactSelect.addEventListener('input', handleArtifactSelectInput);
-        artifactSelect.addEventListener('focus', handleArtifactSelectFocus);
-    }
-
-    if (downloadCsvBtn) {
-        downloadCsvBtn.addEventListener('click', () => downloadCSV(tableData));
-    }
-
-    if (downloadAllZipBtn) {
-        downloadAllZipBtn.addEventListener('click', handleDownloadAllZip);
-    }
-}
-
-export function handleArtifactSelectInput(e) {
-    const inputValue = e.target.value.toLowerCase().trim();
-    const filteredTables = allArtifacts.filter(table =>
-        table.fact.toLowerCase().includes(inputValue)
-    );
-    createAutocompleteList(filteredTables);
-}
-
-export function handleArtifactSelectFocus() {
-    createAutocompleteList(allArtifacts);
-}
-
-function createAutocompleteList(tables) {
-    const autocompleteList = document.getElementById('autocomplete-list');
-    if (!autocompleteList) return;
-
-    autocompleteList.innerHTML = '';
-    autocompleteList.style.display = 'block';
-
-    if (tables.length === 0) {
-        autocompleteList.innerHTML = '<div style="color: #999;">No matching tables found</div>';
-    } else {
-        tables.forEach(table => {
-            const div = document.createElement("div");
-            div.textContent = table.fact;
-            div.addEventListener("click", function () {
-                document.getElementById('artifact-select').value = this.textContent;
-                renderTableArtifact(table);
-                closeAutocompleteList();
-            });
-            autocompleteList.appendChild(div);
-        });
-    }
-}
-
-export function handleDocumentClick(e) {
-    if (e.target.id !== 'artifact-select') {
-        closeAutocompleteList();
-    }
-}
-
-function closeAutocompleteList() {
-    const autocompleteList = document.getElementById('autocomplete-list');
-    if (autocompleteList) {
-        autocompleteList.style.display = 'none';
-    }
-}
-
-async function handleDownloadAllZip() {
-    const tables = allArtifacts.filter(artifact => artifact.type === 'artifact-table');
-    if (tables.length === 0) {
-        alert('No tables available to download.');
-        return;
-    }
-    const zip = new JSZip();
-    tables.forEach(table => {
-        let csv = '';
-        const headers = Object.keys(table.table[0]);
-        csv += headers.join(',') + '\n';
-        table.table.forEach(row => {
-            csv += Object.values(row).join(',') + '\n';
-        });
-        zip.file(`${table.fact}.csv`, csv);
-    });
-    const content = await zip.generateAsync({ type: "blob" });
-    saveAs(content, "all_tables.zip");
-}

datamule/mulebot/mulebot_server/static/scripts/utils.js

@@ -1,28 +0,0 @@
-// utils.js
-export function renderMetadata(artifactData) {
-    const metadataHtml = `
-        <div class="card mb-3">
-            <div class="card-body">
-                <h5 class="card-title">Metadata</h5>
-                <p><strong>Fact:</strong> ${artifactData.fact}</p>
-                ${artifactData.cik ? `<p><strong>CIK:</strong> ${artifactData.cik}</p>` : ''}
-                ${artifactData.category ? `<p><strong>Category:</strong> ${artifactData.category}</p>` : ''}
-                ${artifactData.label ? `<p><strong>Label:</strong> ${artifactData.label}</p>` : ''}
-                ${artifactData.description ? `<p><strong>Description:</strong> ${artifactData.description}</p>` : ''}
-                ${artifactData.unit ? `<p><strong>Unit:</strong> ${artifactData.unit}</p>` : ''}
-            </div>
-        </div>
-    `;
-    document.getElementById('metadata-content').innerHTML = metadataHtml;
-}
-
-export function downloadCSV(table) {
-    let csv = '';
-    const headers = Object.keys(table.table[0]);
-    csv += headers.join(',') + '\n';
-    table.table.forEach(row => {
-        csv += Object.values(row).join(',') + '\n';
-    });
-    const blob = new Blob([csv], { type: 'text/csv;charset=utf-8;' });
-    saveAs(blob, `${table.fact}.csv`);
-}

datamule/mulebot/mulebot_server/templates/chat-minimalist.html

@@ -1,91 +0,0 @@
-<!DOCTYPE html>
-<html lang="en">
-
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Mulebot</title>
-    <link href="https://cdnjs.cloudflare.com/ajax/libs/bootstrap/5.1.3/css/bootstrap.min.css" rel="stylesheet">
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/FileSaver.js/2.0.5/FileSaver.min.js"></script>
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/jszip/3.7.1/jszip.min.js"></script>
-    <link rel="stylesheet" href="{{ url_for('static', filename='css/minimalist.css') }}">
-</head>
-
-<body>
-    <div class="container mt-5">
-        <div class="d-flex justify-content-between align-items-center mb-4">
-            <h1>Mulebot</h1>
-            <div>
-                <a href="https://github.com/john-friedman/datamule-python" target="_blank" class="social-btn">
-                    <svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" fill="currentColor"
-                        class="bi bi-github" viewBox="0 0 16 16">
-                        <path
-                            d="M8 0C3.58 0 0 3.58 0 8c0 3.54 2.29 6.53 5.47 7.59.4.07.55-.17.55-.38 0-.19-.01-.82-.01-1.49-2.01.37-2.53-.49-2.69-.94-.09-.23-.48-.94-.82-1.13-.28-.15-.68-.52-.01-.53.63-.01 1.08.58 1.23.82.72 1.21 1.87.87 2.33.66.07-.52.28-.87.51-1.07-1.78-.2-3.64-.89-3.64-3.95 0-.87.31-1.59.82-2.15-.08-.2-.36-1.02.08-2.12 0 0 .67-.21 2.2.82.64-.18 1.32-.27 2-.27.68 0 1.36.09 2 .27 1.53-1.04 2.2-.82 2.2-.82.44 1.1.16 1.92.08 2.12.51.56.82 1.27.82 2.15 0 3.07-1.87 3.75-3.65 3.95.29.25.54.73.54 1.48 0 1.07-.01 1.93-.01 2.2 0 .21.15.46.55.38A8.012 8.012 0 0 0 16 8c0-4.42-3.58-8-8-8z" />
-                    </svg>
-                    GitHub
-                </a>
-                <a href="https://datamule.xyz" target="_blank" class="social-btn">
-                    <svg xmlns="http://www.w3.org/2000/svg" width="16" height="16" fill="currentColor"
-                        viewBox="0 0 16 16">
-                        <path
-                            d="M0 8a8 8 0 1 1 16 0A8 8 0 0 1 0 8zm7.5-6.923c-.67.204-1.335.82-1.887 1.855-.143.268-.276.56-.395.872.705.157 1.472.257 2.282.287V1.077zM4.249 3.539c.142-.384.304-.744.481-1.078a6.7 6.7 0 0 1 .597-.933A7.01 7.01 0 0 0 3.051 3.05c.362.184.763.349 1.198.49zM3.509 7.5c.036-1.07.188-2.087.436-3.008a9.124 9.124 0 0 1-1.565-.667A6.964 6.964 0 0 0 1.018 7.5h2.49zm1.4-2.741a12.344 12.344 0 0 0-.4 2.741H7.5V5.091c-.91-.03-1.783-.145-2.591-.332zM8.5 5.09V7.5h2.99a12.342 12.342 0 0 0-.399-2.741c-.808.187-1.681.301-2.591.332zM4.51 8.5c.035.987.176 1.914.399 2.741A13.612 13.612 0 0 1 7.5 10.91V8.5H4.51zm3.99 0v2.409c.91.03 1.783.145 2.591.332.223-.827.364-1.754.4-2.741H8.5zm-3.282 3.696c.12.312.252.604.395.872.552 1.035 1.218 1.65 1.887 1.855V11.91c-.81.03-1.577.13-2.282.287zm.11 2.276a6.696 6.696 0 0 1-.598-.933 8.853 8.853 0 0 1-.481-1.079 8.38 8.38 0 0 0-1.198.49 7.01 7.01 0 0 0 2.276 1.522zm-1.383-2.964A13.36 13.36 0 0 1 3.508 8.5h-2.49a6.963 6.963 0 0 0 1.362 3.675c.47-.258.995-.482 1.565-.667zm6.728 2.964a7.009 7.009 0 0 0 2.275-1.521 8.376 8.376 0 0 0-1.197-.49 8.853 8.853 0 0 1-.481 1.078 6.688 6.688 0 0 1-.597.933zM8.5 11.909v3.014c.67-.204 1.335-.82 1.887-1.855.143-.268.276-.56.395-.872A12.63 12.63 0 0 0 8.5 11.91zm3.555-.401c.57.185 1.095.409 1.565.667A6.963 6.963 0 0 0 14.982 8.5h-2.49a13.36 13.36 0 0 1-.437 3.008zM14.982 7.5a6.963 6.963 0 0 0-1.362-3.675c-.47.258-.995.482-1.565.667.248.92.4 1.938.437 3.008h2.49zM11.27 2.461c.177.334.339.694.482 1.078a8.368 8.368 0 0 0 1.196-.49 7.01 7.01 0 0 0-2.275-1.52c.218.283.418.597.597.932zm-.488 1.343a7.765 7.765 0 0 0-.395-.872C9.835 1.897 9.17 1.282 8.5 1.077V4.09c.81-.03 1.577-.13 2.282-.287z" />
-                    </svg>
-                    DataMule Website
-                </a>
-            </div>
-        </div>
-        <div class="row">
-            <div class="col-md-6">
-                <div class="card mb-3">
-                    <div class="card-body">
-                        <div id="chat-outer-container">
-                            <div id="chat-container" class="mb-3"></div>
-                            <div id="thinking-indicator" class="thinking-indicator">
-                                <span>Bot is thinking</span><span class="dot-animation">...</span>
-                            </div>
-                        </div>
-                        <form id="chat-form">
-                            <div class="input-group">
-                                <input type="text" id="user-input" class="form-control"
-                                    placeholder="Type your message...">
-                                <button type="submit" class="btn btn-primary">Send</button>
-                            </div>
-                        </form>
-                    </div>
-                </div>
-            </div>
-            <div class="col-md-6">
-                <button id="toggle-artifacts" class="btn btn-secondary mb-3">Show Artifacts</button>
-                <div id="artifact-container" class="card mb-3" style="display: none;">
-                    <div class="card-body">
-                        <div id="artifact-content"></div>
-                    </div>
-                </div>
-                <div class="suggestion-box">
-                    <h5 class="mb-2">Suggested Commands:</h5>
-                    <ul class="list-unstyled">
-                        <li class="suggestion-item">Get all company facts for Tesla</li>
-                        <li class="suggestion-item">Get Ford's CIK</li>
-                        <li class="suggestion-item">Get all 10-K urls for META</li>
-                        <li class="suggestion-item">Get the management discussion and analysis section from
-                            https://www.sec.gov/Archives/edgar/data/1318605/000095017022000796/tsla-20211231.htm</li>
-                    </ul>
-                </div>
-            </div>
-        </div>
-    </div>
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/bootstrap/5.1.3/js/bootstrap.bundle.min.js"></script>
-
-    <!-- Application scripts -->
-    <script type="module" src="{{ url_for('static', filename='scripts/utils.js') }}"></script>
-    <script type="module" src="{{ url_for('static', filename='scripts/chat.js') }}"></script>
-    <script type="module" src="{{ url_for('static', filename='scripts/listArtifacts.js') }}"></script>
-    <script type="module" src="{{ url_for('static', filename='scripts/filingArtifacts.js') }}"></script>
-    <script type="module" src="{{ url_for('static', filename='scripts/tableArtifacts.js') }}"></script>
-    <script type="module" src="{{ url_for('static', filename='scripts/artifacts.js') }}"></script>
-    <script type="module" src="{{ url_for('static', filename='scripts/suggestions.js') }}"></script>
-    <script type="module" src="{{ url_for('static', filename='scripts/prefilledPrompt.js') }}"></script>
-    <script type="module" src="{{ url_for('static', filename='scripts/main.js') }}"></script>
-</body>
-
-</html>

datamule/mulebot/search.py

@@ -1,52 +0,0 @@
-import difflib
-from typing import Dict, List, Any
-
-def search_filing(query: str, nested_dict: Dict[str, Any], max_matches: int = 20, score_cutoff: float = 0.6) -> List[Dict[str, Any]]:
-    max_matches = min(max_matches, 20)
-    query = query.lower()  # Convert query to lowercase
-    
-    def flatten_dict(d: Dict[str, Any], parent_path: List[str] = None) -> List[Dict[str, Any]]:
-        parent_path = parent_path or []
-        items = []
-        
-        if isinstance(d, dict):
-            for k, v in d.items():
-                new_path = parent_path + [k]
-                if k == 'title' and isinstance(v, str):
-                    items.append({'path': new_path, 'title': v, 'title_lower': v.lower()})
-                items.extend(flatten_dict(v, new_path))
-        elif isinstance(d, list):
-            for i, item in enumerate(d):
-                new_path = parent_path + [str(i)]
-                items.extend(flatten_dict(item, new_path))
-        
-        return items
-
-    flat_list = flatten_dict(nested_dict)
-    all_titles_lower = [item['title_lower'] for item in flat_list]
-    
-    matches = difflib.get_close_matches(query, all_titles_lower, n=max_matches, cutoff=score_cutoff)
-    
-    results = []
-    for match in matches:
-        similarity = difflib.SequenceMatcher(None, query, match).ratio()
-        for item in flat_list:
-            if item['title_lower'] == match:
-                # Navigate to the correct nested dictionary
-                d = nested_dict
-                for key in item['path'][:-1]:  # Exclude the last 'title' key
-                    if key.isdigit():
-                        d = d[int(key)]
-                    else:
-                        d = d[key]
-                results.append({
-                    'path': '.'.join(item['path'][:-1]),  # Exclude the last 'title' key
-                    'content': d,
-                    'similarity': similarity
-                })
-                break
-    
-    results.sort(key=lambda x: x['similarity'], reverse=True)
-
-
-    return [item['content'] for item in results[:max_matches]]