Fix: Final citation and performance enhancements

app.py

@@ -88,14 +88,9 @@ def format_enhanced_medical_response(response: EnhancedMedicalResponse) -> str:
     if response.sources:
         formatted_parts.append("\n\n---\n")
         formatted_parts.append("### 📋 **Clinical Sources**")
-        # Create a numbered list of sources for clarity
+        # Create a numbered list of all sources used for the response
         for i, source in enumerate(response.sources, 1):
-            # Ensure we don't list more sources than were used for citations
-            if f"[{i}]" in final_response_text:
-                 formatted_parts.append(f"{i}. {source}")
-            else:
-                # If the LLM didn't cite this source, we can choose to omit it or list it as an uncited reference
-                pass # For now, only show cited sources to keep the output clean.
+            formatted_parts.append(f"{i}. {source}")
     
     # Enhanced information section
     formatted_parts.append("\n\n---\n")

src/enhanced_groq_medical_rag.py

@@ -481,7 +481,7 @@ class EnhancedGroqMedicalRAG:
             "**Core Principles:**\n"
             "• Use only information from the provided Sri Lankan clinical guidelines\n"
             "• Write in a natural, professional tone that healthcare providers appreciate\n"
-            "• Include relevant citations [1], [2] naturally within your response\n"
+            "• **CRITICAL INSTRUCTION**: You MUST include markdown citations (e.g., [1], [2]) for every piece of medical information you provide. The citation numbers correspond to the `[#] Citation:` markers in the context.\n"
             "• Structure information logically but naturally - no rigid formatting required\n"
             "• Focus on practical, actionable medical information\n\n"