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

Image input directory not required by collate result as image_name already absolute path

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-07-20 02:56:23 +04:00
parent 6c9ffdba57
commit 2a5445216c
2 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/main.py
View File

@@ -94,7 +94,6 @@ def search(q: str, n: Optional[int] = 5, t: Optional[SearchType] = None):
return image_search.collate_results(
hits,
image_names=model.image_search.image_names,
image_directory=config.content_type.image.input_directory,
output_directory=output_directory,
static_files_url='/static',
count=results_count)

6
src/search_type/image_search.py
View File

@@ -169,13 +169,11 @@ def render_results(hits, image_names, image_directory, count):
img.show()
def collate_results(hits, image_names, image_directory, output_directory, static_files_url, count=5):
def collate_results(hits, image_names, output_directory, static_files_url, count=5):
results = []
image_directory = resolve_absolute_path(image_directory, strict=True)
for index, hit in enumerate(hits[:count]):
source_image_name = image_names[hit['corpus_id']]
source_path = image_directory.joinpath(source_image_name)
source_path = image_names[hit['corpus_id']]
target_image_name = f"{index}{source_path.suffix}"
target_path = resolve_absolute_path(f"{output_directory}/{target_image_name}")