Fixed the image selection.

app.py

@@ -89,7 +89,7 @@ def save_image(
 
 with gr.Blocks() as demo:
     # selected_image = gr.State(None)
-    selected_image_index = gr.State()
+    selected_image = gr.State()
     current_index = gr.State(0)
     character_data = gr.State([])
 
@@ -120,15 +120,15 @@ with gr.Blocks() as demo:
     )
 
     # When an image is clicked
-    def get_select_index(evt: gr.SelectData):
-        return evt.index
+    def get_select_index(evt: gr.SelectData, images):
+        return images[evt.index]
 
-    images.select(get_select_index, None, selected_image_index)
+    images.select(get_select_index, images, selected_image)
 
     save_button.click(
         save_image,
         inputs=[
-            images[selected_image_index],
+            selected_image,
             comic_id,
             character_data,
             current_index,