app.py

import gradio as gr
from datasets import load_dataset

dataset = load_dataset("ShoukanLabs/OpenNiji-Dataset")


startimg = 0

def get_dataset_forward():
    global startimg
    final = []
    for idx in range(startimg, startimg + 50):
        url = dataset["train"]["url"][idx]
        prompt = dataset["train"]["prompt"][idx]
        style = dataset["train"]["style"][idx]
        final.append((url, f"{prompt}\n\n Style: {style}"))
    startimg += 50
    return final

def get_dataset_back():
    global startimg
    final = []
    startimg -= 50
    for idx in range(startimg, startimg + 50):
        url = dataset["train"]["url"][idx]
        prompt = dataset["train"]["prompt"][idx]
        style = dataset["train"]["style"][idx]
        final.append((url, f"{prompt}\n\n Style: {style}"))
    return final

with gr.Blocks() as demo:
    with gr.Column():
        with gr.Row():
            back = gr.Button("<").style()
            forward = gr.Button(">").style()
        gallery = gr.Gallery(
            label="Showing 50 images", show_label=True, elem_id="gallery"
        ).style(object_fit="contain", columns=[10], height="auto")

        back.click(get_dataset_back, None, gallery)
        forward.click(get_dataset_forward, None, gallery)

if __name__ == "__main__":
    demo.launch()