Update tools/quran_search.py

tools/quran_search.py

@@ -92,7 +92,7 @@ class QuranSearchEngine:
         
         try:
             self.model = SentenceTransformer(MODEL_NAME)
-            verse_texts = [v['text'] for v in self.all_verses]
+            verse_texts = ["passage: " + v['text'] for v in self.all_verses]  # Add prefix for e5 model
             self.verse_embeddings = []
             for i in range(0, len(verse_texts), CHUNK_SIZE):
                 chunk = verse_texts[i:i + CHUNK_SIZE]
@@ -144,7 +144,10 @@ class QuranSearchEngine:
             return self._keyword_fallback_search(query, top_k)
         
         try:
-            query_embedding = self.model.encode([query], convert_to_tensor=False)
+            # Add context for single words or short queries
+            if len(query.split()) <= 1:
+                query = f"معنى كلمة {query}"  # Add context: "meaning of the word"
+            query_embedding = self.model.encode(["query: " + query], convert_to_tensor=False)
             similarities = cosine_similarity(query_embedding, self.verse_embeddings)[0]
             top_indices = np.argsort(similarities)[-top_k:][::-1]
             
@@ -152,10 +155,12 @@ class QuranSearchEngine:
             for idx in top_indices:
                 verse = self.all_verses[idx]
                 surah_name = next((s.get('surahNameArabicLong', s.get('name', '')) for s in self.surahs if s['id'] == verse['surah_id']), f"سورة {verse['surah_id']}")
+                if surah_name.startswith("سورة "):
+                    surah_name = surah_name[len("سورة "):]
                 results.append(
                     f"سورة {surah_name} - آية {verse['verse_num']}:\n{verse['text']}"
                 )
-            return "\n\n".join(results)
+            return "\n\n".join(results) or "لا توجد نتائج ذات صلة."
         except Exception as e:
             self.logger.error(f"Search failed: {e}")
             return "حدث خطأ أثناء البحث. جرب مرة أخرى."
@@ -166,6 +171,8 @@ class QuranSearchEngine:
         for verse in self.all_verses:
             if query_lower in verse['text'].lower():
                 surah_name = next((s.get('surahNameArabicLong', s.get('name', '')) for s in self.surahs if s['id'] == verse['surah_id']), f"سورة {verse['surah_id']}")
+                if surah_name.startswith("سورة "):
+                    surah_name = surah_name[len("سورة "):]
                 matches.append(f"سورة {surah_name} - آية {verse['verse_num']}:\n{verse['text']}")
         return "\n\n".join(matches[:top_k]) or "لا توجد نتائج مطابقة."