feat: embed interface

.vscode/settings.json

@@ -1,6 +1,6 @@
 {
     "[python]": {
-        "editor.defaultFormatter": "ms-python.autopep8"
+        "editor.defaultFormatter": "ms-python.python"
     },
     "python.formatting.provider": "none"
 }

app.py

@@ -33,8 +33,20 @@ i_search = gr.Interface(
     outputs="json"
 )
 
-demo = gr.TabbedInterface([i_upsert, i_search], [
-                          "upsert role", "search roles"])
+
+def embed_text(text):
+    arr = vdb.embed_text(text)
+    json_data = json.dumps(arr.tolist())
+    return json_data
+
+i_embed = gr.Interface(
+    fn=embed_text,
+    inputs=["text"],
+    outputs="json"
+)
+
+demo = gr.TabbedInterface([i_upsert, i_search, i_embed], [
+                          "upsert role", "search roles", "embed text"])
 
 if __name__ == "__main__":
     demo.launch()

faiss_vdb.py

@@ -76,7 +76,7 @@ def clear_existing(id):
 
 
 def embed_text(txt):
-    emb = [db.embedding_function()]
+    emb = [db.embedding_function(txt)]
     vector = np.array(emb, dtype=np.float32)
     if db._normalize_L2:
         faiss = db.dependable_faiss_import()