kkk

medical_ai.py

@@ -360,9 +360,18 @@ class CompetitionMedicalLLM:
             self.generator = None
 
     def _build_biogpt_prompt(self, question: str, contexts: Dict[str, List[str]], lang: str = "en") -> str:
-        # Simple prompt for BioGPT (no special tokens needed)
+        # Prompt for BioGPT: friendly, empathetic, Cameroonian context, precise, with safety disclaimer
         context_str = "\n".join([f"- {c}" for ctxs in contexts.values() for c in ctxs])
-        prompt = f"You are a helpful medical AI assistant. Contexts:\n{context_str}\nQuestion: {question}\nAnswer:"
+        prompt = (
+            "You are a helpful, friendly, and empathetic medical assistant. "
+            "Always answer in a clear, simple, and reassuring way, as if you are talking to a patient who may be worried. "
+            "Take into account the Cameroonian context, including local realities, resources, and common practices. "
+            "Be precise and concise in your answers. "
+            "At the end of each answer, always add: 'This is an AI-generated answer. Please always contact a doctor for more precision.'\n"
+            f"Contexts:\n{context_str}\n"
+            f"Question: {question}\n"
+            "Answer:"
+        )
         return prompt
 
     def generate_expert_response(self, question: str, contexts: Dict[str, List[str]], lang: str = "en") -> str: