Update app.py

app.py

@@ -148,7 +148,29 @@ def respond(message, history):
     if history is None: history = []
     api_msgs = normalize_history(history)
     
-    sys_prompt = {"role": "system", "content": "You are an oncology research assistant. Format: Summary, Prediction, Factors, Missing Info."}
+    # sys_prompt = {"role": "system", "content": "You are an oncology research assistant. Format: Summary, Prediction, Factors, Missing Info."}
+    SYSTEM_PROMPT = """You are an oncology research assistant. Analyze the patient data and output strictly in this exact format:
+
+=== Summary ===
+The patient is a [Age] year-old [Sex] with Stage [Stage], [Cancer Type] at diagnosis.
+
+=== Prediction ===
+- 1-year cancer survival: [Estimated %]
+- 1-year cancer recurrence: [Estimated %]
+- 5-year cancer survival: [Estimated %]
+- 5-year cancer recurrence: [Estimated %]
+
+=== Factors affecting prediction ===
+- 1-year survival: [1-2 concise reasons based on data]
+- 1-year recurrence: [1-2 concise reasons based on data]
+- 5-year survival: [1-2 concise reasons based on data]
+- 5-year recurrence: [1-2 concise reasons based on data]
+
+=== Missing Information ===
+- [List any missing key clinical fields, or state 'None detected']
+"""
+    
+    sys_prompt = {"role": "system", "content": SYSTEM_PROMPT}
     current_user_msg = {"role": "user", "content": str(message)}
     
     response_text = ollama_chat([sys_prompt] + api_msgs + [current_user_msg])
@@ -172,7 +194,7 @@ CSS = """
 """
 
 with gr.Blocks(css=CSS, title="OncoRisk Public Demo") as demo:
-    gr.HTML('<div id="titlebar"><div class="title-text">OncoRisk Assistant (Remote GPU)</div></div>')
+    gr.HTML('<div id="titlebar"><div class="title-text">Oncology Chat (developing)</div></div>')
     full_data_state = gr.State({})
     
     with gr.Row():
@@ -216,7 +238,7 @@ with gr.Blocks(css=CSS, title="OncoRisk Public Demo") as demo:
         # RIGHT COLUMN: Chat & Analysis
         with gr.Column(scale=2):
             chatbot = gr.Chatbot(height=550)
-            msg_input = gr.Textbox(label="Clinical Input (JSON)", lines=6, placeholder="Data will appear here...")
+            msg_input = gr.Textbox(label="Input Box", lines=6, placeholder="Ask anything oncology related")
             submit_btn = gr.Button("Analyze Case", variant="primary")
             gr.ClearButton([msg_input, chatbot])