Update app.py

app.py

@@ -2,12 +2,13 @@ import cv2 as cv
 import numpy as np
 import gradio as gr
 import datetime
+from pathlib import Path
 from huggingface_hub import hf_hub_download
 
 from facial_fer_model import FacialExpressionRecog
 from yunet import YuNet
 
-# Download ONNX model from Hugging Face
+# Download ONNX modellen van Hugging Face
 FD_MODEL_PATH = hf_hub_download(repo_id="opencv/face_detection_yunet", filename="face_detection_yunet_2023mar.onnx")
 FER_MODEL_PATH = hf_hub_download(repo_id="opencv/facial_expression_recognition", filename="facial_expression_recognition_mobilefacenet_2022july.onnx")
 
@@ -50,12 +51,23 @@ def detect_expression(input_image):
     output = visualize(image, dets, fer_res)
     return cv.cvtColor(output, cv.COLOR_BGR2RGB)
 
+# === Automatisch voorbeelden inladen uit de map "examples/" ===
+IMAGE_EXTS = {".jpg", ".jpeg", ".png", ".bmp", ".webp"}
+EXAMPLES_DIR = Path("examples")
+
+if EXAMPLES_DIR.exists() and EXAMPLES_DIR.is_dir():
+    example_paths = [
+        str(p) for p in sorted(EXAMPLES_DIR.iterdir())
+        if p.is_file() and p.suffix.lower() in IMAGE_EXTS
+    ]
+else:
+    example_paths = []
+
+example_list = [[p] for p in example_paths]
+# =============================================================
+
 # Gradio Interface
-with gr.Blocks(css='''.example * {
-    font-style: italic;
-    font-size: 18px !important;
-    color: #0ea5e9 !important;
-    }''') as demo:
+with gr.Blocks(css='''.example * { font-style: italic; font-size: 18px !important; color: #0ea5e9 !important; }''') as demo:
 
     gr.Markdown("### Facial Expression Recognition (FER) with OpenCV DNN")
     gr.Markdown("Detects faces and recognizes facial expressions using YuNet + MobileFaceNet ONNX models.")
@@ -64,7 +76,7 @@ with gr.Blocks(css='''.example * {
         input_image = gr.Image(type="numpy", label="Upload Image")
         output_image = gr.Image(type="numpy", label="Facial Expression Result")
 
-    # Clear output when new image is uploaded
+    # Output leegmaken bij nieuwe upload
     input_image.change(fn=lambda: (None), outputs=output_image)
 
     with gr.Row():
@@ -72,18 +84,16 @@ with gr.Blocks(css='''.example * {
         clear_btn = gr.Button("Clear")
 
     submit_btn.click(fn=detect_expression, inputs=input_image, outputs=output_image)
-    clear_btn.click(fn=lambda:(None, None), outputs=[input_image, output_image])
+    clear_btn.click(fn=lambda: (None, None), outputs=[input_image, output_image])
 
-    gr.Markdown("Click on any example to try it.", elem_classes=["example"])
+    gr.Markdown("Klik op een voorbeeld om het te proberen.", elem_classes=["example"])
 
     gr.Examples(
-        examples=[
-            ["examples/lena.jpg"],
-            ["examples/gray_face.png"]
-        ],
-        inputs=input_image
+        examples=example_list,        # automatisch ingelezen
+        inputs=input_image,
+        examples_per_page=20,         # handig bij veel afbeeldingen
+        cache_examples=True           # sneller in Hugging Face Spaces
     )
 
-
 if __name__ == "__main__":
     demo.launch()