from pymilvus import connections, utility
from dotenv import load_dotenv
import os
load_dotenv()


milvus_host = os.getenv("MILVUS_HOST", "milvus")  # default localhost
milvus_port = os.getenv("MILVUS_PORT", "19530")       # default 19530

connections.connect("default", host=milvus_host, port=int(milvus_port))


# Delete the collection
collection_name = "faces"
if utility.has_collection(collection_name):
    utility.drop_collection(collection_name)
    print(f"Collection '{collection_name}' deleted.")
else:
    print(f"Collection '{collection_name}' does not exist.")