Feed 'selected_image' as function input

app.py

@@ -82,14 +82,15 @@ def gallery_selected(evt: gr.SelectData) -> tuple[int, gr.Button]:
     return evt.index, gr.Button(interactive=evt.selected)
 
 
-def to_moondream(images: list[tuple[Image.Image, str | None]]) -> tuple[gr.Tabs, Image.Image]:
+def to_moondream(images: list[tuple[Image.Image, str | None]], index: int) -> tuple[gr.Tabs, Image.Image]:
     """
     Listener that sends selected gallery image to the moondream model.
 
     :param images: list of images from yolos_gallery
+    :param index: index of selected gallery image
     :return: new selected tab and selected image (no caption)
     """
-    return gr.Tabs(selected='moondream'), images[selected_image][0]
+    return gr.Tabs(selected='moondream'), images[index][0]
 
 
 if __name__ == "__main__":
@@ -101,7 +102,7 @@ if __name__ == "__main__":
             Final project for IAT 481 at Simon Fraser University, Spring 2024.
             """
         )
-        selected_image = gr.Number(visible=False)
+        selected_image = gr.Number(visible=False, precision=0)
 
         # Referenced: https://github.com/gradio-app/gradio/issues/7726#issuecomment-2028051431
         with gr.Tabs(selected='yolos') as tabs:
@@ -128,7 +129,7 @@ if __name__ == "__main__":
         # --- YOLOS --- #
         yolos_submit.click(detect_objects, [yolos_input], yolos_gallery)
         yolos_gallery.select(gallery_selected, None, [selected_image, proceed_button])
-        proceed_button.click(to_moondream, yolos_gallery, [tabs, moon_img])
+        proceed_button.click(to_moondream, [yolos_gallery, selected_image], [tabs, moon_img])
 
         # --- Moondream --- #
         moon_submit.click(answer_question, [moon_img, moon_prompt], moon_output)