feat: implement dynamic model-specific samples and resolve session stability error

apds/dashboard.py

@@ -14,6 +14,8 @@ from .detector import PotholeDetector
 from . import utils
 from datetime import datetime
 import torch
+import os
+import uuid
 from dotenv import load_dotenv
 
 import cv2
@@ -331,22 +333,33 @@ def tab_image(model, variant, conf, label, use_sahi: bool = False):
                 else:
                     st.session_state.detected_gps = None
     else:
-        st.session_state.detected_gps = None
-        samples = {
-            "Select Sample": None,
-            "Sample 1: Multiple Potholes": "train_0001_jpg.rf.042f2f916fa4f5561dac320039da31b5.jpg",
-            "Sample 2: Complex Road View": "train_0002_jpg.rf.0639c6d4a821da6d40ec8c499710b0f4.jpg",
-            "Sample 3: Shadowy Conditions": "test_0004_jpg.rf.ca7c5c7984d0c49b97d713a94471d3bb.jpg",
-            "Sample 4: Urban Arterial Stress": "test_0162_jpg.rf.2bd99b9e0ae585e50ae1ef3978672473.jpg",
-            "Sample 5: High-Traffic Fatigue": "test_0187_jpg.rf.696b03eb513485044a267f5fa8cb74d1.jpg"
-        }
+        # ── Dynamic Sample Logic ──
+        is_crack = "crack" in label.lower()
+        if is_crack:
+            samples = {
+                "Select Crack Sample": None,
+                "Crack Sample 1: Surface Fatigue": "cracking-2-_jpg.rf.7ca438dfc3d0379a740e37426ae8b60d.jpg",
+                "Crack Sample 2: Transverse stress": "cracking-27-_jpg.rf.0f7a0646abc4786b505562127d7be737.jpg",
+                "Crack Sample 3: Alligator cracking": "cracking-32-_jpg.rf.38387a609dd55cf2f0c6bedf5a79e684.jpg"
+            }
+        else:
+            samples = {
+                "Select Pothole Sample": None,
+                "Sample 1: Multiple Potholes": "train_0001_jpg.rf.042f2f916fa4f5561dac320039da31b5.jpg",
+                "Sample 2: Complex Road View": "train_0002_jpg.rf.0639c6d4a821da6d40ec8c499710b0f4.jpg",
+                "Sample 3: Shadowy Conditions": "test_0004_jpg.rf.ca7c5c7984d0c49b97d713a94471d3bb.jpg",
+                "Sample 4: Urban Arterial Stress": "test_0162_jpg.rf.2bd99b9e0ae585e50ae1ef3978672473.jpg",
+                "Sample 5: High-Traffic Fatigue": "test_0187_jpg.rf.696b03eb513485044a267f5fa8cb74d1.jpg"
+            }
+            
         name = col_sel.selectbox("Pick a field sample", list(samples.keys()))
         if name and samples[name]:
-            s_path = pu.get_sample_path(samples[name])
+            s_path = pu.get_sample_path(samples[name], is_crack=is_crack)
             if s_path.exists():
                 img = Image.open(s_path).convert("RGB")
+                st.info(f"💡 Playing demo with professional {label.lower()} sample.")
             else:
-                st.error(f"Sample not found at {s_path}")
+                st.error(f"Demo file missed in cloud: {samples[name]}")
 
     if img is None: return
     
@@ -677,6 +690,10 @@ def tab_evaluation(variant: str):
 
 
 def render():
+    if "session_id" not in st.session_state:
+        st.session_state.session_id = str(uuid.uuid4())
+        st.session_state.initialized = True
+        
     st.markdown("<h1 style='text-align:center;'>🏗️ Automated Pavement Distress System (APDS)</h1>", unsafe_allow_html=True)
     st.divider()
     

apds/path_utils.py

@@ -60,9 +60,14 @@ def get_training_artifact(model_path: Path, filename: str) -> Path:
     # Fallback to exact filename
     return base_dir / filename
 
-def get_sample_path(filename: str, is_video: bool = False) -> Path:
+def get_sample_path(filename: str, is_video: bool = False, is_crack: bool = False) -> Path:
     """Find a sample file, prioritizing cloud-accessible 'assets/' folder."""
-    sub = "videos" if is_video else "images"
+    if is_video:
+        sub = "videos"
+    elif is_crack:
+        sub = "images/cracks"
+    else:
+        sub = "images"
     
     # Priority 1: Cloud Assets (included in Git)
     p = ROOT / "assets" / "samples" / sub / filename
@@ -75,6 +80,8 @@ def get_sample_path(filename: str, is_video: bool = False) -> Path:
         p = ROOT / "data" / "combined" / "train" / "images" / filename
         if not p.exists():
             p = ROOT / "data" / "combined" / "test" / "images" / filename
+        if not p.exists() and is_crack:
+            p = ROOT / "data" / "bc_projekt" / "test" / "images" / filename
             
     return p