extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-20 18:08:00 +02:00
Code Issues Packages Projects Releases Wiki Activity

simplify DB query

Browse Source
This commit is contained in:
JeLuF 2023-08-08 18:48:08 +02:00
parent 01dde9d23a
commit e828a72b08
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ui/easydiffusion/bucket_manager.py
View File

@ -94,11 +94,10 @@ def init():
def get_image(image_path: str, db: Session = Depends(get_db)):
from easydiffusion.easydb.mappings import Image
image_path = str(abspath(image_path))
amount = len(db.query(Image).filter(Image.path == image_path).all())
if amount > 0:
try:
image = db.query(Image).filter(Image.path == image_path).first()
return FileResponse(image.path)
else:
except Exception as e:
raise HTTPException(status_code=404, detail="Image not found")
@server_api.get("/all_images")