ui/tab_policy.py

"""Policy definition tab UI components."""

import os
import sys

import gradio as gr

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from utils.helpers import load_policy_from_file, load_preset_policy


def build_policy_tab(base_dir: str) -> dict:
    """Build the policy definition tab UI."""
    with gr.Tab("📋 Policy Definition"):
        current_policy_state = gr.State(value="")
        
        # Existing Policy Accordion
        with gr.Accordion("📥 Load Existing Policy", open=False):
            with gr.Row():
                with gr.Column():
                    preset_dropdown = gr.Dropdown(
                        label="Select Preset",
                        choices=["Hate Speech Policy", "Violence Policy", "Toxicity Policy"],
                        value=None
                    )
                    load_preset_btn = gr.Button("Load Preset")
                
                with gr.Column():
                    gr.Markdown("Upload a markdown file:")
                    upload_file = gr.File(label="Upload Markdown File", file_types=[".md"])
        
        # Manual Edition Accordion
        with gr.Accordion("✏️ Manual Edition", open=True):
            manual_text = gr.Textbox(
                label="Policy Text",
                placeholder="Enter or edit policy markdown...",
                lines=20
            )
        
        policy_preview = gr.Markdown(value="*No policy loaded*")
        
        # Clear button
        clear_policy_btn = gr.Button("Clear Policy", variant="secondary")
        
        # Handlers
        def load_preset_handler(name):
            if not name:
                return "", "*No policy loaded*", ""
            policy_text, _ = load_preset_policy(name, base_dir)
            return policy_text, policy_text, policy_text
        
        load_preset_btn.click(
            load_preset_handler,
            inputs=preset_dropdown,
            outputs=[current_policy_state, manual_text, policy_preview],
        )
        
        def load_upload_handler(f):
            if f:
                policy_text, _ = load_policy_from_file(f.name)
                return policy_text, policy_text, policy_text
            return "", "", "*No policy loaded*"
        
        upload_file.change(
            load_upload_handler,
            inputs=upload_file,
            outputs=[current_policy_state, manual_text, policy_preview],
        )
        
        def update_preview(text):
            return text if text else "*No policy loaded*"
        
        manual_text.change(
            lambda t: (t, update_preview(t)),
            inputs=manual_text,
            outputs=[current_policy_state, policy_preview],
        )
        
        clear_policy_btn.click(
            lambda: ("", "", "*No policy loaded*"),
            outputs=[current_policy_state, manual_text, policy_preview],
        )
        
        # Sync UI components when state changes externally (e.g., from dataset load)
        def sync_policy_ui(policy_text):
            preview_text = policy_text if policy_text else "*No policy loaded*"
            return policy_text, preview_text
        
        current_policy_state.change(
            sync_policy_ui,
            inputs=current_policy_state,
            outputs=[manual_text, policy_preview],
        )

    return {
        "current_policy_state": current_policy_state,
    }