updated app

app.py

@@ -39,7 +39,53 @@ else:
     DB_VECTORS = None
     MODEL_NAME = "Unknown"
 
-# --- 2. Define the Search Logic ---
+# --- 2. Helper function to load images with fallbacks ---
+def load_image_with_fallback(image_path, name):
+    """Try to load image from various possible locations, or create placeholder"""
+    # Try original path
+    if os.path.exists(image_path):
+        try:
+            return Image.open(image_path)
+        except:
+            pass
+    
+    # Try just the filename in current directory
+    filename = os.path.basename(image_path)
+    if os.path.exists(filename):
+        try:
+            return Image.open(filename)
+        except:
+            pass
+    
+    # Try in my_dataset directory
+    dataset_path = os.path.join("my_dataset", filename)
+    if os.path.exists(dataset_path):
+        try:
+            return Image.open(dataset_path)
+        except:
+            pass
+    
+    # Create a placeholder image if nothing found
+    placeholder = Image.new('RGB', (200, 200), color=(220, 220, 220))
+    from PIL import ImageDraw, ImageFont
+    draw = ImageDraw.Draw(placeholder)
+    # Try to use default font, fallback to basic if not available
+    try:
+        font = ImageFont.truetype("/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf", 20)
+    except:
+        font = ImageFont.load_default()
+    
+    # Draw text in center
+    text = f"Image\nNot Found"
+    bbox = draw.textbbox((0, 0), text, font=font)
+    text_width = bbox[2] - bbox[0]
+    text_height = bbox[3] - bbox[1]
+    position = ((200 - text_width) // 2, (200 - text_height) // 2)
+    draw.text(position, text, fill=(100, 100, 100), font=font)
+    
+    return placeholder
+
+# --- 3. Define the Search Logic ---
 def find_best_matches(user_image):
     # Error handling for empty inputs
     if user_image is None: 
@@ -75,14 +121,17 @@ def find_best_matches(user_image):
             display_name = f"{row['name']} (Match: {int(score*100)}%)"
             result_text += f"### #{i}: {display_name}\n\n"
             
-            # Load Image - only add to gallery if image exists and loads successfully
+            # Load Image with fallback - always add to gallery (with placeholder if needed)
             try:
-                if os.path.exists(row['image_path']):
-                    img = Image.open(row['image_path'])
-                    gallery_images.append((img, display_name))
-                else:
-                    result_text += f"⚠️ Image file not found: {row['image_path']}\n\n"
+                img = load_image_with_fallback(row['image_path'], row['name'])
+                gallery_images.append((img, display_name))
+                # Check if we used a placeholder
+                if not os.path.exists(row['image_path']) and not os.path.exists(os.path.basename(row['image_path'])):
+                    result_text += f"⚠️ Image not found at: {row['image_path']} (showing placeholder)\n\n"
             except Exception as img_error:
+                # Even if loading fails, create placeholder
+                placeholder = Image.new('RGB', (200, 200), color=(220, 220, 220))
+                gallery_images.append((placeholder, display_name))
                 result_text += f"⚠️ Could not load image: {str(img_error)}\n\n"
         
         # Don't pad with None - Gallery can't handle None images