Add files

app.py

@@ -1,12 +1,11 @@
 import os
-import sys
 import threading
 import queue
 import time
 import tempfile
 import shutil
 from contextlib import redirect_stdout, redirect_stderr
-from typing import List, Optional
+from typing import List
 
 import gradio as gr
 from huggingface_hub import HfApi
@@ -189,67 +188,67 @@ with gr.Blocks(title="LeRobot Episode Deleter", theme=gr.themes.Soft()) as demo:
         "Delete specific episodes from a HuggingFace LeRobot dataset and upload the cleaned version."
     )
     
-    with gr.Tabs():
-        with gr.Tab("Step 1: Select Dataset"):
-            with gr.Row():
-                with gr.Column(scale=3):
-                    search_box = gr.Textbox(
-                        label="Search Datasets",
-                        placeholder="Enter keyword or organization name (e.g., 'lerobot', 'griffinlabs-cortex')",
-                        value="griffinlabs-cortex"
-                    )
-                with gr.Column(scale=1):
-                    search_btn = gr.Button("🔍 Search", variant="primary")
-            
-            dataset_dropdown = gr.Dropdown(
-                label="Available Datasets",
-                choices=search_datasets_fn("griffinlabs-cortex"),
-                interactive=True,
-                allow_custom_value=True,
-            )
-            
-            with gr.Row():
-                download_btn = gr.Button("📥 Download & List Episodes", variant="secondary", size="lg")
-            
-            download_status = gr.Textbox(
-                label="Download Status",
-                lines=2,
-                interactive=False,
-            )
-        
-        with gr.Tab("Step 2: Select Episodes to Delete"):
-            gr.Markdown("Select the episodes you want to **remove** from the dataset.")
-            
-            episodes_selector = gr.CheckboxGroup(
-                label="Episodes (select to delete)",
-                choices=[],
-                interactive=True,
-            )
-            
-            selected_count = gr.Markdown("*No episodes selected*")
-        
-        with gr.Tab("Step 3: Delete & Upload"):
-            gr.Markdown("Configure the destination and start the deletion process.")
-            
-            dest_repo_input = gr.Textbox(
-                label="Destination Repository ID",
-                placeholder="your-org/cleaned-dataset",
-                info="The HuggingFace repo where the cleaned dataset will be uploaded"
-            )
-            
-            with gr.Row():
-                execute_btn = gr.Button(
-                    "🚀 Delete Episodes & Upload",
-                    variant="primary",
-                    size="lg",
-                )
-            
-            progress_log = gr.Textbox(
-                label="Progress Log",
-                lines=25,
-                interactive=False,
-                max_lines=30,
-            )
+    # Load initial datasets
+    _initial_choices = search_datasets_fn("griffinlabs-cortex")
+    
+    with gr.Row():
+        search_box = gr.Textbox(
+            label="Search Datasets (optional)",
+            placeholder="Enter keyword or organization name (e.g., 'lerobot', 'griffinlabs-cortex')",
+            value="griffinlabs-cortex"
+        )
+        search_btn = gr.Button("🔍 Search")
+    
+    with gr.Row():
+        dataset_dropdown = gr.Dropdown(
+            label="Select Dataset",
+            choices=_initial_choices,
+            interactive=True,
+            allow_custom_value=True,
+        )
+    
+    with gr.Row():
+        download_btn = gr.Button("📥 Download & List Episodes", variant="secondary", size="lg")
+    
+    download_status = gr.Textbox(
+        label="Status",
+        lines=2,
+        interactive=False,
+    )
+    
+    gr.Markdown("---")
+    gr.Markdown("### Select Episodes to Delete")
+    
+    episodes_selector = gr.CheckboxGroup(
+        label="Episodes (select to delete)",
+        choices=[],
+        interactive=True,
+    )
+    
+    selected_count = gr.Markdown("*No episodes selected*")
+    
+    gr.Markdown("---")
+    gr.Markdown("### Configure Destination & Execute")
+    
+    dest_repo_input = gr.Textbox(
+        label="Destination Repository ID",
+        placeholder="your-org/cleaned-dataset",
+        info="The HuggingFace repo where the cleaned dataset will be uploaded"
+    )
+    
+    with gr.Row():
+        execute_btn = gr.Button(
+            "🚀 Delete Episodes & Upload",
+            variant="primary",
+            size="lg",
+        )
+    
+    progress_log = gr.Textbox(
+        label="Progress Log",
+        lines=20,
+        interactive=False,
+        max_lines=25,
+    )
     
     # Event handlers
     def update_search_results(query):