Update app.py

app.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI, Request, Query
 from fastapi.templating import Jinja2Templates
+from pydantic import BaseModel
 from sentence_transformers import SentenceTransformer
 import faiss
 import numpy as np
@@ -10,15 +11,19 @@ index = faiss.IndexFlatL2(384)  # 384 is the dimensionality of the MiniLM model
 
 templates = Jinja2Templates(directory=".")
 
+class EmbedRequest(BaseModel):
+    texts: list[str]
+
+    
 @app.get("/")
 def read_root(request: Request):
     return templates.TemplateResponse("index.html", {"request": request})
 
 @app.post("/embed")
-def embed_string(text: str):
-    embedding = model.encode([text])
-    index.add(np.array(embedding))
-    return {"message": "String embedded and added to FAISS database"}
+def embed_strings(request: EmbedRequest):
+    embeddings = model.encode(request.texts)
+    index.add(np.array(embeddings))
+    return {"message": "Strings embedded and added to FAISS database"}
 
 @app.post("/search")
 def search_string(text: str, n: int = 5):