Update main.py

main.py

@@ -25,22 +25,20 @@ def orb_sim(img1, img2):
         return 0
     return len(similar_regions) / len(matches)
 
-async def load_image(image_path: str):
-    """Carrega uma imagem a partir de um caminho de arquivo ou URL."""
-    try:
-        if image_path.startswith("http://") or image_path.startswith("https://"):
-            async with httpx.AsyncClient() as client:
-                response = await client.get(image_path)
-                response.raise_for_status()
-                image_bytes = np.frombuffer(response.content, np.uint8)
-                img = cv2.imdecode(image_bytes, cv2.IMREAD_COLOR)
-                return img
-        else:
-            img = cv2.imread(image_path)
-            return img
-    except Exception as e:
-        print(f"Erro ao carregar a imagem {image_path}: {e}")
-        return None
+def load_image_url(source):
+    Image.MAX_IMAGE_PIXELS = None
+
+    if source.startswith('http'):
+        response = requests.get(source)
+        img = np.asarray(bytearray(response.content), dtype=np.uint8)
+        img = cv2.imdecode(img, cv2.IMREAD_GRAYSCALE)
+    else:
+        img = base64.b64decode(source)
+        img = Image.open(BytesIO(img))
+        img = np.array(img)
+        img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
+
+    return img
 
 app = FastAPI()
 
@@ -51,8 +49,8 @@ async def save(image_data: RequestModel):
     os.makedirs(BASE_DIR, exist_ok=True)
     filename = os.path.join(BASE_DIR, f"{image_data.originId}_{image_data.assetCode}.json")
 
-    img1 = await load_image(image_data.originSource)
-    img2 = await load_image(image_data.source)
+    img1 = load_image(image_data.originSource)
+    img2 = load_image(image_data.source)
 
     similarity_orb = None
     if img1 is not None and img2 is not None: