src/interface/model_handlers.py

from typing import Optional
import gradio as gr
from src.interface.session_manager import SimplifiedSessionData

def apply_model_settings(spiritual_model: str, soft_spiritual_triage_model: str, triage_evaluate_model: str, medical_model: str, soft_triage_model: str, spiritual_care_message_model: str, session: SimplifiedSessionData):
    """Apply custom model settings."""
    if session is None:
        session = SimplifiedSessionData()
    
    # Store model settings in session
    if not hasattr(session, 'custom_models'):
        session.custom_models = {}
    
    session.custom_models = {
        'SpiritualDistressAnalyzer': spiritual_model,
        'SoftSpiritualTriage': soft_spiritual_triage_model,
        'TriageResponseEvaluator': triage_evaluate_model,
        'MedicalAssistant': medical_model,
        'SoftMedicalTriage': soft_triage_model,
        'SpiritualCareMessage': spiritual_care_message_model
    }
    
    status = f"""<div style="padding: 1em; background-color: #ecfdf5; border-left: 4px solid #10b981; border-radius: 4px;">
<h4 style="color: #059669; margin-top: 0;">✅ Model Settings Applied</h4>

<p><strong>🔍 Spiritual Monitor:</strong> <code>{spiritual_model}</code></p>
<p><strong>🟡 Soft Spiritual Triage:</strong> <code>{soft_spiritual_triage_model}</code></p>
<p><strong>📊 Triage Response Evaluator:</strong> <code>{triage_evaluate_model}</code></p>
<p><strong>🏥 Medical Assistant:</strong> <code>{medical_model}</code></p>
<p><strong>🩺 Soft Medical Triage:</strong> <code>{soft_triage_model}</code></p>
<p><strong>💬 Spiritual Care Message:</strong> <code>{spiritual_care_message_model}</code></p>

<p style="color: #d97706; margin-bottom: 0;">
⚠️ <strong>Note:</strong> Model changes apply to this session only.
</p>
</div>"""
    
    return status, session

def reset_model_settings(session: SimplifiedSessionData):
    """Reset models to defaults."""
    if session is None:
        session = SimplifiedSessionData()
    
    # Clear custom models
    if hasattr(session, 'custom_models'):
        session.custom_models = {}
    
    status = """<div style="padding: 1em; background-color: #eff6ff; border-left: 4px solid #3b82f6; border-radius: 4px;">
<h4 style="color: #2563eb; margin-top: 0;">🔄 Models Reset to Defaults</h4>

<p><strong>🔍 Spiritual Monitor:</strong> <code>gemini-2.5-flash</code></p>
<p><strong>🟡 Soft Spiritual Triage:</strong> <code>claude-sonnet-4-5-20250929</code></p>
<p><strong>📊 Triage Response Evaluator:</strong> <code>gemini-2.5-flash</code></p>
<p><strong>🏥 Medical Assistant:</strong> <code>claude-sonnet-4-5-20250929</code></p>
<p><strong>🩺 Soft Medical Triage:</strong> <code>claude-sonnet-4-5-20250929</code></p>
<p><strong>💬 Spiritual Care Message:</strong> <code>claude-sonnet-4-5-20250929</code></p>

<p style="margin-bottom: 0;">Default models are now active.</p>
</div>"""
    
    return status, session