ui/captions.py

import gradio as gr

def render_captions():
    with gr.Column(visible=False) as page:
        gr.Markdown("# 📝 Caption Browser")
        with gr.Row():
            refresh_cap_btn = gr.Button("🔄 Refresh Data", variant="secondary")
            cap_status = gr.Textbox(interactive=False, show_label=False, scale=4)
            
        caption_table = gr.Dataframe(
            headers=["File", "Summary", "Attire", "Tags"],
            datatype=["str", "str", "str", "str"],
            wrap=True, interactive=False,
            column_widths=["15%", "45%", "20%", "20%"],
        )
    return page, refresh_cap_btn, cap_status, caption_table