File size: 434 Bytes
55a76a3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | import shutil
import os
import sqlite3
def reset_data():
print("Resetting operational data...")
if os.path.exists("chroma_db"):
print("Removing chroma_db directory...")
shutil.rmtree("chroma_db")
if os.path.exists("data/app.db"):
print("Removing sqlite database...")
os.remove("data/app.db")
print("Data reset complete.")
if __name__ == "__main__":
reset_data()
|