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

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

@@ -1,129 +1,129 @@
-// 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");
+// 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 +1,28 @@
-// 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`);
+// 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 +1,91 @@
-<!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>
-
+<!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 +1,52 @@
-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)
-
-
+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]]