Update app.py

app.py

@@ -1,4 +1,6 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
+from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 from sentence_transformers import SentenceTransformer
 import faiss
 import numpy as np
@@ -7,12 +9,21 @@ app = FastAPI()
 model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
 index = faiss.IndexFlatL2(384)  # 384 is the dimensionality of the MiniLM model
 
-@app.get("/hello")
-def greet_json():
-    return {"Hello": "World!"}
+app.mount("/static", StaticFiles(directory="static"), name="static")
+templates = Jinja2Templates(directory="templates")
+
+@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"}
+
+@app.post("/search")
+def search_string(text: str, n: int = 5):
+    embedding = model.encode([text])
+    distances, indices = index.search(np.array(embedding), n)
+    return {"distances": distances[0].tolist(), "indices": indices[0].tolist()}