Update app.py

app.py

@@ -69,7 +69,33 @@ def upload_sketch(file):
 
 scripts = """
 async () => {
-    
+
+    globalThis.theSketchDownloadFunction = () => {
+        console.log("test")
+        var link = document.createElement("a");
+        dataUri = document.getElementById('download_sketch').href
+        link.setAttribute("href", dataUri)
+        link.setAttribute("download", "sketch.png")
+        document.body.appendChild(link); // Required for Firefox
+        link.click();
+        document.body.removeChild(link); // Clean up
+      
+        // also call the output download function
+        theOutputDownloadFunction();
+      return false
+    }
+
+    globalThis.theOutputDownloadFunction = () => {
+        console.log("test output download function")
+        var link = document.createElement("a");
+        dataUri = document.getElementById('download_output').href
+        link.setAttribute("href", dataUri);
+        link.setAttribute("download", "output.png");
+        document.body.appendChild(link); // Required for Firefox
+        link.click();
+        document.body.removeChild(link); // Clean up
+      return false
+    }
 
 
 
@@ -147,7 +173,7 @@ with gr.Blocks(css="style.css") as demo:
                 source="canvas", tool="color-sketch", type="pil", image_mode="L",
                 invert_colors=True, shape=(512, 512), brush_radius=4, height=440, width=440,
                 brush_color="#000000", interactive=True, show_download_button=False, elem_id="input_image", show_label=False)
-            
+            download_sketch = gr.Button("Download sketch", scale=1, elem_id="download_sketch")
             gr.HTML("""
             <div class="button-row">
                 <div id="my-div-pencil" class="pad2"> <button id="my-toggle-pencil" onclick="return togglePencil(this)"></button> </div>
@@ -168,9 +194,9 @@ with gr.Blocks(css="style.css") as demo:
 
     demo.load(None,None,None,_js=scripts)
     inputs = [image, item]
-    outputs = [result]
+    outputs = [result, download_sketch, download_output]
     item.change(fn=run, inputs=inputs, outputs=outputs)
-    image.change(run, inputs=inputs, outputs=outputs,)
+    image.change(fn=run, inputs=inputs, outputs=outputs,)
 
 if __name__ == "__main__":
     demo.queue().launch(debug=True)