Fall back to loading the first image in the gallery if none are selected

app.py

@@ -51,8 +51,11 @@ window.settings = {
     crop_pos_x: 0,
     crop_pos_y: 0,
 };
-window.castHologram = async function() {
-    const uri = document.querySelector('#img-display-output .thumbnail-item.selected img').src;
+window.castHologram = async function(gallery) {
+    if (gallery.length == 0)
+        return;
+    const selected = document.querySelector('#img-display-output .thumbnail-item.selected img');
+    const uri = selected ? selected.src : gallery[0].image;
     if (!uri)
         return;
     const Bridge = BridgeClient.getInstance();
@@ -279,13 +282,13 @@ with gr.Blocks(css=css, head=head) as demo:
             reset = gr.Button(value="Reset All Parameters")
 
     gallery.select(fn=None, js="castHologram")
-    gallery.change(fn=zip_gallery, inputs=gallery, outputs=download_btn).then(fn=None, js="castHologram")
+    gallery.change(fn=zip_gallery, inputs=gallery, outputs=download_btn).then(fn=None, js="castHologram", inputs=gallery)
 
     submit.click(
         on_submit,
         inputs=[input_image, batch_images, book, model_size, upscale_model, upscale_method, denoise_level, discard_alpha],
         outputs=[gallery]
-    ).success(fn=zip_gallery, inputs=gallery, outputs=download_btn).then(fn=None, js="castHologram")
+    ).success(fn=zip_gallery, inputs=gallery, outputs=download_btn).then(fn=None, js="castHologram", inputs=gallery)
 
     depthiness.change(fn=None, inputs=depthiness, js="(value) => updateHologram (value, 'depthiness')")
     focus.change(fn=None, inputs=focus, js="(value) => updateHologram (value, 'focus')")
@@ -303,7 +306,7 @@ with gr.Blocks(css=css, head=head) as demo:
         return on_submit(image, None, None, 'vitl', None, None, -1, True)
     example_files = glob.glob('assets/examples/*')
     examples = gr.Examples(examples=example_files, inputs=[input_image], outputs=[gallery], fn=on_submit_example)
-    examples.load_input_event.success(fn=None, js="castHologram")
+    examples.load_input_event.success(fn=None, js="castHologram", inputs=gallery)
 
 if __name__ == '__main__':
     demo.queue().launch()