Update rag_system.py

rag_system.py

@@ -67,13 +67,26 @@ class RAGSystem:
         )
         return self.prompt_template
 
+    GREETINGS = [
+        "hi", "hello", "hey", "good morning", "good afternoon", "good evening"
+    ]
+
     def ask_question(self, user_input):
-        retriever = self.vectorstore.as_retriever(search_kwargs={"k": 5})
-        docs = retriever.invoke(user_input)  # تستخدم invoke بدلاً من get_relevant_documents
+        # التعامل مع التحية
+        if user_input.lower() in [g.lower() for g in self.GREETINGS]:
+            return (
+                "Hi! Please ask a question related to any of the following Saudi cybersecurity documents: "
+                "ECC (Essential Cybersecurity Controls), SCYWF (Saudi Cybersecurity Workforce Framework), or OSMACC (Organizations’ Social Media Accounts Cybersecurity Controls)."
+            )
+
+        # استرجاع الوثائق ذات الصلة
+        retriever = self.vectorstore.as_retriever(search_kwargs={"k": 3})
+        docs = retriever.invoke(user_input)
 
         if not docs:
             return "The answer is not available in the ECC guide."
 
+        # بناء السياق والمصادر
         context = "\n\n".join([d.page_content for d in docs])
         raw_sources = [
             f"source={d.metadata.get('source','?')};page={d.metadata.get('page_label', d.metadata.get('page','?'))}"
@@ -81,7 +94,10 @@ class RAGSystem:
         ]
         sources = " | ".join(set(raw_sources))
 
-        answer_prompt = self.prompt_template.format(context=context, question=user_input, sources=sources)
+        # إعداد السؤال والإجابة
+        answer_prompt = self.prompt_template.format(
+            context=context, question=user_input, sources=sources
+        )
         answer = self.llm(answer_prompt)
 
-        return answer
+        return answer