File size: 3,503 Bytes
9f5c8f7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 |
"""
Gradio UI Dataset Section Module
Contains dataset explorer section component definitions
"""
import gradio as gr
def create_dataset_section(dataset_handler) -> dict:
"""Create dataset explorer section"""
with gr.Accordion("📊 Dataset Explorer", open=False, visible=False):
with gr.Row(equal_height=True):
dataset_type = gr.Dropdown(
choices=["train", "test"],
value="train",
label="Dataset",
info="Choose dataset to explore",
scale=2
)
import_dataset_btn = gr.Button("📥 Import Dataset", variant="primary", scale=1)
search_type = gr.Dropdown(
choices=["keys", "idx", "random"],
value="random",
label="Search Type",
info="How to find items",
scale=1
)
search_value = gr.Textbox(
label="Search Value",
placeholder="Enter keys or index (leave empty for random)",
info="Keys: exact match, Index: 0 to dataset size-1",
scale=2
)
instruction_display = gr.Textbox(
label="📝 Instruction",
interactive=False,
placeholder="No instruction available",
lines=1
)
repaint_viz_plot = gr.Plot()
with gr.Accordion("📋 Item Metadata (JSON)", open=False):
item_info_json = gr.Code(
label="Complete Item Information",
language="json",
interactive=False,
lines=15
)
with gr.Row(equal_height=True):
item_src_audio = gr.Audio(
label="Source Audio",
type="filepath",
interactive=False,
scale=8
)
get_item_btn = gr.Button("🔍 Get Item", variant="secondary", interactive=False, scale=2)
with gr.Row(equal_height=True):
item_target_audio = gr.Audio(
label="Target Audio",
type="filepath",
interactive=False,
scale=8
)
item_refer_audio = gr.Audio(
label="Reference Audio",
type="filepath",
interactive=False,
scale=2
)
with gr.Row():
use_src_checkbox = gr.Checkbox(
label="Use Source Audio from Dataset",
value=True,
info="Check to use the source audio from dataset"
)
data_status = gr.Textbox(label="📊 Data Status", interactive=False, value="❌ No dataset imported")
auto_fill_btn = gr.Button("📋 Auto-fill Generation Form", variant="primary")
return {
"dataset_type": dataset_type,
"import_dataset_btn": import_dataset_btn,
"search_type": search_type,
"search_value": search_value,
"instruction_display": instruction_display,
"repaint_viz_plot": repaint_viz_plot,
"item_info_json": item_info_json,
"item_src_audio": item_src_audio,
"get_item_btn": get_item_btn,
"item_target_audio": item_target_audio,
"item_refer_audio": item_refer_audio,
"use_src_checkbox": use_src_checkbox,
"data_status": data_status,
"auto_fill_btn": auto_fill_btn,
}
|