Change UI layout

app.py

@@ -22,12 +22,9 @@ from nbconvert import HTMLExporter
 load_dotenv()
 
 HF_TOKEN = os.getenv("HF_TOKEN")
-NOTEBOOKS_REPOSITORY = os.getenv("NOTEBOOKS_REPOSITORY")
 assert HF_TOKEN is not None, "You need to set HF_TOKEN in your environment variables"
-assert (
-    NOTEBOOKS_REPOSITORY is not None
-), "You need to set NOTEBOOKS_REPOSITORY in your environment variables"
 
+NOTEBOOKS_REPOSITORY = "asoria/dataset-notebook-creator-content"
 
 BASE_DATASETS_SERVER_URL = "https://datasets-server.huggingface.co"
 HEADERS = {"Accept": "application/json", "Content-Type": "application/json"}
@@ -208,41 +205,40 @@ with gr.Blocks(
     css=css,
 ) as demo:
     gr.Markdown("# 🤖 Dataset notebook creator 🕵️")
-    with gr.Row(equal_height=True):
-        with gr.Column(scale=2):
-            text_input = gr.Textbox(label="Suggested notebook type", visible=False)
-
-            dataset_name = HuggingfaceHubSearch(
-                label="Hub Dataset ID",
-                placeholder="Search for dataset id on Huggingface",
-                search_type="dataset",
-                value="",
-            )
-
-            dataset_samples = gr.Examples(
-                examples=[
-                    [
-                        "scikit-learn/iris",
-                        "Try this dataset for Exploratory Data Analysis",
-                    ],
-                    [
-                        "infinite-dataset-hub/GlobaleCuisineRecipes",
-                        "Try this dataset for Embeddings generation",
-                    ],
-                    [
-                        "infinite-dataset-hub/GlobalBestSellersSummaries",
-                        "Try this dataset for RAG generation",
-                    ],
-                ],
-                inputs=[dataset_name, text_input],
-                cache_examples=False,
-            )
-
-            @gr.render(inputs=dataset_name)
-            def embed(name):
-                if not name:
-                    return gr.Markdown("### No dataset provided")
-                html_code = f"""
+    text_input = gr.Textbox(label="Suggested notebook type", visible=False)
+
+    gr.Markdown("## 1. Select and preview a dataset from Huggingface Hub")
+    dataset_name = HuggingfaceHubSearch(
+        label="Hub Dataset ID",
+        placeholder="Search for dataset id on Huggingface",
+        search_type="dataset",
+        value="",
+    )
+
+    dataset_samples = gr.Examples(
+        examples=[
+            [
+                "scikit-learn/iris",
+                "Try this dataset for Exploratory Data Analysis",
+            ],
+            [
+                "infinite-dataset-hub/GlobaleCuisineRecipes",
+                "Try this dataset for Embeddings generation",
+            ],
+            [
+                "infinite-dataset-hub/GlobalBestSellersSummaries",
+                "Try this dataset for RAG generation",
+            ],
+        ],
+        inputs=[dataset_name, text_input],
+        cache_examples=False,
+    )
+
+    @gr.render(inputs=dataset_name)
+    def embed(name):
+        if not name:
+            return gr.Markdown("### No dataset provided")
+        html_code = f"""
                     <iframe
                     src="https://huggingface.co/datasets/{name}/embed/viewer/default/train"
                     frameborder="0"
@@ -250,19 +246,22 @@ with gr.Blocks(
                     height="350px"
                     ></iframe>
                     """
-                return gr.HTML(value=html_code, elem_classes="viewer")
-
-            with gr.Row():
-                generate_eda_btn = gr.Button("Exploratory Data Analysis")
-                generate_embedding_btn = gr.Button("Embeddings")
-                generate_rag_btn = gr.Button("RAG")
-                generate_training_btn = gr.Button(
-                    "Training - Coming soon", interactive=False
-                )
-
-        with gr.Column(scale=2):
-            code_component = gr.HTML(elem_id="box")
-            go_to_notebook = gr.Markdown("", visible=True)
+        return gr.HTML(value=html_code, elem_classes="viewer")
+
+    gr.Markdown("## 2. Select the type of notebook you want to generate")
+    with gr.Row():
+        with gr.Column():
+            generate_eda_btn = gr.Button("EDA", size="sm")
+        with gr.Column():
+            generate_embedding_btn = gr.Button("Embeddings", size="sm")
+        with gr.Column():
+            generate_rag_btn = gr.Button("RAG", size="sm")
+        with gr.Column():
+            generate_training_btn = gr.Button("Training", interactive=False, size="sm")
+
+    gr.Markdown("## 3. Notebook code result")
+    code_component = gr.HTML(elem_id="box")
+    go_to_notebook = gr.Markdown("", visible=True)
 
     generate_eda_btn.click(
         generate_eda_cells,