klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 21:29:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix extracting image names from multiple image directories

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-07-20 03:40:49 +04:00
parent c9ff97451b
commit c4c7f38b15
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/search_type/image_search.py
View File

@@ -34,11 +34,12 @@ def extract_entries(image_directories, verbose=0):
image_names = [] image_names = []
for image_directory in image_directories: for image_directory in image_directories:
image_directory = resolve_absolute_path(image_directory, strict=True) image_directory = resolve_absolute_path(image_directory, strict=True)
image_names = list(image_directory.glob('*.jpg')) image_names.extend(list(image_directory.glob('*.jpg')))
image_names.extend(list(image_directory.glob('*.jpeg'))) image_names.extend(list(image_directory.glob('*.jpeg')))
if verbose > 0: if verbose > 0:
print(f'Found {len(image_names)} images in {image_directory}') image_directory_names = ', '.join([str(image_directory) for image_directory in image_directories])
print(f'Found {len(image_names)} images in {image_directory_names}')
return image_names return image_names