refactor: reorganize verification modes and add Edit Enhanced Datasets tab

- Add new 'Edit Enhanced Datasets' tab to main menu for dataset management
- Replace 'Enhanced Datasets' mode with 'Standard Verification' in Enhanced Verification
- Create standalone dataset_editor_interface.py for dataset editing functionality
- Remove dataset editing from Enhanced Verification (now in separate tab)
- Update mode selection to show: Standard Verification, Manual Input, File Upload
- Clean up unused event handlers and references

src/interface/dataset_editor_interface.py

@@ -0,0 +1,370 @@
+# dataset_editor_interface.py
+"""
+Dataset Editor Interface for Enhanced Datasets.
+
+Provides a standalone interface for editing test datasets with full CRUD operations.
+"""
+
+import gradio as gr
+from typing import List, Dict, Tuple, Optional, Any
+from datetime import datetime
+import uuid
+
+from src.core.verification_models import TestMessage, TestDataset
+from src.interface.enhanced_dataset_interface import EnhancedDatasetInterfaceController
+
+
+def create_dataset_editor_interface() -> gr.Blocks:
+    """
+    Create the dataset editor interface.
+    
+    Returns:
+        Gradio Blocks component for dataset editing
+    """
+    with gr.Blocks(title="Edit Enhanced Datasets") as interface:
+        # Initialize controller
+        dataset_controller = EnhancedDatasetInterfaceController()
+        
+        # Pre-load dataset choices for initialization
+        try:
+            initial_choices, _, _, _ = dataset_controller.initialize_interface()
+        except Exception:
+            initial_choices = []
+        
+        # Application state
+        current_dataset_state = gr.State(value=None)
+        editing_dataset_state = gr.State(value=None)
+        
+        # Header
+        gr.Markdown("# ✏️ Edit Enhanced Datasets")
+        gr.Markdown("Manage and customize test datasets for verification. Add, modify, or delete test cases to create datasets tailored to your testing needs.")
+        
+        # Dataset selection interface
+        with gr.Row():
+            with gr.Column(scale=2):
+                gr.Markdown("## 📋 Select Dataset")
+                
+                dataset_selector = gr.Dropdown(
+                    choices=initial_choices,
+                    value=initial_choices[0] if initial_choices else None,
+                    label="Available Datasets",
+                    info="Choose a dataset to edit",
+                    interactive=True
+                )
+                
+                with gr.Row():
+                    edit_dataset_btn = gr.Button("✏️ Edit Dataset", variant="primary", scale=2)
+                    create_dataset_btn = gr.Button("➕ Create New Dataset", variant="secondary", scale=1)
+            
+            with gr.Column(scale=1):
+                gr.Markdown("## 📊 Dataset Information")
+                dataset_info_display = gr.Markdown(
+                    "Select a dataset to view details",
+                    label="Dataset Details"
+                )
+        
+        # Create new dataset section (initially hidden)
+        create_section = gr.Row(visible=False)
+        with create_section:
+            with gr.Column():
+                gr.Markdown("## ➕ Create New Dataset")
+                
+                with gr.Row():
+                    with gr.Column(scale=2):
+                        new_dataset_name = gr.Textbox(
+                            label="Dataset Name",
+                            placeholder="Enter dataset name...",
+                            interactive=True
+                        )
+                    with gr.Column(scale=2):
+                        new_dataset_description = gr.Textbox(
+                            label="Description",
+                            placeholder="Describe the dataset purpose...",
+                            interactive=True
+                        )
+                    with gr.Column(scale=1):
+                        template_selector = gr.Dropdown(
+                            choices=["", "empty", "mixed_sample"],
+                            label="Template",
+                            value="",
+                            info="Optional template"
+                        )
+                
+                with gr.Row():
+                    create_confirm_btn = gr.Button("✅ Create Dataset", variant="primary")
+                    create_cancel_btn = gr.Button("❌ Cancel", variant="secondary")
+        
+        # Edit Dataset interface (initially hidden)
+        edit_section = gr.Row(visible=False)
+        with edit_section:
+            with gr.Column():
+                gr.Markdown("## ✏️ Edit Dataset")
+                
+                # Dataset actions
+                with gr.Row():
+                    save_dataset_btn = gr.Button("💾 Save Changes", variant="primary")
+                    close_edit_btn = gr.Button("✖️ Close Editor", variant="secondary")
+                
+                # Test cases display
+                test_cases_display = gr.HTML(
+                    value="<p>Loading test cases...</p>",
+                    label="Test Cases"
+                )
+                
+                gr.Markdown("### ➕ Add New Test Case")
+                with gr.Row():
+                    with gr.Column(scale=3):
+                        new_message_text = gr.Textbox(
+                            label="Message Text",
+                            placeholder="Enter the test message...",
+                            lines=3,
+                            interactive=True
+                        )
+                    with gr.Column(scale=1):
+                        new_message_classification = gr.Radio(
+                            choices=["green", "yellow", "red"],
+                            label="Expected Classification",
+                            value="yellow"
+                        )
+                
+                with gr.Row():
+                    add_test_case_btn = gr.Button("➕ Add Test Case", variant="primary")
+                    clear_form_btn = gr.Button("🗑️ Clear Form", variant="secondary")
+        
+        # Status message
+        status_message = gr.Markdown("", visible=True)
+        
+        # Event handlers
+        def on_dataset_selection_change(dataset_selection):
+            """Handle dataset selection change."""
+            try:
+                dataset_info, dataset_obj = dataset_controller.get_dataset_info(dataset_selection)
+                return (
+                    dataset_info,     # dataset_info_display
+                    dataset_obj       # current_dataset_state
+                )
+            except Exception as e:
+                return (
+                    f"❌ Error loading dataset info: {str(e)}",  # dataset_info_display
+                    None  # current_dataset_state
+                )
+        
+        def on_edit_dataset(current_dataset):
+            """Handle edit dataset button."""
+            try:
+                if not current_dataset:
+                    return (
+                        gr.Row(visible=False),  # edit_section
+                        "",  # test_cases_display
+                        current_dataset,  # editing_dataset_state
+                        "❌ No dataset selected for editing"  # status_message
+                    )
+                
+                # Render test cases HTML
+                test_cases_html = dataset_controller.render_test_cases_display(current_dataset)
+                
+                return (
+                    gr.Row(visible=True),  # edit_section
+                    test_cases_html,  # test_cases_display
+                    current_dataset,  # editing_dataset_state
+                    f"✏️ Editing dataset: '{current_dataset.name}'"  # status_message
+                )
+            except Exception as e:
+                return (
+                    gr.Row(visible=False),  # edit_section
+                    "",  # test_cases_display
+                    None,  # editing_dataset_state
+                    f"❌ Error accessing dataset for editing: {str(e)}"  # status_message
+                )
+        
+        def on_close_edit():
+            """Close the edit section."""
+            return (
+                gr.Row(visible=False),  # edit_section
+                "",  # new_message_text (clear)
+                "yellow",  # new_message_classification (reset)
+                "✅ Editor closed"  # status_message
+            )
+        
+        def on_show_create():
+            """Show create dataset section."""
+            return (
+                gr.Row(visible=True),  # create_section
+                "➕ Create a new dataset"  # status_message
+            )
+        
+        def on_cancel_create():
+            """Cancel create dataset."""
+            return (
+                gr.Row(visible=False),  # create_section
+                "",  # new_dataset_name (clear)
+                "",  # new_dataset_description (clear)
+                "",  # template_selector (reset)
+                "❌ Dataset creation cancelled"  # status_message
+            )
+        
+        def on_create_dataset(name, description, template):
+            """Create a new dataset."""
+            try:
+                success, message, new_dataset = dataset_controller.create_new_dataset(
+                    name, description, template if template else None
+                )
+                
+                if success:
+                    # Refresh dataset choices
+                    updated_choices, _, _, _ = dataset_controller.initialize_interface()
+                    
+                    return (
+                        gr.Row(visible=False),  # create_section
+                        gr.update(choices=updated_choices, value=f"{new_dataset.name} ({new_dataset.message_count} messages)"),  # dataset_selector
+                        "",  # new_dataset_name (clear)
+                        "",  # new_dataset_description (clear)
+                        "",  # template_selector (reset)
+                        new_dataset,  # current_dataset_state
+                        message  # status_message
+                    )
+                else:
+                    return (
+                        gr.Row(visible=True),  # create_section (keep visible)
+                        gr.update(),  # dataset_selector (no change)
+                        name,  # new_dataset_name (keep)
+                        description,  # new_dataset_description (keep)
+                        template,  # template_selector (keep)
+                        None,  # current_dataset_state
+                        message  # status_message
+                    )
+            except Exception as e:
+                return (
+                    gr.Row(visible=True),  # create_section (keep visible)
+                    gr.update(),  # dataset_selector (no change)
+                    name,  # new_dataset_name (keep)
+                    description,  # new_dataset_description (keep)
+                    template,  # template_selector (keep)
+                    None,  # current_dataset_state
+                    f"❌ Error creating dataset: {str(e)}"  # status_message
+                )
+        
+        def on_add_test_case(editing_dataset, message_text, classification):
+            """Add a new test case to the dataset."""
+            try:
+                if not editing_dataset:
+                    return (
+                        "",  # test_cases_display
+                        editing_dataset,  # editing_dataset_state
+                        message_text,  # new_message_text (keep)
+                        classification,  # new_message_classification (keep)
+                        "❌ No dataset being edited"  # status_message
+                    )
+                
+                if not message_text or not message_text.strip():
+                    return (
+                        dataset_controller.render_test_cases_display(editing_dataset),  # test_cases_display
+                        editing_dataset,  # editing_dataset_state
+                        message_text,  # new_message_text (keep)
+                        classification,  # new_message_classification (keep)
+                        "❌ Message text is required"  # status_message
+                    )
+                
+                # Add test case
+                success, message, updated_dataset = dataset_controller.add_test_case(
+                    editing_dataset, message_text.strip(), classification
+                )
+                
+                if success:
+                    # Render updated test cases
+                    test_cases_html = dataset_controller.render_test_cases_display(updated_dataset)
+                    return (
+                        test_cases_html,  # test_cases_display
+                        updated_dataset,  # editing_dataset_state
+                        "",  # new_message_text (clear)
+                        "yellow",  # new_message_classification (reset)
+                        message  # status_message
+                    )
+                else:
+                    return (
+                        dataset_controller.render_test_cases_display(editing_dataset),  # test_cases_display
+                        editing_dataset,  # editing_dataset_state
+                        message_text,  # new_message_text (keep)
+                        classification,  # new_message_classification (keep)
+                        message  # status_message
+                    )
+            except Exception as e:
+                return (
+                    "",  # test_cases_display
+                    editing_dataset,  # editing_dataset_state
+                    message_text,  # new_message_text (keep)
+                    classification,  # new_message_classification (keep)
+                    f"❌ Error adding test case: {str(e)}"  # status_message
+                )
+        
+        def on_clear_form():
+            """Clear the add test case form."""
+            return (
+                "",  # new_message_text
+                "yellow",  # new_message_classification
+                "🗑️ Form cleared"  # status_message
+            )
+        
+        def on_save_dataset(editing_dataset):
+            """Save the current dataset."""
+            try:
+                if not editing_dataset:
+                    return "❌ No dataset being edited"
+                
+                success, message = dataset_controller.save_dataset(editing_dataset)
+                return message
+            except Exception as e:
+                return f"❌ Error saving dataset: {str(e)}"
+        
+        # Bind event handlers
+        dataset_selector.change(
+            on_dataset_selection_change,
+            inputs=[dataset_selector],
+            outputs=[dataset_info_display, current_dataset_state]
+        )
+        
+        edit_dataset_btn.click(
+            on_edit_dataset,
+            inputs=[current_dataset_state],
+            outputs=[edit_section, test_cases_display, editing_dataset_state, status_message]
+        )
+        
+        create_dataset_btn.click(
+            on_show_create,
+            outputs=[create_section, status_message]
+        )
+        
+        create_cancel_btn.click(
+            on_cancel_create,
+            outputs=[create_section, new_dataset_name, new_dataset_description, template_selector, status_message]
+        )
+        
+        create_confirm_btn.click(
+            on_create_dataset,
+            inputs=[new_dataset_name, new_dataset_description, template_selector],
+            outputs=[create_section, dataset_selector, new_dataset_name, new_dataset_description, template_selector, current_dataset_state, status_message]
+        )
+        
+        close_edit_btn.click(
+            on_close_edit,
+            outputs=[edit_section, new_message_text, new_message_classification, status_message]
+        )
+        
+        add_test_case_btn.click(
+            on_add_test_case,
+            inputs=[editing_dataset_state, new_message_text, new_message_classification],
+            outputs=[test_cases_display, editing_dataset_state, new_message_text, new_message_classification, status_message]
+        )
+        
+        clear_form_btn.click(
+            on_clear_form,
+            outputs=[new_message_text, new_message_classification, status_message]
+        )
+        
+        save_dataset_btn.click(
+            on_save_dataset,
+            inputs=[editing_dataset_state],
+            outputs=[status_message]
+        )
+    
+    return interface

src/interface/enhanced_verification_interface.py

@@ -120,18 +120,20 @@ class EnhancedVerificationInterface:
                         gr.Markdown("## 🎯 Select Verification Mode")
                         
                         with gr.Row():
-                            # Enhanced Dataset Mode
+                            # Standard Verification Mode (replaced Enhanced Dataset)
                             with gr.Column(scale=1):
-                                mode_info = EnhancedVerificationUIComponents.MODE_OPTIONS["enhanced_dataset"]
-                                gr.Markdown(f"### {mode_info['icon']} {mode_info['title']}")
-                                gr.Markdown(mode_info["description"])
+                                gr.Markdown("### ✓ Standard Verification")
+                                gr.Markdown("Use predefined test datasets to verify classifier accuracy. Quick and straightforward verification workflow.")
                                 
                                 gr.Markdown("**Features:**")
-                                for feature in mode_info["features"]:
-                                    gr.Markdown(f"• {feature}")
+                                gr.Markdown("• Pre-classified test datasets")
+                                gr.Markdown("• Step-by-step message review")
+                                gr.Markdown("• Accuracy tracking and statistics")
+                                gr.Markdown("• Export verification results")
+                                gr.Markdown("• Session progress preservation")
                                 
-                                enhanced_dataset_btn = gr.Button(
-                                    f"{mode_info['icon']} Start Enhanced Dataset Mode",
+                                standard_verification_btn = gr.Button(
+                                    "✓ Start Standard Verification",
                                     variant="primary",
                                     size="lg"
                                 )
@@ -195,15 +197,16 @@ class EnhancedVerificationInterface:
                             )
                 
                 # Individual mode interfaces (initially hidden)
-                enhanced_dataset_interface = gr.Row(visible=False)
-                with enhanced_dataset_interface:
+                # Standard Verification interface (replaced Enhanced Dataset)
+                standard_verification_interface = gr.Row(visible=False)
+                with standard_verification_interface:
                     with gr.Column():
-                        gr.Markdown("# 📊 Enhanced Dataset Mode")
-                        gr.Markdown("Select and customize test datasets for verification.")
-                        back_from_dataset_btn = gr.Button("← Back to Mode Selection", size="sm")
+                        gr.Markdown("# ✓ Standard Verification Mode")
+                        gr.Markdown("Review classified messages and provide feedback to improve the spiritual distress classifier.")
+                        back_from_standard_btn = gr.Button("← Back to Mode Selection", size="sm")
                         gr.Markdown("---")
                         
-                        # Create dataset interface components directly here
+                        # Create dataset interface components
                         from src.interface.enhanced_dataset_interface import EnhancedDatasetInterfaceController
                         dataset_controller = EnhancedDatasetInterfaceController()
                         
@@ -216,19 +219,17 @@ class EnhancedVerificationInterface:
                         # Dataset selection interface
                         with gr.Row():
                             with gr.Column(scale=2):
-                                gr.Markdown("## 📋 Select Dataset")
+                                gr.Markdown("## 📊 Select Dataset")
                                 
                                 dataset_selector = gr.Dropdown(
                                     choices=initial_choices,
                                     value=initial_choices[0] if initial_choices else None,
                                     label="Available Datasets",
-                                    info="Choose a dataset to verify or edit",
+                                    info="Choose a dataset to verify",
                                     interactive=True
                                 )
                                 
-                                with gr.Row():
-                                    load_dataset_btn = gr.Button("📥 Load Dataset", variant="primary", scale=2)
-                                    edit_dataset_btn = gr.Button("✏️ Edit Dataset", variant="secondary", scale=1)
+                                load_dataset_btn = gr.Button("📥 Load Dataset", variant="primary")
                             
                             with gr.Column(scale=1):
                                 gr.Markdown("## 📊 Dataset Information")
@@ -265,45 +266,12 @@ class EnhancedVerificationInterface:
                                     label="Progress"
                                 )
                         
-                        # Edit Dataset interface (initially hidden)
-                        edit_section = gr.Row(visible=False)
-                        with edit_section:
-                            with gr.Column():
-                                gr.Markdown("## ✏️ Edit Dataset")
-                                
-                                # Test cases display
-                                test_cases_display = gr.HTML(
-                                    value="<p>Loading test cases...</p>",
-                                    label="Test Cases"
-                                )
-                                
-                                gr.Markdown("### ➕ Add New Test Case")
-                                with gr.Row():
-                                    with gr.Column(scale=3):
-                                        new_message_text = gr.Textbox(
-                                            label="Message Text",
-                                            placeholder="Enter the test message...",
-                                            lines=3,
-                                            interactive=True
-                                        )
-                                    with gr.Column(scale=1):
-                                        new_message_classification = gr.Radio(
-                                            choices=["green", "yellow", "red"],
-                                            label="Expected Classification",
-                                            value="yellow"
-                                        )
-                                
-                                with gr.Row():
-                                    add_test_case_btn = gr.Button("➕ Add Test Case", variant="primary")
-                                    close_edit_btn = gr.Button("✖️ Close Editor", variant="secondary")
-                        
                         # Status message for dataset interface
                         dataset_status_message = gr.Markdown("", visible=True)
                         
                         # Dataset state
                         current_dataset_state = gr.State(value=None)
                         verification_session_state = gr.State(value=None)
-                        editing_dataset_state = gr.State(value=None)
                 
                 manual_input_interface = gr.Row(visible=False)
                 with manual_input_interface:
@@ -418,20 +386,25 @@ class EnhancedVerificationInterface:
                     
                     # Update interface visibility
                     mode_selection_visible = mode_type is None
-                    enhanced_dataset_visible = mode_type == "enhanced_dataset"
+                    standard_verification_visible = mode_type == "standard_verification"
                     manual_input_visible = mode_type == "manual_input"
                     file_upload_visible = mode_type == "file_upload"
                     
                     # Create status message
                     if mode_type:
-                        mode_title = EnhancedVerificationUIComponents.MODE_OPTIONS.get(mode_type, {}).get('title', 'Unknown')
+                        mode_titles = {
+                            "standard_verification": "Standard Verification",
+                            "manual_input": "Manual Input",
+                            "file_upload": "File Upload"
+                        }
+                        mode_title = mode_titles.get(mode_type, 'Unknown')
                         status_msg = f"✅ Switched to {mode_title} mode"
                     else:
                         status_msg = "✅ Returned to mode selection"
                     
                     return (
                         gr.Row(visible=mode_selection_visible),  # Mode selection section
-                        gr.Row(visible=enhanced_dataset_visible),  # Enhanced dataset interface
+                        gr.Row(visible=standard_verification_visible),  # Standard verification interface
                         gr.Row(visible=manual_input_visible),  # Manual input interface
                         gr.Row(visible=file_upload_visible),  # File upload interface
                         gr.Row(visible=False),  # Hide confirmation dialog
@@ -445,7 +418,7 @@ class EnhancedVerificationInterface:
                 except Exception as e:
                     return (
                         gr.Row(visible=True),  # Show mode selection on error
-                        gr.Row(visible=False),  # Hide enhanced dataset
+                        gr.Row(visible=False),  # Hide standard verification
                         gr.Row(visible=False),  # Hide manual input
                         gr.Row(visible=False),  # Hide file upload
                         gr.Row(visible=False),  # Hide confirmation dialog
@@ -523,25 +496,30 @@ class EnhancedVerificationInterface:
             def direct_mode_switch(mode_type: str):
                 """Directly switch to a mode without confirmation."""
                 mode_selection_visible = False
-                enhanced_dataset_visible = mode_type == "enhanced_dataset"
+                standard_verification_visible = mode_type == "standard_verification"
                 manual_input_visible = mode_type == "manual_input"
                 file_upload_visible = mode_type == "file_upload"
                 
-                mode_title = EnhancedVerificationUIComponents.MODE_OPTIONS.get(mode_type, {}).get('title', 'Unknown')
+                mode_titles = {
+                    "standard_verification": "Standard Verification",
+                    "manual_input": "Manual Input",
+                    "file_upload": "File Upload"
+                }
+                mode_title = mode_titles.get(mode_type, 'Unknown')
                 status_msg = f"✅ Switched to {mode_title} mode"
                 
                 return (
                     gr.Row(visible=mode_selection_visible),  # mode_selection_section
-                    gr.Row(visible=enhanced_dataset_visible),  # enhanced_dataset_interface
+                    gr.Row(visible=standard_verification_visible),  # standard_verification_interface
                     gr.Row(visible=manual_input_visible),  # manual_input_interface
                     gr.Row(visible=file_upload_visible),  # file_upload_interface
                     mode_type,  # current_mode_state
                     status_msg  # status_message
                 )
             
-            # Special function for enhanced dataset mode initialization
-            def switch_to_enhanced_dataset():
-                """Switch to enhanced dataset mode with proper initialization."""
+            # Special function for standard verification mode initialization
+            def switch_to_standard_verification():
+                """Switch to standard verification mode with proper initialization."""
                 try:
                     # Get first dataset info for display
                     dataset_choices, _, _, _ = dataset_controller.initialize_interface()
@@ -554,33 +532,33 @@ class EnhancedVerificationInterface:
                     
                     return (
                         gr.Row(visible=False),  # mode_selection_section
-                        gr.Row(visible=True),   # enhanced_dataset_interface
+                        gr.Row(visible=True),   # standard_verification_interface
                         gr.Row(visible=False),  # manual_input_interface
                         gr.Row(visible=False),  # file_upload_interface
-                        "enhanced_dataset",     # current_mode_state
-                        "✅ Switched to Enhanced Datasets mode",  # status_message
+                        "standard_verification",  # current_mode_state
+                        "✅ Switched to Standard Verification mode",  # status_message
                         dataset_info,           # dataset_info_display
                         first_dataset           # current_dataset_state
                     )
                 except Exception as e:
                     return (
                         gr.Row(visible=False),  # mode_selection_section
-                        gr.Row(visible=True),   # enhanced_dataset_interface
+                        gr.Row(visible=True),   # standard_verification_interface
                         gr.Row(visible=False),  # manual_input_interface
                         gr.Row(visible=False),  # file_upload_interface
-                        "enhanced_dataset",     # current_mode_state
-                        f"❌ Error initializing dataset mode: {str(e)}",  # status_message
+                        "standard_verification",  # current_mode_state
+                        f"❌ Error initializing verification mode: {str(e)}",  # status_message
                         "Error loading datasets",  # dataset_info_display
                         None                    # current_dataset_state
                     )
             
             # Bind mode selection buttons
-            enhanced_dataset_btn.click(
-                switch_to_enhanced_dataset,
+            standard_verification_btn.click(
+                switch_to_standard_verification,
                 inputs=[],
                 outputs=[
                     mode_selection_section,
-                    enhanced_dataset_interface,
+                    standard_verification_interface,
                     manual_input_interface,
                     file_upload_interface,
                     current_mode_state,
@@ -595,7 +573,7 @@ class EnhancedVerificationInterface:
                 inputs=[],
                 outputs=[
                     mode_selection_section,
-                    enhanced_dataset_interface,
+                    standard_verification_interface,
                     manual_input_interface,
                     file_upload_interface,
                     current_mode_state,
@@ -608,7 +586,7 @@ class EnhancedVerificationInterface:
                 inputs=[],
                 outputs=[
                     mode_selection_section,
-                    enhanced_dataset_interface,
+                    standard_verification_interface,
                     manual_input_interface,
                     file_upload_interface,
                     current_mode_state,
@@ -622,7 +600,7 @@ class EnhancedVerificationInterface:
                 inputs=[pending_mode_switch_state, current_session_state],
                 outputs=[
                     mode_selection_section,
-                    enhanced_dataset_interface,
+                    standard_verification_interface,
                     manual_input_interface,
                     file_upload_interface,
                     mode_switch_dialog,
@@ -639,7 +617,7 @@ class EnhancedVerificationInterface:
                 inputs=[pending_mode_switch_state, current_session_state],
                 outputs=[
                     mode_selection_section,
-                    enhanced_dataset_interface,
+                    standard_verification_interface,
                     manual_input_interface,
                     file_upload_interface,
                     mode_switch_dialog,
@@ -671,7 +649,7 @@ class EnhancedVerificationInterface:
                 """Return to mode selection screen."""
                 return (
                     gr.Row(visible=True),   # mode_selection_section
-                    gr.Row(visible=False),  # enhanced_dataset_interface
+                    gr.Row(visible=False),  # standard_verification_interface
                     gr.Row(visible=False),  # manual_input_interface
                     gr.Row(visible=False),  # file_upload_interface
                     None,                   # current_mode_state
@@ -679,11 +657,11 @@ class EnhancedVerificationInterface:
                 )
             
             # Bind Back buttons for each mode
-            back_from_dataset_btn.click(
+            back_from_standard_btn.click(
                 go_back_to_mode_selection,
                 outputs=[
                     mode_selection_section,
-                    enhanced_dataset_interface,
+                    standard_verification_interface,
                     manual_input_interface,
                     file_upload_interface,
                     current_mode_state,
@@ -695,7 +673,7 @@ class EnhancedVerificationInterface:
                 go_back_to_mode_selection,
                 outputs=[
                     mode_selection_section,
-                    enhanced_dataset_interface,
+                    standard_verification_interface,
                     manual_input_interface,
                     file_upload_interface,
                     current_mode_state,
@@ -707,7 +685,7 @@ class EnhancedVerificationInterface:
                 go_back_to_mode_selection,
                 outputs=[
                     mode_selection_section,
-                    enhanced_dataset_interface,
+                    standard_verification_interface,
                     manual_input_interface,
                     file_upload_interface,
                     current_mode_state,
@@ -771,84 +749,6 @@ class EnhancedVerificationInterface:
                         f"❌ Error loading dataset: {str(e)}"  # dataset_status_message
                     )
             
-            def on_edit_dataset(current_dataset):
-                """Handle edit dataset button."""
-                try:
-                    if not current_dataset:
-                        return (
-                            gr.Row(visible=False),  # edit_section
-                            "",  # test_cases_display
-                            current_dataset,  # editing_dataset_state
-                            "❌ No dataset selected for editing"  # dataset_status_message
-                        )
-                    
-                    # Render test cases HTML
-                    test_cases_html = dataset_controller.render_test_cases_display(current_dataset)
-                    
-                    return (
-                        gr.Row(visible=True),  # edit_section
-                        test_cases_html,  # test_cases_display
-                        current_dataset,  # editing_dataset_state
-                        f"✏️ Editing dataset: '{current_dataset.name}'"  # dataset_status_message
-                    )
-                except Exception as e:
-                    return (
-                        gr.Row(visible=False),  # edit_section
-                        "",  # test_cases_display
-                        None,  # editing_dataset_state
-                        f"❌ Error accessing dataset for editing: {str(e)}"  # dataset_status_message
-                    )
-            
-            def on_close_edit():
-                """Close the edit section."""
-                return (
-                    gr.Row(visible=False),  # edit_section
-                    "✅ Editor closed"  # dataset_status_message
-                )
-            
-            def on_add_test_case(editing_dataset, message_text, classification):
-                """Add a new test case to the dataset."""
-                try:
-                    if not editing_dataset:
-                        return (
-                            "",  # test_cases_display
-                            editing_dataset,  # editing_dataset_state
-                            "❌ No dataset being edited"  # dataset_status_message
-                        )
-                    
-                    if not message_text or not message_text.strip():
-                        return (
-                            dataset_controller.render_test_cases_display(editing_dataset),  # test_cases_display
-                            editing_dataset,  # editing_dataset_state
-                            "❌ Message text is required"  # dataset_status_message
-                        )
-                    
-                    # Add test case
-                    success, message, updated_dataset = dataset_controller.add_test_case(
-                        editing_dataset, message_text.strip(), classification
-                    )
-                    
-                    if success:
-                        # Render updated test cases
-                        test_cases_html = dataset_controller.render_test_cases_display(updated_dataset)
-                        return (
-                            test_cases_html,  # test_cases_display
-                            updated_dataset,  # editing_dataset_state
-                            message  # dataset_status_message
-                        )
-                    else:
-                        return (
-                            dataset_controller.render_test_cases_display(editing_dataset),  # test_cases_display
-                            editing_dataset,  # editing_dataset_state
-                            message  # dataset_status_message
-                        )
-                except Exception as e:
-                    return (
-                        "",  # test_cases_display
-                        editing_dataset,  # editing_dataset_state
-                        f"❌ Error adding test case: {str(e)}"  # dataset_status_message
-                    )
-            
             def on_start_verification(current_dataset, verifier_name):
                 """Handle starting verification session."""
                 try:
@@ -891,23 +791,6 @@ class EnhancedVerificationInterface:
                 outputs=[verification_section, dataset_status_message]
             )
             
-            edit_dataset_btn.click(
-                on_edit_dataset,
-                inputs=[current_dataset_state],
-                outputs=[edit_section, test_cases_display, editing_dataset_state, dataset_status_message]
-            )
-            
-            close_edit_btn.click(
-                on_close_edit,
-                outputs=[edit_section, dataset_status_message]
-            )
-            
-            add_test_case_btn.click(
-                on_add_test_case,
-                inputs=[editing_dataset_state, new_message_text, new_message_classification],
-                outputs=[test_cases_display, editing_dataset_state, dataset_status_message]
-            )
-            
             start_verification_btn.click(
                 on_start_verification,
                 inputs=[current_dataset_state, verifier_name_input],

src/interface/simplified_gradio_app.py

@@ -138,6 +138,11 @@ def create_simplified_interface():
                             """)
                     enhanced_verification_interface = create_enhanced_verification_tab()
             
+            # Edit Enhanced Datasets tab
+            with gr.TabItem("✏️ Edit Enhanced Datasets", id="edit_datasets"):
+                from src.interface.dataset_editor_interface import create_dataset_editor_interface
+                dataset_editor_interface = create_dataset_editor_interface()
+            
             # Standard Verification Mode tab (conditionally shown based on feature flag)
             if is_feature_enabled("standard_verification_enabled"):
                 with gr.TabItem("✓ Standard Verification", id="verification"):