Update app.py

app.py

@@ -1,7 +1,18 @@
 from fastapi import FastAPI
+from sentence_transformers import SentenceTransformer
+import faiss
+import numpy as np
 
 app = FastAPI()
+model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
+index = faiss.IndexFlatL2(384)  # 384 is the dimensionality of the MiniLM model
 
 @app.get("/")
 def greet_json():
     return {"Hello": "World!"}
+
+@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"}