from modules.config.categories import ScriptureCategoryConstants
from modules.google_drive.google_drive_utils import convert_drive_url_to_direct


taitriya_samhitha_config = {
    "name": "taitriya_samhitha",
    "title": "Taitriya Samhitha",
    "banner_url": convert_drive_url_to_direct(
        "https://drive.google.com/file/d/1xxjMzPfqm9Vx95tdmGfSFkzJtqKLuxal/view?usp=drive_link"
    ),
    "category": ScriptureCategoryConstants.VEDAM,
    "output_dir": "./output/taitriya_samhitha",
    "collection_name": "taitriya_samhitha",
    "collection_embedding_fn": "openai",
    "unit": "panchadhi",
    "unit_field": "panchadhi",
    # "chapter_order": lambda: get_chapter_order_from_taitriya_samhitha(),
    "field_mapping": {
        "text": "sanskrit",
        "unit_index": "panchadhi",
        "transliteration": "transliteration",
        "chapter_name": lambda doc: (
            doc.get("prashnam_name")
            if doc.get("prashnam_name") and doc.get("prashnam_name") != "-"
            else (
                doc.get("kandam_name")
                if doc.get("kandam_name") and doc.get("kandam_name") != "-"
                else "-"
            )
        ),
        "relative_path": lambda doc: (
            "॥ तैत्तिरीय-संहिता ॥"
            if not doc.get("panchadhi") or doc.get("panchadhi") == "-"
            else f"॥ तैत्तिरीय-संहिता ॥ {doc.get('kandam')}.{doc.get('prashnam')}.{doc.get('anuvakam')}.{doc.get('panchadhi_index')} || {doc.get('panchadhi_sa')} ||"
        ),
    },
    "metadata_fields": [
        {
            "name": "kandam",
            "datatype": "int",
            "label": "Kandam Number",
            "description": "Kandam Number",
            "show_as_filter": True,
            "is_unique": True,
        },
        {
            "name": "prashnam",
            "datatype": "int",
            "label": "Prashnam Number",
            "description": "Prashnam Number",
            "show_as_filter": True,
            "is_unique": True,
        },
        {
            "name": "anuvakam",
            "datatype": "int",
            "label": "Anuvakam Number",
            "description": "Anuvakam Number",
            "show_as_filter": True,
            "is_unique": True,
        },
        {
            "name": "panchadhi_index",
            "datatype": "int",
            "label": "Relative Panchadhi Number",
            "description": "Relative Panchadhi Number",
            "show_as_filter": True,
            "is_unique": True,
        },
        {
            "name": "panchadhi",
            "datatype": "int",
            "label": "Absolute Panchadhi Number",
            "description": "Absolute Panchadhi Number",
            "show_as_filter": True,
            "is_unique": True,
        },
        {
            "name": "sanskrit",
            "label": "Lyrics in sanskrit",
            "datatype": "str",
            "description": "The original sloka in sanskrit.",
            "show_as_filter": True,
        },
        {
            "name": "transliteration",
            "label": "Transliteration in english",
            "datatype": "str",
            "description": "The original sloka transliterated in English.",
            "show_as_filter": True,
        },
    ],
    "pdf_path": "./data/taitriya_samhitha.pdf",
    "source": "https://sanskritdocuments.org/doc_veda/taittirIyasamhitA.html",
    "language": "san+eng",
    "example_labels": [
        "Taitriya Samhitha",
    ],
    "examples": [
        "Show some verses from Taitriya Samhitha",
    ],
    "llm_hints": [],
    "credits": {
        "art": [
            {
                "name": "Sri TNCN",
                "url": "https://www.linkedin.com/in/narashiman-chakravarthy-54326111/",
                "role": "Illustrator",
                "notes": "Original artwork used with permission",
                "photo_url": "https://drive.google.com/uc?export=download&id=1D4Vo-Mvkp1RdZR6whUkcjvNBIW1yHSUE",
            }
        ],
        "data": [
            {
                "name": "sanskritdocuments.org",
                "url": "https://sanskritdocuments.org/doc_veda/taittirIyasamhitA.html",
                "role": "Data provider",
            }
        ],
        "audio": [
            {
                "name": "Srirangam Sri Vasudevan Srinivasachariar",
                "url": "https://www.youtube.com/@akivasudev",
                "role": "Recitation audio provider",
                "photo_url": "https://drive.google.com/uc?export=download&id=1HB1Nr39j2-55rIeOK_wYNbZZndcGZIaf",
            },
        ],
        "video": [],
    },
}