Update app.py

app.py

@@ -435,6 +435,20 @@ def create_demo():
                 file_paths.append(f[0])
             else:
                 file_paths.append(f)
+                
+                valid_paths = []
+                for path in file_paths:
+                    try:
+                        img = cv2.imread(path)
+                        if img is not None:
+                            valid_paths.append(path)
+                    except:
+                        pass
+                
+                if len(valid_paths) < 2:
+                    return None, f"❌ Only {len(valid_paths)} valid images found", None
+                
+                file_paths = valid_paths
         
         # Update configuration
         basic_stitcher.config['feature_extractor'] = feature_type
@@ -444,10 +458,11 @@ def create_demo():
         basic_stitcher.config['ransac_threshold'] = ransac_thresh
         
         # Load images
-        images = [Image.open(f) for f in file_paths]
-        
-        # Process
-        return basic_stitcher.stitch_images(images)
+        try:
+            images = [Image.open(f) for f in file_paths]
+            return basic_stitcher.stitch_images(images)
+        except Exception as e:
+            return None, f"❌ Error: {str(e)}", None
         
     def process_zip_with_metadata(zip_file, feature_type, matcher_type, use_clahe, detect_labels, ransac_thresh):
         """Process ZIP file containing images and metadata"""