"""
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"""
"""
def _generate_error_status(self, error_message: str) -> str:
"""Generate error status display."""
return f"""
"""
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
}