Update app.py

app.py

@@ -45,8 +45,7 @@ translations = {
         "low_confidence": "Low confidence prediction. Results may not be accurate.",
         "not_a_bird": "The image may not contain a bird. Please upload a clear image of a bird.",
         "other_message": "This image is not in our trained dataset or the image may not be of a bird. Please try uploading a different image.",
-        "example_images": "Example Images",
-        "try_example": "Try Example"
+        "example_images": "Example Images - Click on any image to use it"
     },
     "sw": {
         "app_title": "Mtafiti wa Ndege: Utambuzi wa Kiotomatiki kwa Watafiti",
@@ -70,8 +69,7 @@ translations = {
         "low_confidence": "Utabiri wa uhakika mdogo. Matokeo yanaweza kuwa si sahihi.",
         "not_a_bird": "Picha inaweza isiwe ya ndege. Tafadhali pakia picha wazi ya ndege.",
         "other_message": "Ndege huyu haipatikani katika hifadhidata yetu au picha inaweza isiwe ya ndege. Tafadhali jaribu kupakia picha nyingine.",
-        "example_images": "Mifano ya Picha",
-        "try_example": "Jaribu Mfano"
+        "example_images": "Mifano ya Picha - Bofya picha yoyote kutumia"
     }
 }
 
@@ -576,6 +574,14 @@ def follow_up_question(question, bird_name, language="en"):
         error_msg = "Hitilafu katika kupata taarifa" if language == "sw" else "Error fetching information"
         return f"{error_msg}: {str(e)}"
 
+def get_available_examples():
+    """Get list of available example images that actually exist"""
+    available_examples = []
+    for example in EXAMPLE_IMAGES:
+        if os.path.exists(example["path"]):
+            available_examples.append(example)
+    return available_examples
+
 # Create the Gradio interface
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     # Current language and bird state
@@ -601,19 +607,24 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         with gr.Column(scale=1):
             input_image = gr.Image(type="pil", label=translations['en']['upload_label'])
             
-            # Example images section
+            # Example images section with actual image display
             example_header = gr.Markdown(f"### {translations['en']['example_images']}")
             
-            # Create example image buttons in a grid
-            with gr.Row():
-                example_buttons = []
-                for i, example in enumerate(EXAMPLE_IMAGES):
-                    btn = gr.Button(
-                        f"{translations['en']['try_example']} {i+1}",
-                        variant="secondary",
-                        size="sm"
-                    )
-                    example_buttons.append(btn)
+            # Create example images gallery
+            available_examples = get_available_examples()
+            if available_examples:
+                example_images = gr.Gallery(
+                    value=[(example["path"], example["name"]) for example in available_examples],
+                    label="Click any image to use it",
+                    show_label=False,
+                    elem_id="example-gallery",
+                    columns=2,
+                    rows=2,
+                    height="auto",
+                    allow_preview=False
+                )
+            else:
+                example_images = gr.Markdown("*No example images available*")
             
             submit_btn = gr.Button(translations['en']['identify_button'], variant="primary")
         
@@ -682,24 +693,17 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             t['question_label'],
             t['question_placeholder'],
             t['submit_question'],
-            t['clear_conversation'],
-            # Update example button labels
-            *[f"{t['try_example']} {i+1}" for i in range(len(EXAMPLE_IMAGES))]
+            t['clear_conversation']
         )
     
-    # Functions to handle example image clicks
-    def create_example_handlers():
-        """Create handlers for example image buttons"""
-        handlers = []
-        for example in EXAMPLE_IMAGES:
-            def make_handler(path):
-                def handler():
-                    return load_example_image(path)
-                return handler
-            handlers.append(make_handler(example["path"]))
-        return handlers
-    
-    example_handlers = create_example_handlers()
+    def handle_example_select(evt: gr.SelectData):
+        """Handle selection of example images"""
+        if evt.index is not None:
+            available_examples = get_available_examples()
+            if evt.index < len(available_examples):
+                selected_example = available_examples[evt.index]
+                return load_example_image(selected_example["path"])
+        return None
     
     # Set up event handlers
     language_selector.change(
@@ -718,15 +722,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
             follow_up_input,
             follow_up_input,
             follow_up_btn,
-            clear_btn,
-            *example_buttons
+            clear_btn
         ]
     )
     
-    # Set up example image button handlers
-    for i, (btn, handler) in enumerate(zip(example_buttons, example_handlers)):
-        btn.click(
-            handler,
+    # Handle example image selection
+    if available_examples:
+        example_images.select(
+            handle_example_select,
             outputs=[input_image]
         )