import os
import gradio as gr
from glob import glob
models = glob("models/*")
#cats = glob("models/coco_SDXL_simpleprompt/*")


with gr.Blocks() as demo:
    for model in models:
        with gr.Tab(model.split("/")[-1]) as tab:
            cats = glob(model+"/*")
            for cat in cats:
                with gr.Tab(cat.split("/")[-1]) as tab:
                    gallery = gr.Gallery(value=glob(os.path.join(cat, "*.jpg")),label="Generated images", show_label=False, elem_id="gallery",columns=[5], rows=[1], object_fit="contain", height="auto")
                
    
        
    

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