Create database_utils.py

database_utils.py

@@ -0,0 +1,28 @@
+import sqlite3
+
+def init_db():
+    conn = sqlite3.connect('embeddings.db')
+    c = conn.cursor()
+    c.execute('''CREATE TABLE IF NOT EXISTS embeddings
+                 (sentence TEXT, embedding BLOB)''')
+    conn.commit()
+    conn.close()
+    return "Database initialized successfully!"
+
+def save_embeddings_to_db(sentence, embedding):
+    conn = sqlite3.connect('embeddings.db')
+    c = conn.cursor()
+    embedding_blob = sqlite3.Binary(embedding.tobytes())
+    c.execute("INSERT INTO embeddings (sentence, embedding) VALUES (?, ?)", (sentence, embedding_blob))
+    conn.commit()
+    conn.close()
+
+def get_all_embeddings():
+    conn = sqlite3.connect('embeddings.db')
+    c = conn.cursor()
+    c.execute("SELECT sentence, embedding FROM embeddings")
+    data = c.fetchall()
+    conn.close()
+    embeddings = [np.frombuffer(row[1], dtype=np.float32).reshape(-1, 3) for row in data]
+    sentences = [row[0] for row in data]
+    return embeddings, sentences