button for download

static/index.html

@@ -24,6 +24,7 @@
                 <label for="text-gen-input">Text prompt</label>
                 <input id="text-gen-input" type="text" value="English: Translate There are many ducks. German:" />
                 <button id="text-gen-submit">Submit</button>
+                <button id="download-embeddings" type="button">Download Embeddings</button>
                 <p class="text-gen-output"></p>
             </form>
         </section>

static/script.js

@@ -1,10 +1,11 @@
 const textGenForm = document.querySelector('.text-gen-form');
+let embeddingsList = [];
 
 const embedText = async (text) => {
     const inferResponse = await fetch(`embeddings?input=${text}`);
     const inferJson = await inferResponse.json();
 
-    return inferJson.output;
+    return inferJson.embeddings;
 };
 
 textGenForm.addEventListener('submit', async (event) => {
@@ -14,8 +15,32 @@ textGenForm.addEventListener('submit', async (event) => {
     const textGenParagraph = document.querySelector('.text-gen-output');
 
     try {
-        textGenParagraph.textContent = await embedText(textGenInput.value);
+        const embeddings = await embedText(textGenInput.value);
+        embeddingsList = embeddings; // Store embeddings in the variable
+        textGenParagraph.textContent = JSON.stringify(embeddingsList);
+        updateDownloadButtonState(); // Update button state
     } catch (err) {
         console.error(err);
     }
-});
+});
+
+const downloadButton = document.getElementById('download-embeddings');
+
+const updateDownloadButtonState = () => {
+    downloadButton.disabled = embeddingsList.length === 0;
+};
+
+const downloadEmbeddings = () => {
+    const dataStr = "data:text/json;charset=utf-8," + encodeURIComponent(JSON.stringify(embeddingsList));
+    const downloadAnchorNode = document.createElement('a');
+    downloadAnchorNode.setAttribute("href", dataStr);
+    downloadAnchorNode.setAttribute("download", "embeddings.json");
+    document.body.appendChild(downloadAnchorNode);
+    downloadAnchorNode.click();
+    downloadAnchorNode.remove();
+};
+
+downloadButton.addEventListener('click', downloadEmbeddings);
+
+// Initialize button state
+updateDownloadButtonState();