fix: enhance citation display in response formatting

- Always show numbered citations with [1], [2] format
- Improve source section visibility and organization
- Add clear section headers and better metrics display
- Enhance medical disclaimer formatting

app.py

@@ -79,40 +79,43 @@ def format_enhanced_medical_response(response: EnhancedMedicalResponse) -> str:
     formatted_parts = []
     
     # Main response from the LLM
-    # The new prompt instructs the LLM to include markdown citations like [1], [2]
-    # The final response text is now the primary source of the answer.
-    final_response_text = response.answer
+    final_response_text = response.answer.strip()
     formatted_parts.append(final_response_text)
     
-    # Always add the clinical sources section if sources exist
+    # ALWAYS add the clinical sources section with clear numbering
     if response.sources:
         formatted_parts.append("\n\n---\n")
-        formatted_parts.append("### 📋 **Clinical Sources**")
+        formatted_parts.append("### 📋 **Clinical Sources & Citations**")
+        formatted_parts.append("\nThis response is based on the following Sri Lankan clinical guidelines:")
         # Create a numbered list of all sources used for the response
         for i, source in enumerate(response.sources, 1):
-            formatted_parts.append(f"{i}. {source}")
+            # Make the citation number bold and add a clear label
+            formatted_parts.append(f"\n**[{i}]** Source: {source}")
     
-    # Enhanced information section
+    # Enhanced information section with clear separation
     formatted_parts.append("\n\n---\n")
-    formatted_parts.append("### 📊 **Enhanced Medical Analysis**")
+    formatted_parts.append("### 📊 **Response Analysis**")
     
-    # Safety and verification info
+    # Safety and verification info with clearer formatting
     if response.verification_result:
-        safety_emoji = "🛡️" if response.safety_status == "SAFE" else "⚠️"
-        formatted_parts.append(f"**{safety_emoji} Medical Safety**: {response.safety_status}")
-        formatted_parts.append(f"**🔍 Verification Score**: {response.verification_result.verification_score:.1%}")
-        formatted_parts.append(f"**✅ Verified Claims**: {response.verification_result.verified_claims}/{response.verification_result.total_claims}")
+        safety_status = "✅ SAFE" if response.safety_status == "SAFE" else "⚠️ CAUTION"
+        formatted_parts.append(f"\n**Medical Safety Status**: {safety_status}")
+        formatted_parts.append(f"**Verification Score**: {response.verification_result.verification_score:.1%}")
+        formatted_parts.append(f"**Verified Medical Claims**: {response.verification_result.verified_claims}/{response.verification_result.total_claims}")
     
-    # Enhanced retrieval info
-    formatted_parts.append(f"**🧠 Medical Entities Extracted**: {response.medical_entities_count}")
-    formatted_parts.append(f"**🎯 Context Adherence**: {response.context_adherence_score:.1%}")
-    formatted_parts.append(f"**📚 Sources Used**: {len(response.sources)}")
-    if hasattr(response, 'query_time'): # Changed from processing_time to match the object attribute
-        formatted_parts.append(f"**⚡ Processing Time**: {response.query_time:.2f}s")
+    # Enhanced retrieval metrics
+    formatted_parts.append(f"\n**Medical Information Coverage**:")
+    formatted_parts.append(f"- 🧠 Medical Entities: {response.medical_entities_count}")
+    formatted_parts.append(f"- 🎯 Context Adherence: {response.context_adherence_score:.1%}")
+    formatted_parts.append(f"- 📚 Guidelines Referenced: {len(response.sources)}")
     
-    # Medical disclaimer
-    formatted_parts.append("\n---\n")
-    formatted_parts.append("*This information is for clinical reference based on Sri Lankan guidelines and does not replace professional medical judgment.*")
+    # Always include processing time if available
+    if hasattr(response, 'query_time'):
+        formatted_parts.append(f"- ⚡ Processing Time: {response.query_time:.2f}s")
+    
+    # Medical disclaimer with clear separation
+    formatted_parts.append("\n\n---\n")
+    formatted_parts.append("*⚕️ This information is derived from Sri Lankan clinical guidelines and is for reference only. Always consult with qualified healthcare professionals for patient care decisions.*")
     
     return "\n".join(formatted_parts)