Spaces:
Sleeping
Sleeping
File size: 2,933 Bytes
17f3ad3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
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
|