Update utility/utils.py

utility/utils.py

@@ -47,10 +47,20 @@ client = InferenceClient(model="mistralai/Mistral-7B-Instruct-v0.3", token=HFT)
 
 # Load image using OpenCV
 def load_image(image_path):
-    image = cv2.imread(image_path)
-    if image is None:
-        raise ValueError(f"Could not load image from {image_path}. It may be corrupted or the path is incorrect.")
-    return image
+    ext = os.path.splitext(file_path)[1].lower()
+    if ext in ['.png', '.jpg', '.jpeg']:
+        image = cv2.imread(image_path)
+        return image
+    else:
+        raise ValueError(f"Could not load image from {image_path}. It may be corrupted or the path is incorrect. or in a not supported format")
+   
+def extract_hyperlinks(file_path):
+    ext = os.path.splitext(file_path)[1].lower()
+    
+    if ext in ['.png', '.jpg', '.jpeg']:
+        return 'Not Supported'
+    else:
+        raise ValueError("Unsupported file format")
 
 # Function for upscaling image using OpenCV's INTER_CUBIC
 def upscale_image(image, scale=2):