DominikDoom/a1111-sd-webui-tagcomplete
1
0
Fork 0
mirror of https://github.com/DominikDoom/a1111-sd-webui-tagcomplete.git synced 2026-01-26 19:19:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for lora filenames containing dots

Browse Source 
Since file extensions are already cut off before the client-side request, it's not needed here anymore
This commit is contained in:
DominikDoom
2023-08-07 19:22:50 +02:00
parent 995a5ecdba
commit ebe276ee44
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/tag_autocomplete_helper.py
View File

@@ -453,8 +453,7 @@ def api_tac(_: gr.Blocks, app: FastAPI):
return json.dumps({})
try:
name = Path(filename).stem
json_candidates = glob.glob(base_path.as_posix() + f"/**/{name}.json", recursive=True)
json_candidates = glob.glob(base_path.as_posix() + f"/**/{filename}.json", recursive=True)
if json_candidates is not None and len(json_candidates) > 0:
return FileResponse(json_candidates[0])
except Exception as e:
@@ -465,8 +464,7 @@ def api_tac(_: gr.Blocks, app: FastAPI):
return json.dumps({})
try:
name = Path(filename).stem
img_glob = glob.glob(base_path.as_posix() + f"/**/{name}.*", recursive=True)
img_glob = glob.glob(base_path.as_posix() + f"/**/{filename}.*", recursive=True)
img_candidates = [img for img in img_glob if Path(img).suffix in [".png", ".jpg", ".jpeg", ".webp"]]
if img_candidates is not None and len(img_candidates) > 0:
return JSONResponse({"url": urllib.parse.quote(img_candidates[0])})