Fixed the image selection.

app.py

@@ -87,15 +87,9 @@ def save_image(
     print("Image saved successfully!")
 
 
-# Function to handle image selection and enable the save button
-def select_image(selected_image_index, images):
-    # Get the selected image from its index
-    selected_image = images[selected_image_index]
-    return gr.update(interactive=True), selected_image
-
-
 with gr.Blocks() as demo:
-    selected_image = gr.State(None)
+    # selected_image = gr.State(None)
+    selected_image_index = gr.State()
     current_index = gr.State(0)
     character_data = gr.State([])
 
@@ -126,16 +120,15 @@ with gr.Blocks() as demo:
     )
 
     # When an image is clicked
-    images.select(
-        select_image,
-        inputs=[gr.Number(), images],
-        outputs=[save_button, selected_image],
-    )
+    def get_select_index(evt: gr.SelectData):
+        return evt.index
+
+    images.select(get_select_index, None, selected_image_index)
 
     save_button.click(
         save_image,
         inputs=[
-            selected_image,
+            images[selected_image_index],
             comic_id,
             character_data,
             current_index,