Update app.py

app.py

@@ -7,6 +7,11 @@ def preprocess_image(image_path):
     # Read image
     img = cv2.imread(image_path)
     
+    # Check if the image is loaded correctly
+    if img is None:
+        print(f"Error: Unable to load image at {image_path}")
+        return None, None
+    
     # Convert to grayscale
     gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
     
@@ -58,10 +63,17 @@ def remove_shadows(image):
     
     return result
 
-def main(image_path):
+def main():
+    # Ask the user to provide the image path
+    image_path = input("Please provide the path to the rooftop image (e.g., 'image.jpg'): ")
+    
     # Step 1: Preprocess image
     img, blurred = preprocess_image(image_path)
     
+    # If the image was not loaded successfully, exit the program
+    if img is None:
+        return
+    
     # Step 2: Detect edges
     edges = detect_edges(blurred)
     
@@ -88,5 +100,4 @@ def main(image_path):
     plt.show()
 
 if __name__ == "__main__":
-    image_path = "rooftop_image.jpg"  # Replace with your image path
-    main(image_path)
+    main()