sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-23 23:48:56 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 16

Create images directory if doesn't exist, to store image search results

Browse source
This commit is contained in:
Debanjum Singh Solanky 2022-07-28 21:30:31 +04:00
parent a29af70de5
commit 4d5183063c
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/search_type/image_search.py
View file

@ -190,6 +190,10 @@ def collate_results(hits, image_names, output_directory, image_files_url, count=
target_image_name = f"{index}{source_path.suffix}" target_image_name = f"{index}{source_path.suffix}"
target_path = resolve_absolute_path(f"{output_directory}/{target_image_name}") target_path = resolve_absolute_path(f"{output_directory}/{target_image_name}")
# Create output directory, if it doesn't exist
if not target_path.parent.exists():
target_path.parent.mkdir(exist_ok=True)
# Copy the image to the output directory # Copy the image to the output directory
shutil.copy(source_path, target_path) shutil.copy(source_path, target_path)