""" Enhanced Edit Prompts UI Integration This module provides enhanced UI integration for the Edit Prompts interface, integrating with the centralized PromptController system while maintaining existing UI functionality and adding new features. **Feature: prompt-optimization, Task 11.4: Enhance Edit Prompts UI integration** **Validates: Requirements 9.1, 9.4** """ import gradio as gr from typing import Dict, List, Optional, Tuple, Any from datetime import datetime import sys import os # Add src to path for imports sys.path.append('src') from config.prompt_management.prompt_controller import PromptController from config.prompt_management.data_models import PromptConfig class EnhancedPromptEditor: """Enhanced prompt editor with centralized prompt system integration.""" def __init__(self): self.controller = PromptController() self._agent_mapping = { "🔍 Spiritual Monitor (Classifier)": "spiritual_monitor", "🟡 Soft Spiritual Triage": "triage_question", "📊 Triage Response Evaluator": "triage_evaluator", "🏥 Medical Assistant": "medical_assistant", "🩺 Soft Medical Triage": "soft_medical_triage", "💬 Spiritual Care Message": "spiritual_care_message" } self._reverse_mapping = {v: k for k, v in self._agent_mapping.items()} def get_available_prompts(self) -> List[str]: """Get list of available prompts for the dropdown.""" return list(self._agent_mapping.keys()) def load_prompt_for_editing(self, prompt_name: str, session_id: Optional[str] = None) -> Tuple[str, str, str]: """ Load a prompt for editing with enhanced information display. Args: prompt_name: Display name of the prompt session_id: Optional session ID for session-specific overrides Returns: Tuple of (prompt_content, info_html, status_html) """ try: agent_type = self._agent_mapping.get(prompt_name) if not agent_type: return "", self._generate_error_info("Unknown prompt type"), self._generate_error_status("Invalid prompt selection") # Get prompt configuration config = self.controller.get_prompt(agent_type, session_id=session_id) # Determine prompt source prompt_source = "Default Fallback" if config.session_override: prompt_source = f"Session Override ({session_id[:8]}...)" elif agent_type in ['spiritual_monitor', 'triage_question', 'triage_evaluator']: prompt_source = "Centralized File" # Generate enhanced info display info_html = self._generate_prompt_info( prompt_name=prompt_name, config=config, prompt_source=prompt_source, session_id=session_id ) # Generate status status_html = self._generate_load_status(prompt_name, prompt_source) return config.base_prompt, info_html, status_html except Exception as e: error_info = self._generate_error_info(f"Error loading prompt: {str(e)}") error_status = self._generate_error_status("Failed to load prompt") return "", error_info, error_status def apply_prompt_changes(self, prompt_name: str, prompt_content: str, session_id: str) -> Tuple[str, bool]: """ Apply prompt changes to the session. Args: prompt_name: Display name of the prompt prompt_content: New prompt content session_id: Session identifier Returns: Tuple of (status_html, success) """ try: if not prompt_content.strip(): return self._generate_error_status("Prompt content cannot be empty"), False agent_type = self._agent_mapping.get(prompt_name) if not agent_type: return self._generate_error_status("Invalid prompt type"), False # Set session override success = self.controller.set_session_override(agent_type, prompt_content, session_id) if success: status_html = self._generate_apply_success_status( prompt_name=prompt_name, content_length=len(prompt_content), session_id=session_id ) return status_html, True else: return self._generate_error_status("Failed to apply prompt changes"), False except Exception as e: return self._generate_error_status(f"Error applying changes: {str(e)}"), False def reset_prompt_to_default(self, prompt_name: str, session_id: str) -> Tuple[str, str, str]: """ Reset prompt to default (remove session override). Args: prompt_name: Display name of the prompt session_id: Session identifier Returns: Tuple of (prompt_content, info_html, status_html) """ try: agent_type = self._agent_mapping.get(prompt_name) if not agent_type: error_info = self._generate_error_info("Invalid prompt type") error_status = self._generate_error_status("Reset failed") return "", error_info, error_status # Clear session override for this agent if session_id in self.controller._session_overrides: if agent_type in self.controller._session_overrides[session_id]: del self.controller._session_overrides[session_id][agent_type] # Clear cache entry cache_key = f"{agent_type}_{session_id}" if cache_key in self.controller._prompt_cache: del self.controller._prompt_cache[cache_key] # Reload default prompt return self.load_prompt_for_editing(prompt_name, session_id) except Exception as e: error_info = self._generate_error_info(f"Error resetting prompt: {str(e)}") error_status = self._generate_error_status("Reset failed") return "", error_info, error_status def get_session_prompt_status(self, session_id: str) -> str: """ Get status of all session prompt overrides. Args: session_id: Session identifier Returns: HTML status display """ try: session_overrides = self.controller.get_session_overrides(session_id) if not session_overrides: return """

📋 Session Status

No active prompt overrides in this session.

""" override_list = [] for agent_type, content in session_overrides.items(): display_name = self._reverse_mapping.get(agent_type, agent_type) content_preview = content[:100] + "..." if len(content) > 100 else content override_list.append(f"
  • {display_name}: {len(content)} chars
  • ") return f"""

    ✅ Active Session Overrides

    """ except Exception as e: return f"""

    ❌ Error

    Failed to get session status: {str(e)}

    """ def promote_session_to_file(self, prompt_name: str, session_id: str) -> Tuple[str, bool]: """ Promote session override to permanent file. Args: prompt_name: Display name of the prompt session_id: Session identifier Returns: Tuple of (status_html, success) """ try: agent_type = self._agent_mapping.get(prompt_name) if not agent_type: return self._generate_error_status("Invalid prompt type"), False success = self.controller.promote_session_to_file(agent_type, session_id) if success: status_html = f"""

    ✅ Promoted to File

    Prompt: {prompt_name}

    Action: Session override promoted to permanent file

    ⚠️ Note: Original file backed up with timestamp.

    """ return status_html, True else: return self._generate_error_status("No session override to promote"), False except Exception as e: return self._generate_error_status(f"Error promoting to file: {str(e)}"), False def validate_prompt_syntax(self, prompt_content: str) -> Tuple[str, bool]: """ Validate prompt syntax and structure. Args: prompt_content: Prompt content to validate Returns: Tuple of (validation_html, is_valid) """ try: issues = [] warnings = [] # Basic validation checks if not prompt_content.strip(): issues.append("Prompt cannot be empty") if len(prompt_content) < 50: warnings.append("Prompt is very short (< 50 characters)") if len(prompt_content) > 10000: warnings.append("Prompt is very long (> 10,000 characters)") # Check for common structural elements if "" not in prompt_content: warnings.append("Missing section") if "" not in prompt_content: warnings.append("Missing section") # Check for placeholder usage placeholder_count = prompt_content.count("{{SHARED_") if placeholder_count > 0: warnings.append(f"Contains {placeholder_count} placeholder(s) - will be replaced with actual content") # Generate validation result if issues: validation_html = f"""

    ❌ Validation Errors

      {''.join(f'
    • {issue}
    • ' for issue in issues)}
    """ return validation_html, False elif warnings: validation_html = f"""

    ⚠️ Validation Warnings

      {''.join(f'
    • {warning}
    • ' for warning in warnings)}
    """ return validation_html, True else: validation_html = """

    ✅ Validation Passed

    Prompt structure looks good!

    """ return validation_html, True except Exception as e: error_html = f"""

    ❌ Validation Error

    Failed to validate: {str(e)}

    """ return error_html, False def _generate_prompt_info(self, prompt_name: str, config: PromptConfig, prompt_source: str, session_id: Optional[str]) -> str: """Generate enhanced prompt information display.""" # Calculate statistics content_length = len(config.base_prompt) line_count = len(config.base_prompt.split('\n')) word_count = len(config.base_prompt.split()) # Check for placeholders placeholder_count = config.base_prompt.count("{{SHARED_") # Generate shared components info components_info = f"""

    Shared Components:

    • Indicators: {len(config.shared_indicators)}
    • Rules: {len(config.shared_rules)}
    • Templates: {len(config.templates)}
    """ # Generate session info session_info = "" if session_id: session_info = f"""

    Session: {session_id[:12]}...

    """ # Generate source indicator source_color = "#059669" if "Session Override" in prompt_source else "#3b82f6" source_icon = "🔧" if "Session Override" in prompt_source else "📁" return f"""

    📋 Prompt Information

    Name: {prompt_name}

    Source: {source_icon} {prompt_source}

    {session_info}

    Statistics:

    • Length: {content_length:,} characters
    • Lines: {line_count:,}
    • Words: {word_count:,}
    • Placeholders: {placeholder_count}
    {components_info}

    Last Updated: {config.last_updated.strftime('%Y-%m-%d %H:%M:%S')}

    Version: {config.version}

    """ def _generate_load_status(self, prompt_name: str, prompt_source: str) -> str: """Generate load success status.""" return f"""

    ✅ Prompt Loaded

    Prompt: {prompt_name}

    Source: {prompt_source}

    Ready to edit. Make your changes and click "Apply Changes".

    """ def _generate_apply_success_status(self, prompt_name: str, content_length: int, session_id: str) -> str: """Generate apply success status.""" return f"""

    ✅ Prompt Applied Successfully

    Prompt: {prompt_name}

    Length: {content_length:,} characters

    Session: {session_id[:12]}...

    ⚠️ Note: Changes are active for this session only. Use "Promote to File" to make permanent.

    """ def _generate_error_info(self, error_message: str) -> str: """Generate error information display.""" return f"""

    ❌ Error

    {error_message}

    """ def _generate_error_status(self, error_message: str) -> str: """Generate error status display.""" return f"""

    ❌ Error

    {error_message}

    """ def create_enhanced_prompt_editor_ui() -> Tuple[Any, ...]: """ Create enhanced prompt editor UI components. Returns: Tuple of Gradio components for the enhanced prompt editor """ editor = EnhancedPromptEditor() with gr.TabItem("🔧 Edit Prompts", id="edit_prompts"): gr.Markdown("## 🔧 Enhanced Prompt Editor") gr.Markdown("⚠️ **Note:** Changes apply only to your current session. Use 'Promote to File' to make permanent.") # Session status display with gr.Row(): session_status_display = gr.HTML(value="", visible=True) # Prompt selector and controls with gr.Row(): with gr.Column(scale=2): prompt_selector = gr.Dropdown( choices=editor.get_available_prompts(), value=editor.get_available_prompts()[0] if editor.get_available_prompts() else None, label="Select Prompt to Edit", interactive=True ) with gr.Column(scale=1): load_prompt_btn = gr.Button("📥 Load Prompt", variant="secondary") validate_prompt_btn = gr.Button("🔍 Validate", variant="secondary") # Main editor area with gr.Row(): with gr.Column(scale=3): # Prompt editor prompt_editor = gr.Code( label="System Prompt", value="", language="markdown", lines=25, interactive=True ) # Validation display validation_display = gr.HTML(value="", visible=True) # Action buttons with gr.Row(): apply_prompt_btn = gr.Button("✅ Apply Changes", variant="primary", scale=2) reset_prompt_btn = gr.Button("🔄 Reset to Default", variant="secondary", scale=1) promote_prompt_btn = gr.Button("📤 Promote to File", variant="stop", scale=1) # Status display prompt_status = gr.HTML(value="", visible=True) with gr.Column(scale=1): # Enhanced info panel gr.Markdown("### 📋 Prompt Information") prompt_info_display = gr.HTML( value="""

    Select a prompt to edit

    Enhanced features:

    • 🔧 Session-level editing
    • 📊 Real-time validation
    • 🔄 Easy reset/revert
    • 📤 Promote to permanent
    • 📋 Detailed statistics
    """, visible=True ) return ( prompt_selector, prompt_editor, prompt_info_display, prompt_status, validation_display, session_status_display, load_prompt_btn, apply_prompt_btn, reset_prompt_btn, promote_prompt_btn, validate_prompt_btn ) # Helper function for integration with existing UI def integrate_with_existing_ui(session_data_component): """ Integration helper for existing Gradio UI. Args: session_data_component: Existing session data Gradio component """ editor = EnhancedPromptEditor() def enhanced_load_prompt(prompt_name: str, session_data): """Enhanced load prompt handler.""" session_id = getattr(session_data, 'session_id', 'default_session') if session_data else 'default_session' return editor.load_prompt_for_editing(prompt_name, session_id) def enhanced_apply_prompt(prompt_name: str, prompt_content: str, session_data): """Enhanced apply prompt handler.""" session_id = getattr(session_data, 'session_id', 'default_session') if session_data else 'default_session' status_html, success = editor.apply_prompt_changes(prompt_name, prompt_content, session_id) return status_html, session_data def enhanced_reset_prompt(prompt_name: str, session_data): """Enhanced reset prompt handler.""" session_id = getattr(session_data, 'session_id', 'default_session') if session_data else 'default_session' prompt_content, info_html, status_html = editor.reset_prompt_to_default(prompt_name, session_id) return prompt_content, info_html, status_html, session_data def enhanced_validate_prompt(prompt_content: str): """Enhanced validate prompt handler.""" return editor.validate_prompt_syntax(prompt_content) def enhanced_session_status(session_data): """Enhanced session status handler.""" session_id = getattr(session_data, 'session_id', 'default_session') if session_data else 'default_session' return editor.get_session_prompt_status(session_id) def enhanced_promote_prompt(prompt_name: str, session_data): """Enhanced promote prompt handler.""" session_id = getattr(session_data, 'session_id', 'default_session') if session_data else 'default_session' status_html, success = editor.promote_session_to_file(prompt_name, session_id) return status_html, session_data return { 'load_prompt': enhanced_load_prompt, 'apply_prompt': enhanced_apply_prompt, 'reset_prompt': enhanced_reset_prompt, 'validate_prompt': enhanced_validate_prompt, 'session_status': enhanced_session_status, 'promote_prompt': enhanced_promote_prompt }