Fix: absolute paths and libGL dependencies

Dockerfile

@@ -6,10 +6,13 @@ RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
     git \
+    libgl1-mesa-glx \
+    libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
 COPY requirements.txt ./
 COPY src/ ./src/
+COPY models/ ./models/
 
 RUN pip3 install -r requirements.txt
 

src/pages/image_classification.py

@@ -37,8 +37,9 @@ CLASS_ICONS = {
 # MODEL CONFIGS
 # Preprocessing matches exactly what each training notebook used
 # ─────────────────────────────────────────────────────────────────────────────
-PLOT_DIR = "models/result"
-
+# PLOT_DIR = "models/result"
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+PLOT_DIR = os.path.join(BASE_DIR, "models", "result")
 MODEL_CONFIGS = {
     "EfficientNetB0": {
         "path":        os.path.join(PLOT_DIR, "efficientnetb0_best.keras"),

src/pages/model_performance.py

@@ -10,8 +10,10 @@ from PIL import Image
 # ─────────────────────────────────────────────────────────────────────────────
 # CONFIG
 # ─────────────────────────────────────────────────────────────────────────────
-PLOT_DIR = "models/result"
-
+# PLOT_DIR = "models/result"
+# Use absolute path relative to this script file
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+PLOT_DIR = os.path.join(BASE_DIR, "models", "result")
 MODELS = ["VGG16", "ResNet50", "MobileNetV2", "EfficientNetB0"]
 
 METRICS = {

src/pages/yolo_object_detection.py

@@ -53,8 +53,10 @@ CLASS_ICONS = {
 # ─────────────────────────────────────────────────────────────────────────────
 # MODEL LOADER
 # ─────────────────────────────────────────────────────────────────────────────
-YOLO_PATH = "models/best_yolo_model.pt"
-
+# YOLO_PATH = "models/best_yolo_model.pt"
+import os
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+YOLO_PATH = os.path.join(BASE_DIR, "..", "..", "models", "best_yolo_model.pt")
 @st.cache_resource(show_spinner=False)
 def load_yolo():
     return YOLO(YOLO_PATH)

src/pages/yolo_performance.py

@@ -5,8 +5,10 @@ from PIL import Image
 # ─────────────────────────────────────────────────────────────────────────────
 # CONFIG
 # ─────────────────────────────────────────────────────────────────────────────
-VAL_DIR = "models/val"
-
+# VAL_DIR = "models/val"
+import os
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+VAL_DIR = os.path.join(BASE_DIR, "..", "..", "models", "val")
 # Class names in exact YOLO training order (from data.yaml)
 CLASS_NAMES = [
     'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck',

src/pages/yolo_webcam.py

@@ -25,8 +25,10 @@ CLASS_COLORS_BGR = {
 }
 DEFAULT_COLOR = (0, 255, 0)
 
-YOLO_PATH = "models/best_yolo_model.pt"
-
+# YOLO_PATH = "models/best_yolo_model.pt"
+import os
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+YOLO_PATH = os.path.join(BASE_DIR, "..", "..", "models", "best_yolo_model.pt")
 # ─────────────────────────────────────────────────────────────────────────────
 # MODEL LOADER
 # ─────────────────────────────────────────────────────────────────────────────

src/streamlit_app.py

@@ -37,8 +37,10 @@ st.markdown(
 )
 
 # ✅ FIX 1: Use forward slash (works on Linux/Mac/Windows)
-image_path = "models/smartvision.png"
-
+# image_path = "models/smartvision.png"
+import os
+APP_ROOT   = "/app"
+image_path = os.path.join(APP_ROOT, "models", "smartvision.png")
 # ✅ FIX 2: Check if file exists before loading to give a clear error
 if os.path.exists(image_path):
     st.image(image_path, use_container_width=True)