Update app.py

app.py

@@ -65,40 +65,43 @@ def generate_dataset(dataset_name):
 with gr.Blocks() as demo:
     gr.Markdown("# Dataset Builder")
     
-    with gr.Tab("Create/Edit Dataset"):
-        dataset_name = gr.Textbox(label="Dataset Name")
-        image = gr.Image(label="Upload Image")
-        prompt = gr.Textbox(label="Prompt")
-        add_button = gr.Button("Add to Dataset")
-        add_output = gr.Textbox()
-        add_button.click(add_to_dataset, inputs=[dataset_name, image, prompt], outputs=add_output)
+    with gr.Row():
+        with gr.Column(scale=2):
+            gr.Markdown("## Create/Edit Dataset")
+            dataset_name = gr.Textbox(label="Dataset Name")
+            image = gr.Image(label="Upload Image")
+            prompt = gr.Textbox(label="Prompt")
+            add_button = gr.Button("Add to Dataset")
+            add_output = gr.Textbox()
+            add_button.click(add_to_dataset, inputs=[dataset_name, image, prompt], outputs=add_output)
+            
+            index = gr.Number(label="Index to Edit/Delete", precision=0)
+            new_prompt = gr.Textbox(label="New Prompt")
+            edit_button = gr.Button("Edit Prompt")
+            edit_output = gr.Textbox()
+            edit_button.click(edit_dataset, inputs=[dataset_name, index, new_prompt], outputs=edit_output)
+            
+            delete_button = gr.Button("Delete from Dataset")
+            delete_output = gr.Textbox()
+            delete_button.click(delete_from_dataset, inputs=[dataset_name, index], outputs=delete_output)
         
-        index = gr.Number(label="Index to Edit/Delete", precision=0)
-        new_prompt = gr.Textbox(label="New Prompt")
-        edit_button = gr.Button("Edit Prompt")
-        edit_output = gr.Textbox()
-        edit_button.click(edit_dataset, inputs=[dataset_name, index, new_prompt], outputs=edit_output)
-        
-        delete_button = gr.Button("Delete from Dataset")
-        delete_output = gr.Textbox()
-        delete_button.click(delete_from_dataset, inputs=[dataset_name, index], outputs=delete_output)
-    
-    with gr.Tab("Upload Dataset"):
-        upload_dataset_name = gr.Textbox(label="Dataset Name")
-        upload_file = gr.File(label="Upload JSONL File")
-        upload_button = gr.Button("Upload Dataset")
-        upload_output = gr.Textbox()
-        upload_button.click(load_dataset, inputs=[upload_dataset_name, upload_file], outputs=upload_output)
-    
-    with gr.Tab("Generate/Download Dataset"):
-        generate_button = gr.Button("Generate Dataset")
-        generate_output = gr.File()
-        generate_button.click(generate_dataset, inputs=[dataset_name], outputs=generate_output)
-    
-    with gr.Tab("View Dataset"):
-        view_dataset_name = gr.Textbox(label="Dataset Name")
-        view_button = gr.Button("View Dataset")
-        view_output = gr.JSON()
-        view_button.click(lambda name: datasets.get(name, []), inputs=[view_dataset_name], outputs=view_output)
+        with gr.Column(scale=1):
+            gr.Markdown("## Upload Dataset")
+            upload_dataset_name = gr.Textbox(label="Dataset Name")
+            upload_file = gr.File(label="Upload JSONL File")
+            upload_button = gr.Button("Upload Dataset")
+            upload_output = gr.Textbox()
+            upload_button.click(load_dataset, inputs=[upload_dataset_name, upload_file], outputs=upload_output)
+            
+            gr.Markdown("## Generate/Download Dataset")
+            generate_button = gr.Button("Generate Dataset")
+            generate_output = gr.File()
+            generate_button.click(generate_dataset, inputs=[dataset_name], outputs=generate_output)
+            
+            gr.Markdown("## View Dataset")
+            view_dataset_name = gr.Textbox(label="Dataset Name")
+            view_button = gr.Button("View Dataset")
+            view_output = gr.JSON()
+            view_button.click(lambda name: datasets.get(name, []), inputs=[view_dataset_name], outputs=view_output)
 
 demo.launch()