Spaces:
Sleeping
Sleeping
| from io import BytesIO | |
| import imagehash | |
| import requests | |
| from PIL import Image | |
| from src.application.image.search_yandex import YandexReverseImageSearcher | |
| def get_image_from_url(url): | |
| try: | |
| response = requests.get(url) | |
| return Image.open(BytesIO(response.content)) | |
| except Exception as e: | |
| print(f"Error opening image: {e}") | |
| return None | |
| def get_image_from_file(file_path): | |
| try: | |
| return Image.open(file_path) | |
| except FileNotFoundError: | |
| print(f"Error occurred while opening image from file: {file_path}") | |
| return None | |
| def standardize_image(image): | |
| # Convert to RGB if needed | |
| if image.mode in ("RGBA", "LA"): | |
| background = Image.new("RGB", image.size, (255, 255, 255)) | |
| background.paste(image, mask=image.split()[-1]) | |
| image = background | |
| elif image.mode != "RGB": | |
| image = image.convert("RGB") | |
| # Resize to standard size (e.g. 256x256) | |
| standard_size = (256, 256) | |
| image = image.resize(standard_size) | |
| return image | |
| def compare_images(image1, image2): | |
| # Standardize both images first | |
| img1_std = standardize_image(image1) | |
| img2_std = standardize_image(image2) | |
| hash1 = imagehash.average_hash(img1_std) | |
| hash2 = imagehash.average_hash(img2_std) | |
| return hash1 - hash2 # Returns the Hamming distance between the hashes | |
| if __name__ == "__main__": | |
| image_url = "https://i.pinimg.com/originals/c4/50/35/c450352ac6ea8645ead206721673e8fb.png" # noqa: E501 | |
| # Get the image from URL | |
| url_image = get_image_from_url(image_url) | |
| # Search image | |
| rev_img_searcher = YandexReverseImageSearcher() | |
| res = rev_img_searcher.search(image_url) | |
| for search_item in res: | |
| print(f"Title: {search_item.page_title}") | |
| # print(f'Site: {search_item.page_url}') | |
| print(f"Img: {search_item.image_url}\n") | |
| # Compare each search result image with the input image | |
| result_image = get_image_from_url(search_item.image_url) | |
| result_difference = compare_images(result_image, url_image) | |
| print(f"Difference with search result: {result_difference}") | |
| if result_difference == 0: | |
| break | |