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


taitriya_aranyakam_config = {
    "name": "taitriya_aranyakam",
    "title": "Taitriya Aranyakam",
    "banner_url": convert_drive_url_to_direct(
        "https://drive.google.com/file/d/1-8weVFe58DWPef2fuVyu5It3r1pCXgc2/view?usp=drive_link"
    ),
    "category": ScriptureCategoryConstants.VEDAM,
    "output_dir": "./output/taitriya_aranyakam",
    "collection_name": "taitriya_aranyakam",
    "collection_embedding_fn": "openai",
    "unit": "panchadhi",
    "unit_field": "panchadhi",
    # "chapter_order": lambda: get_chapter_order_from_taitriya_aranyakam(),
    "field_mapping": {
        "text": "sanskrit",
        "unit_index": "panchadhi",
        "transliteration": "transliteration",
        "chapter_name": "prashnam_name",
        "relative_path": lambda doc: (
            "॥ तैत्तिरीय-आरण्यक ॥"
            if not doc.get("panchadhi") or doc.get("panchadhi") == "-"
            else f"॥ तैत्तिरीय-आरण्यक ॥ {doc.get('prashnam')}.{doc.get('anuvakam')}.{doc.get('panchadhi_index')} || {doc.get('panchadhi_sa')} ||"
        ),
    },
    "metadata_fields": [
        {
            "name": "prashnam_name",
            "datatype": "str",
            "label": "Prashnam Name",
            "description": "Name of the Prashnam",
            "show_as_filter": True,
            "component": "dropdown",
            "lov": lambda: [
                "आरण्यके प्रथमः प्रश्नः १",
                "आरण्यके द्वितीयः प्रश्नः २",
                "आरण्यके तृतीयः प्रश्नः ३",
                "आरण्यके चतुर्थः प्रश्नः ४",
                "आरण्यके पञ्चमः प्रश्नः ५",
                "आरण्यके षष्ठः प्रश्नः ६",
                "आरण्यके सप्तमः प्रश्नः ७",
                "आरण्यके अष्टमः प्रश्नः ८",
                "आरण्यके नवमः प्रश्नः ९",
                "आरण्यके दशमः प्रश्नः १०",
                "एकाग्निकाण्डे प्रथमः प्रश्नः १",
                "एकाग्निकाण्डे द्वितीयः प्रश्नः २",
            ],
            "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_aranyakam.pdf",
    "source": "https://sanskritdocuments.org/doc_veda/taittirIyaAraNyaka.html",
    "language": "san+eng",
    "example_labels": [
        "Taitriya AraNyaka",
    ],
    "examples": [
        "Show some verses from Taitriya AraNyaka",
    ],
    "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/taittirIyaAraNyaka.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": [],
    },    
}