Adjust layout and test gallery event listener

app.py

@@ -72,6 +72,13 @@ def detect_objects(img: Image.Image):
     return box_images
 
 
+def get_selected_index(evt: gr.SelectData):
+    """
+    Returns the selected image from the object detection output gallery.
+    """
+    return evt.index
+
+
 if __name__ == "__main__":
     with gr.Blocks() as app:
         gr.Markdown(
@@ -83,11 +90,12 @@ if __name__ == "__main__":
         )
 
         with gr.Tab("Object Detection"):
-            with gr.Column():
-                yolos_input = gr.Image(type="pil")
-                yolos_button = gr.Button("Submit")
-            with gr.Column():
-                yolos_output = gr.Gallery(label="Detected Objects", object_fit="contain", columns=3)
+            with gr.Row():
+                with gr.Column():
+                    yolos_input = gr.Image(type="pil")
+                    yolos_button = gr.Button("Submit")
+                yolos_output = gr.Gallery(label="Detected Objects", object_fit="scale-down", columns=3, scale=1)
+            yolos_selected = gr.Number()
 
         with gr.Tab("Inference"):
             with gr.Row():
@@ -97,7 +105,11 @@ if __name__ == "__main__":
                 moon_img = gr.Image(label="Image", type="pil")
                 moon_output = gr.TextArea(label="Output")
 
-        moon_submit.click(answer_question, [moon_img, moon_prompt], moon_output)
+        # --- YOLOS --- #
         yolos_button.click(detect_objects, [yolos_input], yolos_output)
+        yolos_output.select(get_selected_index, None, yolos_selected)
+
+        # --- Moondream --- #
+        moon_submit.click(answer_question, [moon_img, moon_prompt], moon_output)
 
     app.queue().launch()