ulysia/szurubooru
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

If attempting to reverse-search using a PSD, convert to PNG first

Browse source
This commit is contained in:
Deer-Spangle 2024-09-01 15:40:01 +01:00
parent d102578b54
commit 9ccff1b69c
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
server/szurubooru/func/posts.py
View file

@ -928,6 +928,12 @@ def search_by_image_exact(image_content: bytes) -> Optional[model.Post]:
def search_by_image(image_content: bytes) -> List[Tuple[float, model.Post]]: def search_by_image(image_content: bytes) -> List[Tuple[float, model.Post]]:
# Convert PSD files to PNG so pillow can open them
if image_content[0:4] == b"\x38\x42\x50\x53":
image = images.Image(image_content)
image.resize_fill(image.width, image.height)
image_content = image.content
query_signature = image_hash.generate_signature(image_content) query_signature = image_hash.generate_signature(image_content)
query_words = image_hash.generate_words(query_signature) query_words = image_hash.generate_words(query_signature)