Update app.py

app.py

@@ -87,6 +87,18 @@ MODEL_CONFIG = {
     }
 }
 
+# Heuristic filename patterns for mapping examples per model
+MODEL_EXAMPLE_PATTERNS = {
+    "Pneumonia": ["pneumonia", "normal-"],
+    "Tuberculosis": ["tuberculosis", "tb-"],
+    "Brain Tumor": ["glioma", "meningioma", "notumor", "pituitary", "brain"],
+    "Skin Cancer": ["melanoma", "nev", "keratos", "carcinoma", "vascular", "dermatofibroma", "skin"],
+    "Kvasir": [
+        "dyedlifted", "dyedresection", "esophagitis", "normalceacum", "normalpylorus",
+        "normalzline", "polypus", "ulcerative"
+    ],
+}
+
 models = {}
 
 def load_all_models():
@@ -193,10 +205,16 @@ def serve_test_image(filename):
 def example_images():
     try:
         files = []
+        selected_model = (request.args.get('model') or '').strip()
+        patterns = MODEL_EXAMPLE_PATTERNS.get(selected_model, []) if selected_model else []
         if os.path.isdir(TEST_IMAGES_DIR):
             for f in os.listdir(TEST_IMAGES_DIR):
                 lf = f.lower()
                 if lf.endswith(('.png', '.jpg', '.jpeg')):
+                    # If a model is selected and patterns exist, filter by them
+                    if patterns:
+                        if not any(p in lf for p in patterns):
+                            continue
                     files.append(url_for('serve_test_image', filename=f))
         return jsonify({"images": files})
     except Exception as e: