Update app.py

app.py

@@ -111,7 +111,15 @@ class SearchEngine:
         """
         query_embedding = self.model.encode(query, convert_to_tensor=True)
         distances, indices = self.index.search(query_embedding.cpu().detach().numpy().reshape(1, -1), top_k)
-        return [(self.documents[i], distances[0][i]) for i in indices[0]]
+        
+        # Convert NumPy data types to native Python types
+        results = []
+        for i in indices[0]:
+            document = self.documents[i]
+            distance = float(distances[0][i])  # Convert numpy.float32 to float
+            results.append({"document": document, "distance": distance})
+        
+        return results
 
 
 # Conversational Model using GPT-2
@@ -228,8 +236,11 @@ async def search(request: QueryRequest):
     if not query:
         raise HTTPException(status_code=400, detail="No query provided")
 
-    results = search_engine.search(query)
-    return {"results": results}
+    try:
+        results = search_engine.search(query)
+        return {"results": results}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
 
 @app.post("/topics")