Update services/pdf_service.py

services/pdf_service.py

@@ -129,17 +129,26 @@ class PDFService:
         
         try:
             # Get query embedding using thread pool
-            loop = asyncio.get_running_loop()
-            query_embedding = await loop.run_in_executor(
-                None,
-                lambda: self.embedder.encode(
-                    [query],
-                    convert_to_tensor=True
-                ).cpu().detach().numpy()
-            )
+            #loop = asyncio.get_running_loop()
+            #query_embedding = await loop.run_in_executor(
+            #    None,
+            #    lambda: self.embedder.encode(
+            #        [query],
+            #        convert_to_tensor=True
+            #    ).cpu().detach().numpy()
+            #)
             
             # Search
-            distances, indices = self.index.search(query_embedding, top_k * 2)
+            #distances, indices = self.index.search(query_embedding, top_k * 2)
+
+            # Create query embedding
+            query_embedding = self.embedder.encode([query], convert_to_tensor=True)
+            query_embedding_np = query_embedding.cpu().detach().numpy()
+
+            # Search in FAISS index
+            distances, indices = self.faiss_index.search(query_embedding_np, top_k)
+            
+
             
             # Process results
             results = []