feat: app initialize

app.py

@@ -0,0 +1,499 @@
+import os
+import json
+import copy
+import gradio as gr
+from patientsim import PatientAgent
+
+
+# ---------------------------------------------------------------------------
+# Constants
+# ---------------------------------------------------------------------------
+CEFR_CHOICES = [
+    ("A — Beginner", "A"),
+    ("B — Intermediate", "B"),
+    ("C — Advanced", "C"),
+]
+PERSONALITY_CHOICES = [
+    ("Neutral", "plain"),
+    ("Talkative", "verbose"),
+    ("Distrustful", "distrust"),
+    ("Pleasing", "pleasing"),
+    ("Impatient", "impatient"),
+    ("Overanxious", "overanxious"),
+]
+RECALL_CHOICES = [
+    ("No History", "no_history"),
+    ("Low", "low"),
+    ("High", "high"),
+]
+CONFUSION_CHOICES = [
+    ("Normal", "normal"),
+    ("Moderate", "moderate"),
+    ("High", "high"),
+]
+BACKEND_MODELS = [
+    "gemini-2.5-flash",
+    "gemini-3-flash-preview",
+    "gemini-3.1-flash-lite-preview",
+]
+
+# ---------------------------------------------------------------------------
+# Patient data
+# ---------------------------------------------------------------------------
+_DATA_PATH = os.path.join(os.path.dirname(__file__), "demo", "data", "demo_data.json")
+with open(_DATA_PATH) as _f:
+    PATIENT_DATA: list[dict] = json.load(_f)
+PATIENT_DICT: dict[str, dict] = {p["hadm_id"]: p for p in PATIENT_DATA}
+
+
+def _patient_choices() -> list[tuple[str, str]]:
+    choices = []
+    for idx, p in enumerate(sorted(PATIENT_DATA, key=lambda x: x["hadm_id"])):
+        label = (
+            f"{idx + 1}. Age {p['age']} | {p['gender']} "
+            f"| {p.get('diagnosis', 'Unknown')}"
+        )
+        choices.append((label, p["hadm_id"]))
+    return choices
+
+
+# ---------------------------------------------------------------------------
+# HTML helpers
+# ---------------------------------------------------------------------------
+def _row(label: str, val) -> str:
+    val = str(val) if val not in (None, "") else "N/A"
+    return (
+        f"<tr>"
+        f"<td style='padding:3px 8px;white-space:nowrap;vertical-align:top'><b>{label}:</b></td>"
+        f"<td style='padding:3px 8px'>{val}</td>"
+        f"</tr>"
+    )
+
+
+def _section(title: str, rows_html: str) -> str:
+    return (
+        f"<div style='font-weight:bold;color:#2c3e50;border-bottom:1px solid #ccc;"
+        f"margin:10px 0 4px;padding-bottom:2px'>{title}</div>"
+        f"<table style='border-collapse:collapse;font-size:13px;margin-bottom:4px'>"
+        f"{rows_html}</table>"
+    )
+
+
+def build_profile_html(p: dict) -> str:
+    basic = (
+        "<h3 style='margin:0 0 6px'>Patient Profile</h3>"
+        f"<table style='border-collapse:collapse;font-size:14px;margin-bottom:4px'>"
+        f"{_row('Age', p.get('age'))}"
+        f"{_row('Gender', p.get('gender'))}"
+        f"{_row('Race', p.get('race'))}"
+        f"{_row('Arrival Transport', p.get('arrival_transport'))}"
+        f"</table>"
+    )
+    social = _section(
+        "Social History",
+        _row("Tobacco", p.get("tobacco"))
+        + _row("Alcohol", p.get("alcohol"))
+        + _row("Illicit Drug", p.get("illicit_drug"))
+        + _row("Exercise", p.get("exercise"))
+        + _row("Marital Status", p.get("marital_status"))
+        + _row("Children", p.get("children"))
+        + _row("Living Situation", p.get("living_situation"))
+        + _row("Occupation", p.get("occupation"))
+        + _row("Insurance", p.get("insurance")),
+    )
+    history = _section(
+        "Previous Medical History",
+        _row("Allergies", p.get("allergies"))
+        + _row("Family History", p.get("family_medical_history"))
+        + _row("Medical Devices", p.get("medical_device"))
+        + _row("Prior Medical History", p.get("medical_history")),
+    )
+    visit = _section(
+        "Current Visit",
+        _row("Present Illness (+)", p.get("present_illness_positive"))
+        + _row("Present Illness (−)", p.get("present_illness_negative"))
+        + _row("Chief Complaint", p.get("chiefcomplaint"))
+        + _row("Pain (0–10)", p.get("pain"))
+        + _row("Medications", p.get("medication"))
+        + _row("Disposition", p.get("disposition"))
+        + _row("Diagnosis", p.get("diagnosis")),
+    )
+    return f"<div style='font-family:Noto Sans KR,Noto Sans,Malgun Gothic,Apple SD Gothic Neo,sans-serif;font-size:14px;line-height:1.5'>{basic}{social}{history}{visit}</div>"
+
+
+# ---------------------------------------------------------------------------
+# Custom CSS
+# ---------------------------------------------------------------------------
+CUSTOM_CSS = """
+/* ── Global gothic (sans-serif) font ───────────────────────────── */
+@import url('https://fonts.googleapis.com/css2?family=Noto+Sans+KR:wght@300;400;500;600;700&family=Noto+Sans:wght@300;400;500;600;700&display=swap');
+
+*, *::before, *::after {
+    font-family: 'Noto Sans KR', 'Noto Sans', 'Malgun Gothic', 'Apple SD Gothic Neo',
+                 'Segoe UI', sans-serif !important;
+}
+.prose h1, .prose h2, .prose h3, .prose h4,
+.markdown-text h1, .markdown-text h2, .markdown-text h3, .markdown-text h4 {
+    font-family: 'Noto Sans KR', 'Noto Sans', 'Malgun Gothic', 'Apple SD Gothic Neo',
+                 'Segoe UI', sans-serif !important;
+}
+
+/* ── Compact radio pills ────────────────────────────────────────── */
+.compact-radio .wrap {
+    gap: 6px !important;
+}
+.compact-radio label {
+    padding: 6px 14px !important;
+    border-radius: 8px !important;
+    font-size: 13px !important;
+    min-width: unset !important;
+}
+
+/* ── Tooltip descriptions ───────────────────────────────────────── */
+.option-desc {
+    font-size: 12px;
+    color: var(--body-text-color-subdued);
+    margin-top: 2px;
+    line-height: 1.4;
+}
+"""
+
+# Tooltip maps for descriptions shown below each radio group
+PERSONALITY_TIPS = {
+    "plain": "No strong emotions or noticeable behavior.",
+    "verbose": "Speaks a lot, gives highly detailed responses.",
+    "distrust": "Questions the doctor's expertise and care.",
+    "pleasing": "Overly positive, tends to minimize problems.",
+    "impatient": "Easily irritated and lacks patience.",
+    "overanxious": "Expresses concern beyond what is typical.",
+}
+RECALL_TIPS = {
+    "no_history": "No significant past medical history.",
+    "low": "Often forgets even major medical events.",
+    "high": "Usually recalls medical events accurately.",
+}
+CONFUSION_TIPS = {
+    "normal": "Clear mental status.",
+    "moderate": "Somewhat confused and off-topic at times.",
+    "high": "Highly dazed and extremely confused.",
+}
+CEFR_TIPS = {
+    "A": "Can make simple sentences.",
+    "B": "Can have daily conversations.",
+    "C": "Can freely use advanced medical terms.",
+}
+
+
+def _make_tip_html(tips: dict, selected: str) -> str:
+    desc = tips.get(selected, "")
+    return f"<p class='option-desc'>{desc}</p>" if desc else ""
+
+
+# ---------------------------------------------------------------------------
+# Callbacks
+# ---------------------------------------------------------------------------
+def start_simulation(
+    hadm_id: str,
+    api_key_input: str,
+    model: str,
+    cefr: str,
+    personality: str,
+    recall: str,
+    confusion: str,
+):
+    # Keep-current helper: returns all outputs unchanged so button resets cleanly
+    def _no_change():
+        return (
+            gr.update(),             # patient_agent_state (keep)
+            gr.update(),             # profile_display (keep)
+            gr.update(),             # chatbot (keep)
+            gr.update(visible=True), # setup_section stays visible
+            gr.update(visible=False),# chat_section stays hidden
+        )
+
+    if not hadm_id:
+        gr.Warning("Please select a patient first.")
+        return _no_change()
+
+    # Resolve API key: user input → env var
+    api_key = (
+        api_key_input.strip()
+        or os.environ.get("GOOGLE_API_KEY", "")
+        or os.environ.get("GENAI_API_KEY", "")
+    )
+    use_vertex = bool(os.environ.get("GOOGLE_PROJECT_ID", ""))
+
+    if not api_key and not use_vertex:
+        gr.Warning(
+            "API key is required. Enter your Google AI API key above, "
+            "or set the GOOGLE_API_KEY environment variable."
+        )
+        return _no_change()
+
+    patient = copy.deepcopy(PATIENT_DICT[hadm_id])
+
+    try:
+        agent = PatientAgent(
+            model=model,
+            visit_type="emergency_department",
+            personality=personality,
+            recall_level=recall,
+            confusion_level=confusion,
+            lang_proficiency_level=cefr,
+            api_key=api_key if not use_vertex else None,
+            use_vertex=use_vertex,
+            temperature=0.7,
+            num_word_sample=10,
+            random_seed=42,
+            log_verbose=False,
+            **patient,
+        )
+    except Exception as e:
+        gr.Warning(f"Failed to initialize patient agent: {e}")
+        return _no_change()
+
+    profile_html = build_profile_html(patient)
+    return (
+        agent,
+        profile_html,
+        [],                        # empty chat history
+        gr.update(visible=False),  # hide setup section
+        gr.update(visible=True),   # show chat section
+    )
+
+
+def chat(message: str, history: list, agent):
+    if agent is None:
+        raise gr.Error("No simulation running. Please start a simulation first.")
+    if not message.strip():
+        return history, ""
+
+    response = agent(user_prompt=message, using_multi_turn=True, verbose=False)
+    history = history + [
+        {"role": "user", "content": message},
+        {"role": "assistant", "content": response},
+    ]
+    return history, ""
+
+
+def reset_chat(agent):
+    if agent is not None:
+        agent.reset_history(verbose=False)
+    return []
+
+
+def return_to_setup():
+    return (
+        None,                      # clear agent state
+        [],                        # clear chat
+        "",                        # clear profile html
+        gr.update(visible=True),   # show setup section
+        gr.update(visible=False),  # hide chat section
+    )
+
+
+# ---------------------------------------------------------------------------
+# UI
+# ---------------------------------------------------------------------------
+with gr.Blocks(title="PatientSim", theme=gr.themes.Soft(), css=CUSTOM_CSS) as demo:
+    patient_agent_state = gr.State(None)
+
+    # ── Header ──────────────────────────────────────────────────────────────
+    gr.Markdown(
+        "# 🏥 PatientSim — ED Consultation Demo\n"
+        "An interactive simulator for realistic doctor–patient interactions "
+        "([NeurIPS 2025](https://openreview.net/forum?id=1THAjdP4QJ)). "
+        "Configure a patient and persona below, then converse as the doctor."
+    )
+
+    # ── Setup section ────────────────────────────────────────────────────────
+    with gr.Column(visible=True) as setup_section:
+
+        # ── Row 1: Patient & Model & API Key ─────────────────────────────
+        with gr.Group():
+            gr.Markdown("#### 🔑 Connection")
+            with gr.Row():
+                api_key_input = gr.Textbox(
+                    label="Google AI API Key",
+                    placeholder="Enter your API key (or set GOOGLE_API_KEY env var)",
+                    type="password",
+                    scale=2,
+                )
+                model_dd = gr.Dropdown(
+                    choices=BACKEND_MODELS,
+                    value=BACKEND_MODELS[0],
+                    label="Backend Model",
+                    scale=1,
+                )
+
+        # ── Row 2: Patient selector ──────────────────────────────────────
+        with gr.Group():
+            gr.Markdown("#### 🩺 Patient Case")
+            patient_dd = gr.Dropdown(
+                choices=_patient_choices(),
+                label="Select a patient case",
+            )
+
+        # ── Row 3: Persona settings (2 × 2 grid) ────────────────────────
+        with gr.Group():
+            gr.Markdown("#### 🎭 Patient Persona")
+            with gr.Row():
+                with gr.Column(min_width=200):
+                    personality_radio = gr.Radio(
+                        choices=PERSONALITY_CHOICES,
+                        value="plain",
+                        label="Personality",
+                        elem_classes=["compact-radio"],
+                    )
+                    personality_tip = gr.HTML(
+                        value=_make_tip_html(PERSONALITY_TIPS, "plain")
+                    )
+                with gr.Column(min_width=200):
+                    cefr_radio = gr.Radio(
+                        choices=CEFR_CHOICES,
+                        value="C",
+                        label="Language Proficiency (CEFR)",
+                        elem_classes=["compact-radio"],
+                    )
+                    cefr_tip = gr.HTML(
+                        value=_make_tip_html(CEFR_TIPS, "C")
+                    )
+            with gr.Row():
+                with gr.Column(min_width=200):
+                    recall_radio = gr.Radio(
+                        choices=RECALL_CHOICES,
+                        value="no_history",
+                        label="Medical History Recall",
+                        elem_classes=["compact-radio"],
+                    )
+                    recall_tip = gr.HTML(
+                        value=_make_tip_html(RECALL_TIPS, "no_history")
+                    )
+                with gr.Column(min_width=200):
+                    confusion_radio = gr.Radio(
+                        choices=CONFUSION_CHOICES,
+                        value="normal",
+                        label="Cognitive Confusion",
+                        elem_classes=["compact-radio"],
+                    )
+                    confusion_tip = gr.HTML(
+                        value=_make_tip_html(CONFUSION_TIPS, "normal")
+                    )
+
+        start_btn = gr.Button("▶  Start Simulation", variant="primary", size="lg")
+
+    # ── Chat section ─────────────────────────────────────────────────────────
+    with gr.Column(visible=False) as chat_section:
+        with gr.Row():
+            return_btn = gr.Button("← Return to Patient Selection", scale=1)
+            reset_btn = gr.Button("↺  Reset Conversation", scale=1)
+
+        with gr.Row(equal_height=False):
+            with gr.Column(scale=1, min_width=280):
+                profile_display = gr.HTML()
+
+            with gr.Column(scale=2):
+                chatbot = gr.Chatbot(
+                    label="Consultation",
+                    height=560,
+                    show_label=True,
+                    avatar_images=(
+                        "https://cdn-icons-png.flaticon.com/512/4140/4140048.png",
+                        "https://cdn-icons-png.flaticon.com/512/3774/3774299.png",
+                    ),
+                    placeholder=(
+                        "The conversation will appear here. "
+                        "Type your message below to begin."
+                    ),
+                )
+                with gr.Row():
+                    msg_box = gr.Textbox(
+                        placeholder="Type your message as the doctor…",
+                        label="Doctor's Message",
+                        lines=1,
+                        scale=5,
+                        show_label=False,
+                        container=False,
+                    )
+                    send_btn = gr.Button(
+                        "Send", variant="primary", scale=1, min_width=80
+                    )
+
+    # ── Event wiring ─────────────────────────────────────────────────────────
+
+    # Tooltip updates when radio selection changes
+    personality_radio.change(
+        fn=lambda v: _make_tip_html(PERSONALITY_TIPS, v),
+        inputs=[personality_radio],
+        outputs=[personality_tip],
+    )
+    cefr_radio.change(
+        fn=lambda v: _make_tip_html(CEFR_TIPS, v),
+        inputs=[cefr_radio],
+        outputs=[cefr_tip],
+    )
+    recall_radio.change(
+        fn=lambda v: _make_tip_html(RECALL_TIPS, v),
+        inputs=[recall_radio],
+        outputs=[recall_tip],
+    )
+    confusion_radio.change(
+        fn=lambda v: _make_tip_html(CONFUSION_TIPS, v),
+        inputs=[confusion_radio],
+        outputs=[confusion_tip],
+    )
+
+    # Start simulation
+    _start_outputs = [
+        patient_agent_state,
+        profile_display,
+        chatbot,
+        setup_section,
+        chat_section,
+    ]
+    start_btn.click(
+        fn=start_simulation,
+        inputs=[
+            patient_dd,
+            api_key_input,
+            model_dd,
+            cefr_radio,
+            personality_radio,
+            recall_radio,
+            confusion_radio,
+        ],
+        outputs=_start_outputs,
+    )
+
+    # Chat
+    _chat_outputs = [chatbot, msg_box]
+    send_btn.click(
+        fn=chat,
+        inputs=[msg_box, chatbot, patient_agent_state],
+        outputs=_chat_outputs,
+    )
+    msg_box.submit(
+        fn=chat,
+        inputs=[msg_box, chatbot, patient_agent_state],
+        outputs=_chat_outputs,
+    )
+
+    # Reset & return
+    reset_btn.click(
+        fn=reset_chat,
+        inputs=[patient_agent_state],
+        outputs=[chatbot],
+    )
+
+    _return_outputs = [
+        patient_agent_state,
+        chatbot,
+        profile_display,
+        setup_section,
+        chat_section,
+    ]
+    return_btn.click(fn=return_to_setup, outputs=_return_outputs)
+
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+patientsim
+gradio
+jsonlines
+pyyaml
+numpy