Update database_utils.py

database_utils.py

@@ -2,7 +2,6 @@ import sqlite3
 import pandas as pd
 import numpy as np
 
-
 def init_db():
     conn = sqlite3.connect('embeddings.db')
     c = conn.cursor()
@@ -10,7 +9,6 @@ def init_db():
                  (sentence TEXT, embedding BLOB)''')
     conn.commit()
     conn.close()
-    return "Database initialized successfully!"
 
 def save_embeddings_to_db(sentence, embedding):
     conn = sqlite3.connect('embeddings.db')
@@ -20,28 +18,26 @@ def save_embeddings_to_db(sentence, embedding):
     conn.commit()
     conn.close()
 
-def clear_all_entries():
-    conn = sqlite3.connect('embeddings.db')
-    c = conn.cursor()
-    c.execute("DELETE FROM embeddings")
-    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]
+    embeddings = [np.frombuffer(row[1], dtype=np.float32) for row in data]
     sentences = [row[0] for row in data]
     return embeddings, sentences
 
+def clear_all_entries():
+    conn = sqlite3.connect('embeddings.db')
+    c = conn.cursor()
+    c.execute("DELETE FROM embeddings")
+    conn.commit()
+    conn.close()
 
 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)
+    return df.to_csv(index=False)