Commit inicial

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,12 @@
+---
+title: Facial Comparison
+emoji: 🐨
+colorFrom: indigo
+colorTo: green
+sdk: gradio
+sdk_version: 3.44.4
+app_file: app.py
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,65 @@
+import os
+import gradio as gr
+from services.aws_service import AwsService
+from dotenv import load_dotenv
+import cv2
+import face_recognition
+import numpy as np
+
+load_dotenv()
+   
+def find_face_encodings(image_path):
+  # reading image
+  image = cv2.imread(image_path)
+  # get face encodings from the image
+  face_enc = face_recognition.face_encodings(image)
+  # return face encodings
+  return face_enc[0]
+
+def face_similarity(img1, img2):
+  image_1 = find_face_encodings(img1)
+  image_2  = find_face_encodings(img2)
+
+  # checking both images are same
+  is_same = face_recognition.compare_faces([image_1], image_2)[0]
+  accuracy = 0
+  if is_same:
+    # finding the distance level between images
+    distance = face_recognition.face_distance([image_1], image_2)
+    distance = round(distance[0] * 100)
+    
+    # calcuating accuracy level between images
+    accuracy = 100 - round(distance)
+  return {"is_same": is_same, "accuracy": accuracy}
+
+def convert_pil_to_open_cv(pil_image):
+  open_cv_image = np.array(pil_image) 
+  open_cv_image = open_cv_image[:, :, ::-1].copy() 
+  return open_cv_image
+
+def check_image_similarity(main_image_folder, compare_image_folder):
+
+  main_image = AwsService.get_image_from_s3(os.environ.get('AWS_S3_BUCKET'), main_image_folder)
+  compare_image = AwsService.get_image_from_s3(os.environ.get('AWS_S3_BUCKET'), compare_image_folder)
+  
+  # download image to local storage
+  main_image["pil"].save("main_image.jpg")
+  compare_image["pil"].save("compare_image.jpg")
+
+  result = face_similarity("main_image.jpg", "compare_image.jpg")
+
+  os.remove("main_image.jpg")
+  os.remove("compare_image.jpg")
+
+  return result
+
+iface = gr.Interface(
+  fn=check_image_similarity, 
+  inputs=[
+    gr.Textbox(lines=1, placeholder="Main Image Key", value="PhotoShoots/653fea0aad75f4caca672886/Croppeds/IMG-1698687849160-2.jpg"),
+    gr.Textbox(lines=1, placeholder="Compare Image Key", value="PhotoShoots/653fea0aad75f4caca672886/Croppeds/IMG-1698687855097-1.jpg")
+  ], 
+  outputs="text"
+)
+
+iface.launch()

requirements.txt

@@ -0,0 +1,68 @@
+aiofiles==23.2.1
+altair==5.1.2
+annotated-types==0.6.0
+anyio==3.7.1
+attrs==23.1.0
+boto3==1.26.25
+botocore==1.29.25
+certifi==2023.7.22
+charset-normalizer==3.3.2
+click==8.1.7
+contourpy==1.1.1
+cycler==0.12.1
+dlib==19.24.2
+exceptiongroup==1.1.3
+face-recognition==1.3.0
+face-recognition-models==0.3.0
+fastapi==0.104.1
+ffmpy==0.3.1
+filelock==3.13.1
+fonttools==4.44.0
+fsspec==2023.10.0
+gradio==3.44.4
+gradio-client==0.5.1
+h11==0.14.0
+httpcore==1.0.2
+httpx==0.25.1
+huggingface-hub==0.19.1
+idna==3.4
+importlib-resources==6.1.1
+Jinja2==3.1.2
+jmespath==1.0.1
+jsonschema==4.19.2
+jsonschema-specifications==2023.7.1
+kiwisolver==1.4.5
+MarkupSafe==2.0.1
+matplotlib==3.7.3
+numpy==1.24.4
+opencv-python==4.8.1.78
+orjson==3.9.10
+packaging==23.2
+pandas==2.0.3
+Pillow==10.1.0
+pkgutil-resolve-name==1.3.10
+pydantic==2.5.0
+pydantic-core==2.14.1
+pydub==0.25.1
+pyparsing==3.1.1
+python-dateutil==2.8.2
+python-dotenv==1.0.0
+python-multipart==0.0.6
+pytz==2023.3.post1
+PyYAML==6.0.1
+referencing==0.30.2
+requests==2.31.0
+rpds-py==0.12.0
+s3transfer==0.6.2
+semantic-version==2.10.0
+six==1.16.0
+sniffio==1.3.0
+starlette==0.27.0
+toolz==0.12.0
+tqdm==4.66.1
+typing-extensions==4.8.0
+tzdata==2023.3
+urllib3==1.26.18
+uvicorn==0.24.0.post1
+websockets==11.0.3
+zipp==3.17.0

services/aws_service.py

@@ -0,0 +1,29 @@
+import os
+import boto3
+from PIL import Image
+from io import BytesIO
+
+class AwsService:
+  def session():
+    return boto3.Session(
+      aws_access_key_id = os.environ.get('AWS_ACCESS_KEY_ID'),
+      aws_secret_access_key = os.environ.get('AWS_SECRET_ACCESS_KEY'),
+      region_name=os.environ.get('AWS_REGION')
+    )
+  
+  def s3_client():
+    return AwsService.session().client('s3')     
+
+  def get_files_from_s3(bucket, prefix):
+    results = AwsService.s3_client().list_objects(Bucket=bucket, Prefix=prefix)
+    if 'Contents' in results:
+      return results['Contents']
+    else:
+      return []
+  
+  def get_image_from_s3(bucket, key):
+    file_byte_string = AwsService.s3_client().get_object(Bucket=bucket, Key=key)['Body'].read()
+    return {
+      'key': key.split('/')[-1].split('.')[0],
+      'pil': Image.open(BytesIO(file_byte_string))
+    }