Upload app.py

app.py

@@ -121,16 +121,25 @@ def system_instruction():
         "4. Never recommend or rank treatments. If the patient asks for medical advice, kindly "
         "redirect them to discuss it with their care team, then continue with a follow-up question.\n"
         "5. Tone: empathetic, respectful, never alarming.\n"
-        "6. OPENING TURN: The patient's first message will ask what the best treatment options are "
-        "for their myeloma. Do NOT recommend, rank, or prescribe any specific treatment. Instead, "
-        "respond in 3-5 sentences that (a) acknowledge their question warmly, (b) explain in plain "
-        "language that the right choice depends on clinical factors only their care team can assess "
-        "plus the personal priorities in their profile, and (c) briefly note the general CATEGORIES "
-        "of approaches that relapsed/refractory myeloma patients and their teams commonly discuss "
-        "(for example: combination drug regimens, immunotherapies such as CAR-T or bispecific "
-        "antibodies, stem cell transplant, clinical trials, and supportive/comfort-focused care) "
-        "framed as things to discuss with their team, NOT as advice. Then ask your FIRST single "
-        "follow-up question drawn from their profile and the flagged tensions.\n"
+        "6. TREATMENT-OPTIONS QUESTION: When the patient asks what the best treatment options are for "
+        "their myeloma (e.g. 'Using my profile, what are the best treatment options for my myeloma?'), "
+        "your answer must be grounded STRICTLY in the 16 yes/no profile answers provided to you. Do "
+        "this:\n"
+        "   - Open with one warm sentence.\n"
+        "   - Then walk through how THEIR specific stated priorities map to the kinds of treatment "
+        "approaches commonly discussed for relapsed/refractory myeloma. Tie each point explicitly back "
+        "to their answers (for example: a preference for taking medications at home and the fewest pills "
+        "points toward asking the team about convenient outpatient or oral regimens; openness to clinical "
+        "trials means trials are worth raising; willingness to endure side effects and wanting an "
+        "aggressive approach points toward asking about more intensive options; prioritizing quality of "
+        "life points toward asking about regimens that protect daily functioning; living alone with "
+        "limited support points toward asking about logistics and monitoring).\n"
+        "   - Do NOT invent clinical details, drug names beyond broad well-known categories, dosing, or "
+        "anything not implied by their answers. Do NOT rank, prescribe, or state which option is "
+        "medically best. Frame everything as 'based on what you told us, here are options and questions "
+        "to raise with your care team.'\n"
+        "   - End by reminding them their care team must confirm what is medically appropriate, then ask "
+        "your first single follow-up question drawn from their profile and the flagged tensions.\n"
         "7. When you judge that you have enough to summarize their priorities (or after several "
         "exchanges), instead of asking another question, output a final summary. Begin that final "
         "message with the exact token <<SUMMARY>> on its own line, then 4-7 short bullet points "
@@ -197,18 +206,18 @@ def start_chat(api_key, *radio_values):
     flags = detect_tensions(answers)
     flag_text = "\n".join(f"- {f}" for f in flags) if flags else "- (No obvious contradictions; explore their highest-stakes priorities.)"
 
-    kickoff = (
+    # Seed the conversation with the profile as the first user turn so every later
+    # turn (including when the user pastes the sample question) can be grounded in it.
+    seed = (
         "Here is the patient's completed profile (16 yes/no answers):\n\n"
         f"{profile}\n\n"
         "FLAGGED TENSIONS / PRIORITIES TO EXPLORE:\n"
         f"{flag_text}\n\n"
-        "This is the start of the conversation. The patient now asks the question below. "
-        "Follow your OPENING TURN instructions: give the educational, non-prescriptive response "
-        "and then ask your first follow-up question.\n\n"
-        f'Patient: "{FIRST_QUESTION}"'
+        "Acknowledge in one short sentence that you have their profile. Do NOT ask a question yet "
+        "and do NOT list treatment options yet. Simply invite them to ask their question."
     )
 
-    convo = [{"role": "user", "content": kickoff}]
+    convo = [{"role": "user", "content": seed}]
     try:
         reply = gemini_reply(client, convo, 0)
     except Exception as e:
@@ -216,16 +225,12 @@ def start_chat(api_key, *radio_values):
         return (gr.update(), [], [], 0, gr.update(visible=True), gr.update(visible=False))
 
     convo.append({"role": "assistant", "content": reply})
-    # Show the fixed first question to the user as their own message, then the reply.
-    chat_display = [
-        {"role": "user", "content": FIRST_QUESTION},
-        {"role": "assistant", "content": reply},
-    ]
+    chat_display = [{"role": "assistant", "content": reply}]
     return (
         chat_display,          # chatbot
         chat_display,          # display_state
         convo,                 # convo_state
-        1,                     # followup count
+        0,                     # followup count (no follow-up asked yet)
         gr.update(visible=False),  # hide intake
         gr.update(visible=True),   # show chat
     )
@@ -378,6 +383,33 @@ button.secondary, .gradio-container button.secondary {
     color: #111111 !important;
 }
 
+/* ---- Sample question hint ---- */
+.sample-q {
+    background: #f7f7f7 !important;
+    border: 1px solid #e2e2e2 !important;
+    border-left: 3px solid #7a0c2e !important;
+    border-radius: 6px !important;
+    padding: 10px 14px !important;
+    margin: 12px 0 4px 0 !important;
+}
+.sample-q-label {
+    display: block;
+    font-size: 0.85rem;
+    font-weight: 600;
+    color: #5a5a5a !important;
+    margin-bottom: 4px;
+}
+.sample-q-text {
+    display: block;
+    font-size: 0.95rem;
+    color: #111111 !important;
+    background: #ffffff !important;
+    border: 1px solid #e2e2e2 !important;
+    border-radius: 4px !important;
+    padding: 6px 10px !important;
+    font-family: 'Inter', sans-serif !important;
+}
+
 footer {visibility: hidden;}
 """
 
@@ -393,9 +425,8 @@ with gr.Blocks(title="P3 Myeloma Follow-up Chatbot", css=CSS,
             "<p id='app-subtitle'>Follow-up Questions for Shared Decision Making</p>"
         )
 
-    with gr.Accordion("Gemini API Key (optional if set as Space secret GEMINI_API_KEY)", open=False):
-        api_key = gr.Textbox(label="GEMINI_API_KEY", type="password",
-                             placeholder="Paste your key here, or leave blank to use the Space secret")
+    # API key is read from the Space secret GEMINI_API_KEY (hidden, not user-facing).
+    api_key = gr.State("")
 
     convo_state = gr.State([])
     display_state = gr.State([])
@@ -414,8 +445,14 @@ with gr.Blocks(title="P3 Myeloma Follow-up Chatbot", css=CSS,
     # ---- Chat panel ----
     with gr.Group(visible=False) as chat_panel:
         chatbot = gr.Chatbot(label="Follow-up Conversation", type="messages", height=460)
+        gr.HTML(
+            "<div class='sample-q'>"
+            "<span class='sample-q-label'>Sample question &mdash; copy and paste it into the box below to begin:</span>"
+            f"<code class='sample-q-text'>{FIRST_QUESTION}</code>"
+            "</div>"
+        )
         with gr.Row():
-            msg = gr.Textbox(placeholder="Type your answer...", show_label=False, scale=8)
+            msg = gr.Textbox(placeholder="Type or paste your question here...", show_label=False, scale=8)
             send_btn = gr.Button("Send", variant="primary", scale=1)
         restart_btn = gr.Button("Start Over")