Fix: Agent chat now works properly - Update all remaining functions to use per-user agent storage

app.py

@@ -726,9 +726,10 @@ def build_ui():
             agent_name, user_text, histories,
             deescalation_culture, deescalation_meds,
             stewardship_site, stewardship_biofilm, stewardship_response, stewardship_crcl, stewardship_severity, stewardship_allergies,
-            empiric_age, empiric_allergies, empiric_labs, empiric_culture, empiric_meds, empiric_site, empiric_biofilm, empiric_response, empiric_crcl, empiric_severity
+            empiric_age, empiric_allergies, empiric_labs, empiric_culture, empiric_meds, empiric_site, empiric_biofilm, empiric_response, empiric_crcl, empiric_severity,
+            request: gr.Request
         ):
-            agent_json = agents_config.get(agent_name)
+            agent_json = get_user_agent(request, agent_name)
             if agent_json:
                 agent_data = json.loads(agent_json)
                 skills = agent_data.get("skills", [])
@@ -822,7 +823,7 @@ def build_ui():
             # Call chatpanel_handle, but get extra challenger info from simulate_agent_response_stream
             import asyncio
             from core.agents.chat_orchestrator import simulate_agent_response_stream
-            agent_json_val = agents_config.get(agent_name)
+            agent_json_val = get_user_agent(request, agent_name)
             history_val = histories.get(agent_name, [])
             result = None
             async def run_stream():
@@ -916,7 +917,7 @@ def build_ui():
             # Dropdown to pick which agent to chat with
             agent_picker = gr.Dropdown(
                 label="Select Agent",
-                choices=list(agents_config.keys()),
+                choices=[],  # Will be populated per-user by refresh_chat_dropdown
                 interactive=True
             )
 
@@ -1550,8 +1551,8 @@ def build_ui():
 
         # --- Show/hide stewardship fields based on agent selection ---
 
-        def update_dynamic_vars_visibility(agent_name):
-            agent_json = agents_config.get(agent_name)
+        def update_dynamic_vars_visibility(agent_name, request: gr.Request):
+            agent_json = get_user_agent(request, agent_name)
             if agent_json:
                 agent_data = json.loads(agent_json)
                 skills = agent_data.get("skills", [])
@@ -1761,9 +1762,10 @@ def build_ui():
             agent_name, user_text, histories,
             deescalation_culture, deescalation_meds,
             stewardship_site, stewardship_biofilm, stewardship_response, stewardship_crcl, stewardship_severity, stewardship_allergies,
-            empiric_age, empiric_allergies, empiric_labs, empiric_culture, empiric_meds, empiric_site, empiric_biofilm, empiric_response, empiric_crcl, empiric_severity
+            empiric_age, empiric_allergies, empiric_labs, empiric_culture, empiric_meds, empiric_site, empiric_biofilm, empiric_response, empiric_crcl, empiric_severity,
+            request: gr.Request
         ):
-            agent_json = agents_config.get(agent_name)
+            agent_json = get_user_agent(request, agent_name)
             # Prevent tool invocation on empty/whitespace input (e.g., initial greeting or agent selection)
             if user_text is None or not str(user_text).strip():
                 # Just return the current history, do not invoke any tool
@@ -1977,7 +1979,7 @@ def build_ui():
             # Use simulate_agent_response_stream for all agents to ensure challenger logic is applied
             import asyncio
             from core.agents.chat_orchestrator import simulate_agent_response_stream
-            agent_json_val = agents_config.get(agent_name)
+            agent_json_val = get_user_agent(request, agent_name)
             history_val = histories.get(agent_name, [])
             result = None
             # Prevent repeated tool invocation: if the last assistant message is a tool request for the same required fields, do not re-invoke