Update database.py

database.py

@@ -4,11 +4,12 @@ from parser import parse_python_code
 import os
 from sklearn.metrics.pairwise import cosine_similarity
 import numpy as np
+import dotenv
 
 # User-configurable variables
 DB_NAME = "python_programs"  # ChromaDB collection name
 HF_DATASET_NAME = "python_program_vectors"  # Hugging Face Dataset name
-HF_TOKEN = "YOUR_HUGGINGFACE_TOKEN"  # Replace with your Hugging Face API token
+HF_TOKEN = os.getenv("HF_KEY")  # Replace with your Hugging Face API token
 PERSIST_DIR = "./chroma_data"  # Directory for persistent storage (optional)
 
 def init_chromadb(persist_dir=PERSIST_DIR):