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

Added /all_images route that serves all images in DB

Browse Source
This commit is contained in:
ManInDark 2023-08-06 18:22:25 +02:00
parent 9cfc7511b4
commit 8994378bdb
No known key found for this signature in database
GPG Key ID: 72EC12A5B2D62779
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ui/easydiffusion/bucket_manager.py
View File

@ -94,12 +94,21 @@ def init():
from easydiffusion.easydb.mappings import Image from easydiffusion.easydb.mappings import Image
image_path = image_path.replace("/", "\\") image_path = image_path.replace("/", "\\")
amount = len(db.query(Image).filter(Image.path == image_path).all()) amount = len(db.query(Image).filter(Image.path == image_path).all())
print(image_path, amount)
if amount > 0: if amount > 0:
image = db.query(Image).filter(Image.path == image_path).first() image = db.query(Image).filter(Image.path == image_path).first()
return FileResponse(image.path) return FileResponse(image.path)
else: else:
raise HTTPException(status_code=404, detail="Image not found") raise HTTPException(status_code=404, detail="Image not found")
@server_api.get("/all_images")
def get_all_images(db: Session = Depends(get_db)):
from easydiffusion.easydb.mappings import Image
images = db.query(Image).all()
sum_string = ""
for img in images:
options = f"Path: {img.path}\nPrompt: \nNegative Prompt: \nSeed: {img.seed}\nModel: {img.use_stable_diffusion_model}\nSize: {img.height}x{img.width}\nSampler: {img.sampler_name}\nSteps: {img.num_inference_steps}\nGuidance Scale: {img.guidance_scale}\nLoRA: {img.lora}\nUpscaling: {img.use_upscale}\nFace Correction: {img.use_face_correction}\n"
sum_string += f"<img src='/image/{img.path}' title='{options}'>"
return Response(content=sum_string, media_type="text/html")
def get_filename_from_url(url): def get_filename_from_url(url):