Update llm_client.py

llm_client.py

@@ -259,9 +259,10 @@ class LLMClient:
 
         # Custom Prompt Template
         # Custom Prompt Template - Stricter Format
-        template = """You are CyberGuard - an AI specialized in Phishing Detection.
+        template = """<|im_start|>system
+        You are CyberGuard - an AI specialized in Phishing Detection.
         Task: Analyze the provided URL and HTML snippet to classify the website as 'PHISHING' or 'BENIGN'.
-        
+        Check specifically for BRAND IMPERSONATION (e.g. Facebook, Google, Banks).
         Classification Rules:
         - PHISHING: Typosquatting URLs (e.g., paypa1.com), hidden login forms, obfuscated javascript, mismatched branding vs URL.
         - BENIGN: Legitimate website, clean code, URL matches the content/brand.
@@ -271,7 +272,16 @@ class LLMClient:
         CLASSIFICATION: [PHISHING or BENIGN]
         CONFIDENCE SCORE: [0-100]%
         EXPLANATION: [Write 3-4 concise sentences explaining the main reason]
-        """
+<|im_end|>
+<|im_start|>user
+Context from knowledge base:
+{context}
+
+Input to analyze:
+{question}
+<|im_end|>
+<|im_start|>assistant
+"""
         
         PROMPT = PromptTemplate(
             template=template,