src/ui/tabs/images/flux_tab.py

import gradio as gr

def flux_tab():
    with gr.Column():
        prompt = gr.Textbox(lines=5, label="Prompt")
    with gr.Column():
        with gr.Group():
            output_image = gr.Image()