diff --git a/ui/easydiffusion/bucket_manager.py b/ui/easydiffusion/bucket_manager.py
index f14c7bc0..e8d65a48 100644
--- a/ui/easydiffusion/bucket_manager.py
+++ b/ui/easydiffusion/bucket_manager.py
@@ -100,13 +100,14 @@ def init():
raise HTTPException(status_code=404, detail="Image not found")
@server_api.get("/all_images")
- def get_all_images(prompt: str = "", model: str = "", db: Session = Depends(get_db)):
+ def get_all_images(prompt: str = "", model: str = "", page = 0, db: Session = Depends(get_db)):
from easydiffusion.easydb.mappings import GalleryImage
images = db.query(GalleryImage)
if prompt != "":
images = images.filter(GalleryImage.path.like("%"+prompt+"%"))
if model != "":
images = images.filter(GalleryImage.use_stable_diffusion_model.like("%"+model+"%"))
+ images = images.offset(page*50).limit(50)
return images.all()
@server_api.get("/single_image")
diff --git a/ui/index.html b/ui/index.html
index 9aeb7ab2..5676ab5c 100644
--- a/ui/index.html
+++ b/ui/index.html
@@ -520,6 +520,7 @@
+
diff --git a/ui/media/js/main.js b/ui/media/js/main.js
index d54913fa..34f84399 100644
--- a/ui/media/js/main.js
+++ b/ui/media/js/main.js
@@ -3110,12 +3110,13 @@ function galleryImage(item) {
function refreshGallery() {
let container = document.getElementById("imagecontainer")
- let promptsearchfield = document.getElementById("gallery-prompt-search").value
- let promptsearch = promptsearchfield.length > 0 ? "prompt=" + promptsearchfield + "&" : ""
- let modelsearchfield = document.getElementById("gallery-model-search").value
- let modelsearch = modelsearchfield.length > 0 ? "model=" + modelsearchfield + "&" : ""
+ params = new URLSearchParams({
+ prompt: document.getElementById("gallery-prompt-search").value,
+ model: document.getElementById("gallery-model-search").value,
+ page: document.getElementById("gallery-page").value
+ })
container.innerHTML=""
- fetch('/all_images?' + promptsearch + modelsearch)
+ fetch('/all_images?' + params)
.then(response => response.json())
.then(json => {
console.log(json)