update dataframe visualizations

src/distilabel_dataset_generator/apps/eval.py

@@ -185,7 +185,7 @@ def push_to_hub(
 
 with gr.Blocks() as app:
     gr.Markdown("## 1. Select your input dataset")
-    with gr.Row():
+    with gr.Row(equal_height=False):
         with gr.Column(scale=1):
             search_in = HuggingfaceHubSearch(
                 label="Search",
@@ -199,7 +199,7 @@ with gr.Blocks() as app:
 
     gr.HTML("<hr>")
     gr.Markdown("## 2. Configure your task")
-    with gr.Row():
+    with gr.Row(equal_height=False):
         with gr.Column(scale=1):
             eval_type = gr.Dropdown(
                 label="Evaluation Type",
@@ -252,7 +252,7 @@ with gr.Blocks() as app:
                 )
             btn_apply_to_sample_dataset = gr.Button("Refresh dataset")
         with gr.Column(scale=3):
-            dataframe = gr.Dataframe()
+            dataframe = gr.Dataframe(wrap=True, height=300)
 
     gr.HTML("<hr>")
     gr.Markdown("## 3. Generate your dataset")

src/distilabel_dataset_generator/apps/sft.py

@@ -376,7 +376,7 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
 
         gr.HTML(value="<hr>")
         gr.Markdown(value="## 2. Configure your task")
-        with gr.Row():
+        with gr.Row(equal_height=False):
             with gr.Column(scale=1):
                 system_prompt = gr.Textbox(
                     label="System prompt",
@@ -393,7 +393,9 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
                 )
                 btn_apply_to_sample_dataset = gr.Button("Refresh dataset")
             with gr.Column(scale=3):
-                dataframe = gr.Dataframe()
+                dataframe = gr.Dataframe(
+                    headers=["prompt", "completion"], wrap=True, height=300
+                )
 
         gr.HTML(value="<hr>")
         gr.Markdown(value="## 3. Generate your dataset")

src/distilabel_dataset_generator/apps/textcat.py

@@ -384,7 +384,7 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
 
         gr.HTML("<hr>")
         gr.Markdown("## 2. Configure your task")
-        with gr.Row():
+        with gr.Row(equal_height=False):
             with gr.Column(scale=1):
                 system_prompt = gr.Textbox(
                     label="System prompt",
@@ -436,7 +436,9 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
                 )
                 btn_apply_to_sample_dataset = gr.Button("Refresh dataset")
             with gr.Column(scale=3):
-                dataframe = gr.Dataframe()
+                dataframe = gr.Dataframe(
+                    headers=["labels", "text"], wrap=True, height=300, column_widths=[1, 3]
+                )
 
         gr.HTML("<hr>")
         gr.Markdown("## 3. Generate your dataset")