Update app.py

app.py

@@ -4,12 +4,19 @@ from geti_sdk.deployment import Deployment
 from geti_sdk.utils import show_image_with_annotation_scene
 import numpy as np
 from urllib.request import urlopen
+import re
+
 
 # Step 1: Load the deployment
 deployment = Deployment.from_folder("deployment")
 deployment.load_inference_models(device="CPU")
 
 
+def is_valid_url(url):
+    pattern = r'^(https?|ftp)://[^\s/$.?#].[^\s]*$'
+    return re.match(pattern, url) is not None
+
+
 def resize_image(image, target_dimension):
     height, width = image.shape[:2]
     max_dimension = max(height, width)
@@ -22,17 +29,22 @@ def resize_image(image, target_dimension):
 
 def infer(image=None, url:str=None):
     if image is None and url is None:
-        return [None,'no image or URL provided']
+        return [None,'Error: No image or URL provided']
 
     if image is None:
         if isinstance(url, str):
+
+            if not is_valid_url(url):
+                return [None,'Error: URL appears to be invalid']
+
             req = urlopen(url)
             arr = np.asarray(bytearray(req.read()), dtype=np.uint8)
             image = cv2.imdecode(arr, -1)
             if image is None:
-                return [None, f'Unable to fetch image from {url}']
+                return [None, f'Error: Unable to fetch image from {url}']
             image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
+
     image = resize_image(image, 1200)
     image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     prediction = deployment.infer(image)