Better login and description

yourbench_space/app.py

@@ -17,15 +17,14 @@ from yourbench_space.utils import (
 
 project_description = """
 # YourBench 🚀  
-A Dynamic Benchmark Generation Framework 
-- Produce diverse, up-to-date questions from real-world source documents
-- Seamlessly handles ingestion, summarization, and multi-hop chunking for large or specialized datasets
-- Emulates real-world usage scenarios by creating fresh tasks that guard against memorized knowledge
-- Out-of-the-box pipeline stages, plus an easy plugin mechanism to accommodate custom models or domain constraints
+**Dynamic Benchmark Generation for Language Models**
 
-[📖 Github Page](https://github.com/huggingface/yourbench/tree/v0.2-alpha-space)
+Quickly create zero-shot benchmarks from your documents – keeping models accurate and adaptable
+- 📖 [FAQ](#) 
+- 💻 [GitHub](https://github.com/huggingface/yourbench/tree/v0.2-alpha-space)
 """
 
+
 UPLOAD_DIRECTORY.mkdir(parents=True, exist_ok=True)
 
 logger.remove()
@@ -41,6 +40,7 @@ citation_content = (
     else "# Citation\n\nDocumentation file not found."
 )
 
+
 def generate_and_return(hf_org, hf_prefix):
     generate_and_save_config(hf_org, hf_prefix)
     for _ in range(5):
@@ -84,88 +84,103 @@ def update_hf_org_dropdown(oauth_token: gr.OAuthToken | None):
         return gr.Dropdown([], label="Organization")
 
 
+def switch_to_run_generation_tab():
+    return gr.Tabs(selected=1)
+
+
 def enable_button(files):
     return gr.update(interactive=bool(files))
 
 
-with gr.Blocks() as app:
+with gr.Blocks(theme=gr.themes.Default()) as app:
     gr.Markdown(project_description)
-    gr.Markdown("## YourBench Setup")
 
     with gr.Row():
-        login_btn = gr.LoginButton()
-
-    with gr.Tab("Setup"):
-        with gr.Row():
-            with gr.Accordion("Hugging Face Settings"):
-                hf_org_dropdown = gr.Dropdown(
-                    choices=[], label="Organization", allow_custom_value=True
-                )
-                app.load(update_hf_org_dropdown, inputs=None, outputs=hf_org_dropdown)
-
-                hf_dataset_prefix = gr.Textbox(
-                    label="Dataset Prefix",
-                    value="yourbench",
-                    info="Prefix applied to all datasets",
-                )
-
-            with gr.Accordion("Upload documents"):
-                file_input = gr.File(
-                    label="Upload text files",
-                    file_count="multiple",
-                    file_types=[".txt", ".md", ".html"],
-                )
-                output = gr.Textbox(label="Log")
-                file_input.upload(
-                    lambda files: save_files([file.name for file in files]),
-                    file_input,
-                    output,
-                )
-
-        preview_button = gr.Button("Generate New Config", interactive=False)
-        log_message = gr.Textbox(label="Log Message", visible=True)
-        download_button = gr.File(
-            label="Download Config", visible=False, interactive=False
-        )
-
-        file_input.change(enable_button, inputs=file_input, outputs=preview_button)
-
-        preview_button.click(
-            generate_and_return,
-            inputs=[hf_org_dropdown, hf_dataset_prefix],
-            outputs=[log_message, download_button],
-        )
-
-    with gr.Tab("Run Generation"):
-        with gr.Row():
-            with gr.Accordion("Log Output", open=True):
-                log_output = gr.Code(language=None, lines=20, interactive=False)
-
-            with gr.Accordion("Stages", open=True):
-                stages_table = gr.CheckboxGroup(
-                    choices=STAGES,
-                    value=[],
-                    label="Pipeline Stages Completed",
-                    interactive=False,
-                )
-
-        log_timer = gr.Timer(1.0, active=True)
-        log_timer.tick(manager.read_and_get_output, outputs=[log_output, stages_table])
-
-        with gr.Row():
-            process_status = gr.Checkbox(label="Process Status", interactive=False)
-            status_timer = gr.Timer(1.0, active=True)
-            status_timer.tick(manager.is_running, outputs=process_status)
-
-        with gr.Row():
-            start_button = gr.Button("Start Task")
-            start_button.click(prepare_task, inputs=[login_btn, hf_dataset_prefix])
-
-            stop_button = gr.Button("Stop Task")
-            stop_button.click(manager.stop_process)
-
-            kill_button = gr.Button("Kill Task")
-            kill_button.click(manager.kill_process)
+        gr.Markdown("## ⚙️ YourBench Setup", elem_id="setup-title")
+        with gr.Column(scale=0, min_width=150):
+            login_btn = gr.LoginButton()
+
+    with gr.Tabs() as tabs:
+        with gr.Tab("Setup", id=0):
+            with gr.Row():
+                with gr.Accordion("Hugging Face Settings"):
+                    hf_org_dropdown = gr.Dropdown(
+                        choices=[], label="Organization", allow_custom_value=True
+                    )
+                    app.load(
+                        update_hf_org_dropdown, inputs=None, outputs=hf_org_dropdown
+                    )
+
+                    hf_dataset_prefix = gr.Textbox(
+                        label="Dataset Prefix",
+                        value="yourbench",
+                        info="Prefix applied to all datasets",
+                    )
+
+                with gr.Accordion("Upload documents"):
+                    file_input = gr.File(
+                        label="Upload text files",
+                        file_count="multiple",
+                        file_types=[".txt", ".md", ".html"],
+                    )
+                    output = gr.Textbox(label="Log")
+                    file_input.upload(
+                        lambda files: save_files([file.name for file in files]),
+                        file_input,
+                        output,
+                    )
+
+            preview_button = gr.Button("Generate New Config", interactive=False)
+            log_message = gr.Textbox(label="Log Message", visible=True)
+            download_button = gr.File(
+                label="Download Config", visible=False, interactive=False
+            )
+
+            file_input.change(enable_button, inputs=file_input, outputs=preview_button)
+
+            preview_button.click(
+                generate_and_return,
+                inputs=[hf_org_dropdown, hf_dataset_prefix],
+                outputs=[log_message, download_button],
+            )
+            preview_button.click(
+                switch_to_run_generation_tab,
+                inputs=None,
+                outputs=tabs,
+            )
+
+        with gr.Tab("Run Generation", id=1):
+            with gr.Row():
+                with gr.Accordion("Log Output", open=True):
+                    log_output = gr.Code(language=None, lines=20, interactive=False)
+
+                with gr.Accordion("Stages", open=True):
+                    stages_table = gr.CheckboxGroup(
+                        choices=STAGES,
+                        value=[],
+                        label="Pipeline Stages Completed",
+                        interactive=False,
+                    )
+
+            log_timer = gr.Timer(1.0, active=True)
+            log_timer.tick(
+                manager.read_and_get_output, outputs=[log_output, stages_table]
+            )
+
+            with gr.Row():
+                process_status = gr.Checkbox(label="Process Status", interactive=False)
+                status_timer = gr.Timer(1.0, active=True)
+                status_timer.tick(manager.is_running, outputs=process_status)
+
+            with gr.Row():
+                start_button = gr.Button("Start Task")
+                start_button.click(prepare_task, inputs=[login_btn, hf_dataset_prefix])
+
+                stop_button = gr.Button("Stop Task")
+                stop_button.click(manager.stop_process)
+
+                kill_button = gr.Button("Kill Task")
+                kill_button.click(manager.kill_process)
 
     with gr.Accordion("📜 Citation", open=False):
         gr.Markdown(citation_content)