* More Stuff

public/js/main.js

@@ -8,6 +8,7 @@ const randomPrompts = [
 ]
 
 var generatedImage = ``;
+var generatedImageDownloadBase64 = ``;
 var nextSubmitAllowedTime = 0;
 
 function generateRandomPrompt() {
@@ -101,6 +102,15 @@ async function handleImageGeneratorForm() {
 
         })
 
+        imageGeneratorForm.querySelector(`#download-image-button`).addEventListener( `click`, e => {
+            e.preventDefault();
+
+            if ( !!generatedImageDownloadBase64 ) {
+                window.location.href = 'data:application/octet-stream;base64,' + generatedImageDownloadBase64;
+            }
+
+        })
+
         imageGeneratorForm.querySelector(`#submit-image-button`).addEventListener( `click`, async (e) => {
 
             // Cancel the default post, we wills end the data to an API endpoint
@@ -181,6 +191,7 @@ async function handleImageGeneratorForm() {
             document.getElementById(`submit-image-button`).removeAttribute(`disabled`);
 
             generatedImage = r.images[0];
+            generatedImageDownloadBase64 = r.images[0];
 
         })
     }