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


valmiki_ramayanam_config = {
    "name": "valmiki_ramayanam",
    "title": "Valmiki Ramayanam",
    "banner_url": convert_drive_url_to_direct(
        "https://drive.google.com/file/d/1tr6uwDYw3kec4WeqRS9IrXzZfja1vmci/view?usp=drive_link"
    ),
    "category": ScriptureCategoryConstants.ITHIHASAM,
    "output_dir": "./output/valmiki_ramayanam",
    "collection_name": "valmiki_ramayanam_openai",
    "collection_embedding_fn": "openai",
    "audio_storage": "external",
    "unit": "sloka",
    "unit_field": "sloka",
    # "chapter_order": lambda: get_chapter_order_from_taitriya_samhitha(),
    "field_mapping": {
        "text": "sloka_text_sa",
        "unit_index": "sloka",
        "translation": "explanation",
        "word_by_word_native": "translation",
        "audio": "audio_url",
        "chapter_name": "kanda_name",
        "relative_path": lambda doc: (
            f"॥ वाल्मीकि रामायणम् ॥ {doc.get('kanda_name')} {doc.get('kanda')}.{doc.get('sarga')}.{doc.get('sloka')}"
        ),
        "reference_link": "link",
        "author": lambda doc: "Maharishi Valmiki",
    },
    "metadata_fields": [
        {
            "name": "kanda_name",
            "datatype": "str",
            "label": "Kandam Name",
            "description": "Kandam Name",
            "show_as_filter": True,
            "is_unique": True,
            "lov": lambda: sorted(
                [
                    "BALAKANDA",
                    "AYODHYAKANDA",
                    "ARANYAKANDA",
                    "KISHKINDAKANDA",
                    "SUNDARAKANDA",
                    "YUDDHAKANDA",
                ]
            ),
        },
        {
            "name": "kanda",
            "datatype": "int",
            "label": "Kandam Number",
            "description": "Kandam Number",
            "show_as_filter": True,
            "is_unique": True,
        },
        {
            "name": "sarga",
            "datatype": "int",
            "label": "Sarga Number",
            "description": "Sarga Number",
            "show_as_filter": True,
            "is_unique": True,
        },
        {
            "name": "sloka",
            "datatype": "int",
            "label": "Relative sloka Number",
            "description": "Relative sloka Number",
            "show_as_filter": True,
            "is_unique": True,
        },
        {
            "name": "_global_index",
            "datatype": "int",
            "label": "Absolute sloka Number",
            "description": "Absolute sloka Number",
            "show_as_filter": True,
            "is_unique": True,
        },
        {
            "name": "sloka_text_sa",
            "label": "Lyrics in sanskrit",
            "datatype": "str",
            "description": "The original sloka in sanskrit.",
            "show_as_filter": True,
        },
    ],
    "pdf_path": "./data/valmiki_ramayanam.pdf",
    "source": "https://www.valmiki.iitk.ac.in/",
    "language": "san+eng",
    "example_labels": [
        "About Jatayu",
        "About Hanuman",
        "About Vali",
        "About Sita",
        "About Ravana",
        "A slokam by name",
        "Vibheeshana sharanagathi slokam",
    ],
    "examples": [
        "What is the significance of Jatayu? show some sanskrit verses to support the argument",
        "Show some verses where Hanuman is mentioned",
        "How did Rama kill Vali",
        "How was Sita abducted",
        "How did Rama kill Ravana?",
        "explain sakrudeva prapannaaya shlokam in ramayana",
        "give the shlokam in ramayanam that vibheeshana uses to perform sharanagathi to rama, give the sanskrit shlokam and its meaning",
    ],
    "llm_hints": [],
    "credits": {
        "art": [
            {
                "name": "Sri TNCN",
                "url": "https://www.linkedin.com/in/narashiman-chakravarthy-54326111/",
                "role": "Illustrator",
                "notes": "Original artwork used with permission",
            }
        ],
        "data": [
            {
                "name": "IIT Kanpur",
                "url": "https://www.valmiki.iitk.ac.in/",
                "role": "Data provider",
            }
        ],
        "audio": [
            {
                "name": "IIT Kanpur",
                "url": "https://www.valmiki.iitk.ac.in/",
                "role": "Recitation audio provider",
            },
        ],
        "video": [],
    },    
}