Refactor MEDITRON_PROMPT in app.py to enhance medical assessment and recommendations, ensuring evidence-based practices and clear communication. Update name and age extraction logic for improved accuracy in user responses.

app.py

@@ -31,13 +31,18 @@ Do NOT make specific prescriptions for prescription-only drugs.
 Respond empathetically and clearly. Always be professional and thorough."""
 
 MEDITRON_PROMPT = """<|im_start|>system
-You are a specialized medical assistant focusing ONLY on suggesting over-the-counter medicines and home remedies based on patient information.
-Based on the following patient information, provide ONLY:
-1. One specific over-the-counter medicine with proper adult dosing instructions
-2. One practical home remedy that might help
-3. Clear guidance on when to seek professional medical care
-Be concise, practical, and focus only on general symptom relief. Do not diagnose. Include a disclaimer that you are not a licensed medical professional.
-<|im_end|>
+You are a board-certified physician with extensive clinical experience. Your role is to provide evidence-based medical assessment and recommendations following standard medical practice.
+
+For each patient case:
+1. Analyze presented symptoms systematically using medical terminology
+2. Create a structured differential diagnosis with most likely conditions first
+3. Recommend appropriate next steps (testing, monitoring, or treatment)
+4. Provide specific medication recommendations with precise dosing regimens
+5. Include clear red flags that would necessitate urgent medical attention
+6. Base all recommendations on current clinical guidelines and evidence-based medicine
+7. Maintain professional, clear, and compassionate communication
+
+Follow standard clinical documentation format when appropriate and prioritize patient safety at all times. Remember to include appropriate medical disclaimers.
 <|im_start|>user
 Patient information: {patient_info}
 <|im_end|>
@@ -111,12 +116,21 @@ def extract_name_age(messages):
     name, age = None, None
     for msg in messages:
         if msg.type == "human":
-            age_match = re.search(r"(?:I am|I'm|age is|aged|My age is)\s*(\d{1,3})", msg.content, re.IGNORECASE)
+            # Try to extract age
+            age_match = re.search(r"(?:I am|I'm|age is|aged|My age is|im|i'm)\s*(\d{1,3})", msg.content, re.IGNORECASE)
             if age_match and not age:
                 age = age_match.group(1)
-            name_match = re.search(r"(?:my name is|I'm|I am)\s*([A-Za-z]+)", msg.content, re.IGNORECASE)
+            # Try to extract name (avoid matching 'I'm' as name if age is present)
+            name_match = re.search(r"my name is\s*([A-Za-z]+)", msg.content, re.IGNORECASE)
             if name_match and not name:
                 name = name_match.group(1)
+            # Fallback: if user says 'I'm <name> and <age>'
+            fallback_match = re.search(r"i['’`]?m\s*([A-Za-z]+)\s*(?:and|,)?\s*(\d{1,3})", msg.content, re.IGNORECASE)
+            if fallback_match:
+                if not name:
+                    name = fallback_match.group(1)
+                if not age:
+                    age = fallback_match.group(2)
     return name, age
 
 @spaces.GPU