import json
import logging

from db import SanatanDatabase
from metadata import MetadataFilter, MetadataWhereClause


if __name__ == "__main__":
    logging.basicConfig()
    collection_name = "divya_prabandham"
    database = SanatanDatabase()
    result = database.chroma_client.get_collection(collection_name).get(include=["metadatas","documents"],where={"verse" : 21})

    for index, metadata in enumerate(result["metadatas"]):
        global_index = metadata.get("_global_index",-1)
        verse = metadata["verse"]
        print(f"global_index: {global_index} | verse: {verse} | metadata = {metadata}")