Update main.py

main.py

@@ -4,6 +4,41 @@ from fastapi.responses import JSONResponse
 import os
 import json
 from models import RequestModel
+import httpx  # Para baixar imagens de URLs, se necessário
+
+BASE_DIR = "saved_data"
+app = FastAPI()
+
+def orb_sim(img1, img2):
+    # ORB
+    orb = cv2.ORB_create()
+    kp_a, desc_a = orb.detectAndCompute(img1, None)
+    kp_b, desc_b = orb.detectAndCompute(img2, None)
+
+    # Brute-force matcher
+    bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)
+    matches = bf.match(desc_a, desc_b)
+    similar_regions = [i for i in matches if i.distance < 50]
+    if len(matches) == 0:
+        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
 
 app = FastAPI()
 
@@ -15,8 +50,21 @@ async def save(image_data: RequestModel, request: Request):
     print("RAW BODY RECEIVED →", body.decode())
     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)
+
+    similarity_orb = None
+    if img1 is not None and img2 is not None:
+        similarity_orb = orb_sim(img1, img2)
+        print(f"Similaridade ORB entre {image_data.originSource} e {image_data.source}: {similarity_orb}")
+
+    data_to_save = image_data.dict()
+    if similarity_orb is not None:
+        data_to_save["similarityOrb"] = similarity_orb
+
     with open(filename, "w") as f:
-        json.dump(image_data.dict(), f, indent=4)
+        json.dump(data_to_save, f, indent=4)
     return True
 
 @app.get("/files")