src/ui/images.py

import os

import gradio as gr

from .tabs import (
    flux_tab,
)

def image_tab():
    with gr.Tabs():
        with gr.Tab(label="Flux"):
            flux_tab()
        with gr.Tab(label="SDXL"):
            gr.Label("Coming soon!")