Upload 5 files

app.py

@@ -1,22 +1,41 @@
 import io
-from skimage.segmentation import mark_boundaries
-from lime import lime_image
-import shap
 import cv2
-from PIL import Image
-from tensorflow.keras import layers as L, models
-import tensorflow as tf
-import numpy as np
-import gradio as gr
-import os
 import matplotlib.pyplot as plt
 import matplotlib
-matplotlib.use('Agg')  # Use non-interactive backend
-
+import gradio as gr
+import numpy as np
+import tensorflow as tf
+from tensorflow.keras import layers as L, models
+from PIL import Image
+import os
 # Disable Gradio queue for direct REST API access - MUST be before gradio import
 os.environ["GRADIO_QUEUE"] = "false"
 os.environ["HF_HUB_DISABLE_GRADIO_QUEUE"] = "1"
 
+matplotlib.use('Agg')  # Use non-interactive backend
+
+# Import XAI libraries with error handling
+try:
+    import shap
+    SHAP_AVAILABLE = True
+except ImportError:
+    SHAP_AVAILABLE = False
+    print("Warning: SHAP not available")
+
+try:
+    from lime import lime_image
+    LIME_AVAILABLE = True
+except ImportError:
+    LIME_AVAILABLE = False
+    print("Warning: LIME not available")
+
+try:
+    from skimage.segmentation import mark_boundaries
+    SKIMAGE_AVAILABLE = True
+except ImportError:
+    SKIMAGE_AVAILABLE = False
+    print("Warning: scikit-image not available")
+
 
 # -----------------------------
 # Model Architecture Components
@@ -297,6 +316,9 @@ def generate_shap(image):
     """
     Generate SHAP explanation visualization.
     """
+    if not SHAP_AVAILABLE:
+        return None, "SHAP library not available. Please install shap package."
+
     if model is None or image is None:
         return None, "Model not loaded or no image provided"
 
@@ -363,6 +385,9 @@ def generate_lime(image):
     """
     Generate LIME explanation visualization.
     """
+    if not LIME_AVAILABLE or not SKIMAGE_AVAILABLE:
+        return None, None, "LIME or scikit-image library not available. Please install lime and scikit-image packages."
+
     if model is None or image is None:
         return None, None, "Model not loaded or no image provided"
 

requirements.txt

@@ -3,3 +3,8 @@ tensorflow>=2.13.0
 numpy>=1.24.0
 pillow>=10.0.0
 huggingface_hub==0.17.3
+matplotlib>=3.7.0
+opencv-python-headless>=4.8.0
+scikit-image>=0.21.0
+shap>=0.43.0
+lime>=0.2.0.1