Update app.py

app.py

@@ -1,10 +1,13 @@
 from fastapi import FastAPI, Request, Query
 from fastapi.templating import Jinja2Templates
 from fastapi import File, UploadFile
+from fastapi.responses import FileResponse
+
 from pydantic import BaseModel
 from sentence_transformers import SentenceTransformer
 import faiss
 import numpy as np
+import json
 
 app = FastAPI()
 model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
@@ -40,11 +43,11 @@ def embed_strings(request: EmbedRequest):
 def search_string(request: SearchRequest):
     embedding = model.encode([request.text])
     distances, indices = index.search(np.array(embedding), request.n)
-    #found_documents = index.reconstruct_n(indices[0], request.n)
+    found_documents = index.reconstruct_n(indices[0], request.n)
     return {
         "distances": distances[0].tolist(),
         "indices": indices[0].tolist(),
-        #"documents": found_documents.tolist()
+        "documents": found_documents.tolist()
     }
 
 #########################