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

geethartha_sangraha_config = {
    "name": "geethartha_sangraha",
    "title": "Geethartha Sangraha",
    "banner_url": convert_drive_url_to_direct(
        "https://drive.google.com/file/d/1uLPo1Upx1vLJ6InxqOEMspAOz4UT1F4r/view?usp=drive_link"
    ),
    "category": ScriptureCategoryConstants.SRI_SOOKTHI,
    "output_dir": "./outputs/geethartha_sangraha",
    "collection_name": "geethartha_sangraha",
    "collection_embedding_fn": "openai",
    "unit": "slokam",
    "unit_field": "_global_index",
    "field_mapping": {
        "text": "lyrics_sa",
        "chapter_name": "chapter_name",
        "unit_index": "_global_index",
        "relative_path": lambda doc: f"{doc.get('chapter_name')} — {doc.get('slokam_index')}",
        "author": lambda doc: "Shri Yamunacharya",
    },
    "metadata_fields": [
        {
            "name": "chapter_name",
            "datatype": "str",
            "label": "Chapter Name",
            "description": "Chapter title",
            "show_as_filter": True,
            "component": "dropdown",
            "lov": lambda: ["SLOKAM"],
        },
        {
            "name": "_global_index",
            "datatype": "int",
            "label": "Global Index",
            "description": "Absolute verse index",
            "show_as_filter": True,
            "is_unique": True,
        },
        {
            "name": "chapter_index",
            "datatype": "int",
            "label": "Chapter Index",
            "description": "Chapter index",
        },
        {
            "name": "slokam_index",
            "datatype": "int",
            "label": "Verse Number (in chapter)",
            "description": "Sequential verse index inside the chapter",
            "show_as_filter": True,
        },
        {
            "name": "lyrics_sa",
            "datatype": "str",
            "label": "Sanskrit Verse",
            "description": "Original verse text in Sanskrit",
            "show_as_filter": True,
        },
    ],
    "pdf_path": "./data/geethartha_sangraha.pdf",
    "source": "sanskritdocuments.org",
    "language": "san",
    "example_labels": [],
    "examples": [],
    "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_giitaa/gitaarthasangraha.html",
                "role": "Data Provider",
            }
        ],
        "audio": [
            {
                "name": "Sri Vasudevan Srinivasachariar Swamy(Srirangam)",
                "url": "https://www.youtube.com/@akivasudev",
                "role": "Recitation audio contributor",
                "context": [],
                "photo_url": "https://drive.google.com/uc?export=download&id=1HB1Nr39j2-55rIeOK_wYNbZZndcGZIaf",
            },
        ],
        "video": [],
    },
    "audio_modes": {
        "recitation": "Recitation",
    },
    "title_translations": {
        "en": "Geethartha Sangraha",
        "ta": "கீதார்த்த சங்கிரகம்",
        "te": "గీతార్థ సంగ్రహం",
        "hi": "गीतार्थ संग्रह",
        "mr": "गीतार्थ संग्रह",
        "kn": "ಗೀತಾರ್ಥ ಸಂಗ್ರಹ",
        "ml": "ഗീതാർത്ഥ സംഗ്രഹം",
        "or": "ଗୀତାର୍ଥ ସଂଗ୍ରହ",
        "pa": "ਗੀਤਾਰਥ ਸੰਗ੍ਰਹ",
    },
}