Update database_utils.py

database_utils.py

@@ -1,4 +1,6 @@
 import sqlite3
+import pandas as pd
+
 
 def init_db():
     conn = sqlite3.connect('embeddings.db')
@@ -26,3 +28,11 @@ def get_all_embeddings():
     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
+
+
+def fetch_data_as_csv():
+    conn = sqlite3.connect('embeddings.db')
+    query = "SELECT sentence, embedding FROM embeddings"
+    df = pd.read_sql_query(query, conn)
+    conn.close()
+    return df.to_csv(index=False)