UI suggestions

app.py

@@ -180,26 +180,23 @@ def generate_cells(dataset_id, notebook_title):
     )
     html_content = create_notebook_file(cells, notebook_name=notebook_name)
     _push_to_hub(dataset_id, notebook_name)
-    notebook_link = f"https://colab.research.google.com/#fileId=https%3A//huggingface.co/datasets/asoria/dataset-notebook-creator-content/blob/main/{notebook_name}"
+    notebook_link = f"https://colab.research.google.com/#fileId=https%3A//huggingface.co/datasets/{NOTEBOOKS_REPOSITORY}/blob/main/{notebook_name}"
     return (
         html_content,
-        f"## 🎉 Ready to explore? Play and run the generated notebook 👉 [here]({notebook_link})!",
+        f"[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)]({notebook_link})",
     )
 
 
 css = """
-#box {
-  height: 650px;
-  overflow-y: scroll !important;
+.prose :where(pre):not(:where([class~=not-prose],[class~=not-prose] *)) {
+    background-color: var(--table-even-background-fill);  /* Fix dark mode */
 }
 """
 
-with gr.Blocks(
-    fill_height=True,
-    fill_width=True,
-    css=css,
-) as demo:
+with gr.Blocks(css=css) as demo:
     gr.Markdown("# 🤖 Dataset notebook creator 🕵️")
+    gr.Markdown(f"[![Notebooks: {len(notebook_templates)}](https://img.shields.io/badge/Notebooks-{len(notebook_templates)}-blue.svg)](tree/main/notebooks)")
+    gr.Markdown("[![Contribute a Notebook](https://img.shields.io/badge/Contribute%20a%20Notebook-8A2BE2)](blob/main/CONTRIBUTING.md)")
     text_input = gr.Textbox(label="Suggested notebook type", visible=False)
 
     gr.Markdown("## 1. Select and preview a dataset from Huggingface Hub")
@@ -244,24 +241,16 @@ with gr.Blocks(
         return gr.HTML(value=html_code, elem_classes="viewer")
 
     gr.Markdown("## 2. Select the type of notebook you want to generate")
-    with gr.Row():
-        notebook_type = gr.Dropdown(
-            choices=notebook_templates.keys(),
-            label="Notebook type",
-            value="Text Embeddings",
-        )
-        generate_button = gr.Button("Generate Notebook", variant="primary")
-        contribute_btn = gr.Button(
-            "Or Contribute",
-            visible=True,
-            variant="secondary",
-            size="sm",
-            link="https://huggingface.co/spaces/asoria/auto-notebook-creator/blob/main/CONTRIBUTING.md",
-        )
+    notebook_type = gr.Dropdown(
+        choices=notebook_templates.keys(),
+        label="Notebook type",
+        value="Text Embeddings",
+    )
+    generate_button = gr.Button("Generate Notebook", variant="primary")
 
-    gr.Markdown("## 3. Notebook code result")
-    code_component = gr.HTML(elem_id="box")
-    go_to_notebook = gr.Markdown("", visible=True)
+    gr.Markdown("## 3. Notebook result + Open in Colab")
+    go_to_notebook = gr.Markdown()
+    code_component = gr.HTML()
 
     generate_button.click(
         generate_cells,
@@ -269,8 +258,6 @@ with gr.Blocks(
         outputs=[code_component, go_to_notebook],
     )
 
-    gr.Markdown(
-        "🚧 Note: Some code may not be compatible with datasets that contain binary data or complex structures. 🚧"
-    )
+    gr.Markdown("🚧 Note: Some code may not be compatible with datasets that contain binary data or complex structures. 🚧")
 
 demo.launch()