Update app.py

app.py

@@ -58,13 +58,15 @@ def clean_text_response(text: str) -> str:
 
 def extract_section(text: str, heading: str) -> str:
     try:
-        pattern = rf"{heading}:\n(.*?)(?=\n\w|\Z)"
-        match = re.search(pattern, text, re.DOTALL)
-        return clean_text_response(match.group(1)) if match else ""
+        # Accept formats like "Heading:" or "**Heading:**"
+        pattern = rf"(\*\*)?{re.escape(heading)}(\*\*)?:\s*\n(.*?)(?=\n\s*\*?\*?\w|$)"
+        match = re.search(pattern, text, re.DOTALL | re.IGNORECASE)
+        return clean_text_response(match.group(3)) if match else ""
     except Exception as e:
-        logger.error(f"Section extraction failed: {e}")
+        logger.error(f"Section extraction failed for heading '{heading}': {e}")
         return ""
 
+
 def structure_medical_response(text: str) -> Dict:
     return {
         "summary": extract_section(text, "Summary"),