Update similarity.py

similarity.py

@@ -137,13 +137,7 @@ def load_image_url(source, assetCode, contentType=None, ffmpeg_path='ffmpeg', fr
             try:
                 img_bytes = base64.b64decode(source)
 
-                if contentType and contentType.startswith('image'):
-                    print(f"[INFO] Base64 de {assetCode} identificado como imagem")
-                    img = Image.open(BytesIO(img_bytes))
-                    img = np.array(img)
-                    img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
-                    return img
-                else:
+                if contentType and contentType.startswith('video'):
                     print(f"[INFO] Base64 de {assetCode} identificado como vídeo")
                     with tempfile.NamedTemporaryFile(suffix='.mp4', delete=False) as temp_video:
                         temp_video.write(img_bytes)
@@ -152,6 +146,13 @@ def load_image_url(source, assetCode, contentType=None, ffmpeg_path='ffmpeg', fr
                     frame = extract_frame_from_video(temp_video_path, frame_time)
                     os.remove(temp_video_path)
                     return frame
+                else:
+                    print(f"[INFO] Base64 de {assetCode} identificado como imagem")
+                    img = Image.open(BytesIO(img_bytes))
+                    img = np.array(img)
+                    img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
+                    return img
+                    
 
             except Exception as e:
                 print(f"[ERRO] Falha ao processar base64 de {assetCode}: {e}")