Update app.py

app.py

@@ -14,6 +14,9 @@ templates = Jinja2Templates(directory=".")
 class EmbedRequest(BaseModel):
     texts: list[str]
 
+class SearchRequest(BaseModel):
+    text: str
+    n: int = 5
     
 @app.get("/")
 def read_root(request: Request):
@@ -26,7 +29,7 @@ def embed_strings(request: EmbedRequest):
     return {"message": "Strings embedded and added to FAISS database"}
 
 @app.post("/search")
-def search_string(text: str, n: int = 5):
-    embedding = model.encode([text])
-    distances, indices = index.search(np.array(embedding), n)
+def search_string(request: SearchRequest):
+    embedding = model.encode([request.text])
+    distances, indices = index.search(np.array(embedding), request.n)
     return {"distances": distances[0].tolist(), "indices": indices[0].tolist()}