from modules.config.categories import ScriptureCategoryConstants
from modules.google_drive.google_drive_utils import convert_drive_url_to_direct
from modules.scripture_helpers.vishnu_sahasranamam_helper import (
    get_chapter_order_from_sahasranamam,
    get_chapters_from_sahasranamam,
)

vishnu_sahasranamam_config = {
    "name": "vishnu_sahasranamam",
    "title": "Vishnu Sahasranamam",
    "banner_url": convert_drive_url_to_direct(
        "https://drive.google.com/file/d/1JBfQT9cA4HZZk2niUx96U67ZyoebVJHa/view?usp=drive_link"
    ),
    "category" : ScriptureCategoryConstants.STOTRAM,
    "output_dir": "./output/vishnu_sahasranamam",
    "collection_name": "vishnu_sahasranamam_openai",
    "collection_embedding_fn": "openai",
    "unit": "verse",
    "chapter_order": lambda: get_chapter_order_from_sahasranamam(),
    "field_mapping": {
        "text": "sanskrit",
        "title": lambda doc: f"Verse {doc.get('verse','')}",
        "unit_index": "verse",
        "transliteration": "transliteration",
        "translation": "translation",
        "author": lambda doc: f"Sage Vyasa",
        # "chapter_name": "chapter",
        "relative_path": lambda doc: f"॥ {doc.get("chapter","")} ॥ {doc.get("verse","")}",
    },
    "metadata_fields": [
        {
            "name": "chapter",
            "datatype": "str",
            "label": "Chapter Name",
            "description": "Name of the Chapter",
            "show_as_filter": True,
            "component": "dropdown",
            "lov": lambda: get_chapters_from_sahasranamam(),
            "is_unique": True,
        },
        {
            "name": "verse",
            "label": "Verse Number",
            "datatype": "int",
            "description": "The verse number of the sloka.",
            "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": "translation",
            "label": "English Translation",
            "datatype": "str",
            "description": "The english translation.",
            "show_as_filter": True,
        },
        {
            "name": "transliteration",
            "label": "English Transliteration",
            "datatype": "str",
            "description": "The english transliteration.",
            "show_as_filter": True,
        },
    ],
    "pdf_path": "./data/vishnu_sahasranamam.pdf",
    "source": "https://www.swami-krishnananda.org/vishnu/Sri_Vishnu_Sahasranama_Stotram.pdf",
    "language": "san+eng",
    "example_labels": ["Vanamali", "1000 names", "Sanskrit text search"],
    "examples": [
        "Vanamali",
        "Show some of the 1000 names of Vishnu along with their meaning",
        "show the verse that begins with शुक्लाम्बरधरं",
    ],
    "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": "swami-krishnananda.org",
                "url": "https://www.swami-krishnananda.org/vishnu/Sri_Vishnu_Sahasranama_Stotram.pdf",
                "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",
            },
            {
                "name": "Srirangam Vikram Vasudevan",
                "url": "https://www.youtube.com/@onedayonepasuram6126",
                "role": "Recitation audio provider",
                "photo_url": "https://yt3.googleusercontent.com/lA5fiJuRDVGxOLkszb6O5DaiR1N7WcRVDHtJSQjO8IxCK-Y3VfMtYZP2Wm1SP0gUx8vj87jA=s160-c-k-c0x00ffffff-no-rj",
            },            
            {
                "name": "Srirangam Sashank Vikram",
                "url": "https://www.youtube.com/@onedayonepasuram6126",
                "role": "Recitation audio provider",
                "photo_url": "https://drive.google.com/uc?export=download&id=1WukF9EBkxg00GjcuoX8enGhY3XOqwlVr",
            },            
        ],
        "video": [],
    },    
}