Update app.py

app.py

@@ -61,12 +61,25 @@ with gr.Blocks(title="Animal Classifier", css=css) as demo:
     
     inputs.change(update_state, inputs, all_files_state)
     
-    # Handle example selection - FIXED INDEX ACCESS
+    # Handle example selection - FIXED WITH VALIDATION
     def add_example(example_index, current_files):
-        # Get first selected index (even for single selection)
-        selected_idx = example_index[0] if isinstance(example_index, list) else example_index
-        selected_path = example_images[selected_idx]
-        return current_files + [selected_path]
+        try:
+            # Handle Gradio's selection format
+            if isinstance(example_index, list):
+                if not example_index:  # Empty selection
+                    return current_files
+                selected_idx = example_index[0]
+            else:
+                selected_idx = example_index
+            
+            # Validate index range
+            if 0 <= selected_idx < len(example_images):
+                selected_path = example_images[selected_idx]
+                return current_files + [selected_path]
+            return current_files
+        except Exception as e:
+            print(f"Error handling example selection: {str(e)}")
+            return current_files
     
     # Corrected gallery select handler
     examples_gallery.select(