from pymilvus import Collection, connections
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))


# Now, connect to the collection
collection = Collection("faces")

# Query the collection to find entries where the 'name' field is empty or None
query = 'name == "YAB DATO SERI ANWAR IBRAHIM"'  # Looking for entities where 'name' is empty

# Perform the query to find entities with empty 'name' fields
results = collection.query(query, output_fields=["id", "name"])

# Check and delete entities with empty 'name'
if results:
    ids_to_delete = [str(result["id"]) for result in results]
    id_expr = f"id in [{', '.join(ids_to_delete)}]"
    collection.delete(expr=id_expr)
    print(f"✅ Deleted entities: {ids_to_delete}")
else:
    print("❌ No entities found for deletion.")