wip

app.py

@@ -346,26 +346,27 @@ def main():
                                 
                             with gr.Column():
                                 output_dataset_name = gr.Textbox(label="Output Dataset Name", placeholder="e.g., my-generated-dataset, must be unique. Will be created under the org 'synthetic-data-universe'")
-                                num_output_samples = gr.Slider(label="Number of samples, leave as '0' for all", value=0, minimum=0, maximum=MAX_SAMPLES_FREE, step=1)
-                    
-                    with gr.Accordion("Advanced Options (generation parameters, dataset options)", open=False):
-                        with gr.Row():
-                            input_dataset_config = gr.Textbox(label="Input Dataset Config", value="default", placeholder="e.g., default, custom")
-                            input_dataset_split = gr.Textbox(label="Input Dataset Split", value="train", placeholder="e.g., train, test, validation")
-                            model_revision = gr.Textbox(label="Model Revision", value="main", placeholder="e.g., main, v1.0")
                         
-                        with gr.Group():
-                            gr.Markdown("### Generation Parameters")
+                        with gr.Row():
+                            with gr.Column():
+                                input_dataset_config = gr.Dropdown(label="Dataset Config", choices=[], value=None, interactive=False, info="Click Load Info to populate")
+                                prompt_column = gr.Dropdown(label="Prompt Column", choices=[], value=None, interactive=False, info="Click Load Info to populate")
+                                
+                            with gr.Column():
+                                input_dataset_split = gr.Dropdown(label="Dataset Split", choices=[], value=None, interactive=False, info="Click Load Info to populate")
+                                num_output_samples = gr.Slider(label="Number of samples, leave as '0' for all", value=0, minimum=0, maximum=MAX_SAMPLES_FREE, step=1, interactive=False, info="Click Load Info to populate")
+                    
+                    gr.Markdown("### Generation Parameters")
+                    with gr.Row():
+                        with gr.Column():
                             with gr.Row():
-                                with gr.Column():
-                                    with gr.Row():
-                                        max_tokens = gr.Slider(label="Max Tokens", value=1024, minimum=256, maximum=MAX_TOKENS, step=256)
-                                        temperature = gr.Slider(label="Temperature", minimum=0.0, maximum=2.0, value=0.7, step=0.1)
-                                    with gr.Row():
-                                        top_k = gr.Slider(label="Top K", value=50, minimum=5, maximum=100, step=5)
-                                        top_p = gr.Slider(label="Top P", minimum=0.0, maximum=1.0, value=0.95, step=0.05)
-                                    with gr.Row():
-                                        system_prompt = gr.Textbox(label="System Prompt (Optional)", lines=3, placeholder="Optional system prompt... e.g., You are a helpful assistant.")
+                                max_tokens = gr.Slider(label="Max Tokens", value=1024, minimum=256, maximum=MAX_TOKENS, step=256)
+                                temperature = gr.Slider(label="Temperature", minimum=0.0, maximum=2.0, value=0.7, step=0.1)
+                            with gr.Row():
+                                top_k = gr.Slider(label="Top K", value=50, minimum=5, maximum=100, step=5)
+                                top_p = gr.Slider(label="Top P", minimum=0.0, maximum=1.0, value=0.95, step=0.05)
+                    with gr.Row():
+                        system_prompt = gr.Textbox(label="System Prompt (Optional)", lines=3, placeholder="Optional system prompt... e.g., You are a helpful assistant.")
 
                     with gr.Group():
                         gr.Markdown("##  User Information, for notification when your job is completed (still TODO)")