Spaces:
Sleeping
Sleeping
| import json | |
| import chardet | |
| import streamlit as st | |
| import pandas as pd | |
| from llama_index.core.tools.tool_spec.base import BaseToolSpec | |
| class ScriptureDescriptionToolSpec(BaseToolSpec): | |
| ''' | |
| Purpose: Obtains the description or summary about vedas, mandalas, kandas, shuktas, archakah, adhyaya, and other scriptural elements. | |
| Returns: A dictionary containing the description or basic information about the specified scriptural element. | |
| Sample query: | |
| 1. Describe the first kandah, second shukta from Atharvaveda? | |
| 2. Summarize ShuklaYajurVeda? | |
| 3. What is the difference between ShuklaYajurVeda and KrishnaYajurVeda? | |
| ''' | |
| # Define the functions that we export to the LLM | |
| spec_functions = ["get_description"] | |
| with open("Data/scripture_descriptions.csv", 'rb') as f: | |
| result = chardet.detect(f.read()) | |
| encoding = result['encoding'] | |
| df = pd.read_csv("Data/scripture_descriptions.csv", encoding=encoding) | |
| def get_description(_self, level_0, level_1:int=None, level_2:int=None, level_3:int=None): | |
| """ | |
| To get the description or basic information about vedas/mandalas/kandas/shukatas/archakah/adhyaya and others. | |
| """ | |
| try: | |
| if level_3 is not None: | |
| # Case with Level-2 specified | |
| result = _self.df[(_self.df['scripture_name'].str.lower() == level_0.lower()) | |
| & (_self.df['level_1'] == str(level_1)) | |
| & (_self.df['level_2'] == str(level_2)) & (_self.df['level_3'] == str(level_3))] | |
| elif level_2 is not None: | |
| # Case with Level-2 specified | |
| result = _self.df[(_self.df['scripture_name'].str.lower() == level_0.lower()) | |
| & (_self.df['level_1'] == str(level_1)) & (_self.df['level_2'] == str(level_2))] | |
| elif level_1 is not None: | |
| # Case with Level-1 specified | |
| result = _self.df[(_self.df['scripture_name'].str.lower() == level_0.lower()) | |
| & (_self.df['level_1'] == str(level_1))] | |
| else: | |
| # Case with only Level-0 specified | |
| result = _self.df[_self.df['scripture_name'].str.lower() == level_0.lower()] | |
| return result.iloc[0].to_dict() | |
| except IndexError as e: | |
| return json.dumps({"error": f"Failed to get scripture description. {e}"}) | |
| class MantraToolSpec(BaseToolSpec): | |
| ''' | |
| To obtain translations or meaning of vedamantras from RigVeda and AtharvaVeda using the function `get_translation`. | |
| The mantra details such as vedamantra, padapatha, rishi, chandah, devata, and swarah from the vedas accessible through the function `get_vedamantra_details`. | |
| The mantra summary like anvaya, adhibautic, ahyatmic, adhidaivic meaning of vedamantra accessible using the function 'get_vedamantra_summary' | |
| Sample Query: | |
| 1. What is the vedamantra of the mantra from Rigveda, first mandala, first shukta, and first mantra? | |
| 2. What is the devata of the vedamantra from Rigveda, first mandala, first shukta, and first mantra? | |
| 3. What is the meaning of the vedamantra from Rigveda, first mandala, first shukta, and first mantra written by Tulsi Ram? | |
| 4. What is the (adhibautic) meaning of the vedamantra from RigVeda, first mandala, first shukta, and first mantra? | |
| ''' | |
| spec_functions = ["get_translation", "get_vedamantra_details", "get_vedamantra_summary"] | |
| TRANSLATION_CSV_PATH = 'Data/trans_Rig_Ath_index_v2.csv' | |
| VEDAMANTRA_CSV_PATH = "Data/veda_content_modified_v3.csv" | |
| def __init__(self): | |
| super().__init__() | |
| self.df_translation = pd.read_csv(self.TRANSLATION_CSV_PATH, encoding='utf-8') | |
| self.df_vedamantra = pd.read_csv(self.VEDAMANTRA_CSV_PATH, encoding='utf-8') | |
| def get_translation(_self, mantraid=None, scripture_name=None, MahatmaName=None, KandahNumber=None, | |
| MandalaNumber=None, ArchikahNumber=None, ShuktaNumber=None, | |
| AnvayaNumber=None, PrapatakNumber=None, MantraNumber=None, | |
| AnuvakNumber=None, AdhyayaNumber=None): | |
| """ | |
| Get the translation of mantras from RigVeda and AtharvaVeda. | |
| Sample Query: | |
| 1. What is the translation of Tulsi Ram of the vedamantra from Rigveda, first mandala, first shukta, and first mantra? | |
| 2. What is the translation or adhibautic meaning of the vedamantra from RigVeda, first mandala, first shukta, and first mantra? | |
| 3. What is the subject of the mantra 1.1.84.1? | |
| """ | |
| try: | |
| if mantraid is None: | |
| scripture_name_lower = scripture_name.lower() if scripture_name is not None else False | |
| if scripture_name_lower == 'rigveda': | |
| details = _self.df_translation[ | |
| (_self.df_translation['scripture_name'].str.lower() == scripture_name_lower) | |
| & (_self.df_translation['MandalaNumber'] == MandalaNumber) | |
| & (_self.df_translation['ShuktaNumber'] == int(ShuktaNumber)) | |
| & (_self.df_translation['MantraNumber'] == int(MantraNumber)) | |
| ].to_dict(orient='records') | |
| elif scripture_name_lower == 'atharvaveda': | |
| details = _self.df_translation[ | |
| (_self.df_translation['scripture_name'].str.lower() == scripture_name_lower) | |
| & (_self.df_translation['KandahNumber'] == KandahNumber) | |
| & (_self.df_translation['ShuktaNumber'] == ShuktaNumber) | |
| & (_self.df_translation['MantraNumber'] == MantraNumber)].to_dict(orient='records') | |
| elif scripture_name_lower == 'samaveda': | |
| details = _self.df_translation[ | |
| (_self.df_translation['scripture_name'].str.lower() == scripture_name_lower) | |
| & (_self.df_translation['ArchikahNumber'] == ArchikahNumber) | |
| & (_self.df_translation['ShuktaNumber'] == ShuktaNumber) | |
| & (_self.df_translation['MantraNumber'] == MantraNumber)].to_dict(orient='records') | |
| elif scripture_name_lower == 'krishnayajurveda': | |
| details = _self.df_translation[ | |
| (_self.df_translation['scripture_name'].str.lower() == scripture_name_lower) | |
| & (_self.df_translation['PrapatakNumber'] == PrapatakNumber) | |
| & (_self.df_translation['AnuvakNumber'] == AnuvakNumber) | |
| & (_self.df_translation['MantraNumber'] == MantraNumber)].to_dict(orient='records') | |
| else: | |
| details = _self.df_translation[ | |
| (_self.df_translation['scripture_name'].str.lower() == scripture_name_lower) | |
| & (_self.df_translation['AdhyayaNumber'] == AdhyayaNumber) | |
| & (_self.df_translation['MantraNumber'] == MantraNumber) | |
| ].to_dict(orient='records') | |
| else: | |
| details = _self.df_translation[_self.df_translation['mantra_id'] == mantraid].to_dict(orient='records') | |
| if MahatmaName is not None: | |
| for item in details: | |
| if item['MahatmaName'] == MahatmaName: | |
| return item | |
| else: | |
| return details | |
| except Exception as e: | |
| return json.dumps({"error": f"Failed to get translation. {e}"}) | |
| def get_vedamantra_details(_self, mantraid=None, scripture_name=None, KandahNumber=None, | |
| MandalaNumber=None, ArchikahNumber=None, ShuktaNumber=None, | |
| AnvayaNumber=None, PrapatakNumber=None, MantraNumber=None, | |
| AnuvakNumber=None, AdhyayaNumber=None): | |
| """ | |
| To obtain the vedamantra details such as vedamantra, padapata, devata, rishi, swarah, and chandah. | |
| 1. What is the vedamantra of the mantra from Rigveda, first mandala, first shukta, and first mantra? | |
| 2. What is the devata of the vedamantra from Rigveda, first mandala, first shukta, and first mantra? | |
| """ | |
| try: | |
| if mantraid is None: | |
| scripture_name_lower = scripture_name.lower() if scripture_name is not None else False | |
| if scripture_name_lower == 'rigveda': | |
| conditions = (_self.df_vedamantra['scripture_name'].str.lower() == scripture_name_lower) & \ | |
| (_self.df_vedamantra['MandalaNumber'] == MandalaNumber) & \ | |
| (_self.df_vedamantra['ShuktaNumber'] == ShuktaNumber) & \ | |
| (_self.df_vedamantra['MantraNumber'] == str(MantraNumber)) | |
| details = _self.df_vedamantra[conditions]['mantra_json'].values | |
| vedamantra_details = json.loads(details[0])['mantraHeader']['language'][1]['mandala']['shukta']['mantra'] | |
| elif scripture_name_lower == 'atharvaveda': | |
| conditions = (_self.df_vedamantra['scripture_name'].str.lower() == scripture_name_lower) & \ | |
| (_self.df_vedamantra['KandahNumber'] == KandahNumber) & \ | |
| (_self.df_vedamantra['ShuktaNumber'] == ShuktaNumber) & \ | |
| (_self.df_vedamantra['MantraNumber'] == str(MantraNumber)) | |
| details = _self.df_vedamantra[conditions]['mantra_json'].values | |
| vedamantra_details = json.loads(details[0])['mantraHeader']['language'][1]['kandah']['shukta']['mantra'] | |
| elif scripture_name_lower == 'samaveda': | |
| conditions = (_self.df_vedamantra['scripture_name'].str.lower() == scripture_name_lower) & \ | |
| (_self.df_vedamantra['ArchikahNumber'] == ArchikahNumber) & \ | |
| (_self.df_vedamantra['ShuktaNumber'] == ShuktaNumber) & \ | |
| (_self.df_vedamantra['MantraNumber'] == str(MantraNumber)) | |
| details = _self.df_vedamantra[conditions]['mantra_json'].values | |
| vedamantra_details = json.loads(details[0])['mantraHeader']['language'][1]['archikah'] | |
| elif scripture_name_lower == 'krishnayajurveda': | |
| conditions = (_self.df_vedamantra['scripture_name'].str.lower() == scripture_name_lower) & \ | |
| (_self.df_vedamantra['PrapatakNumber'] == PrapatakNumber) & \ | |
| (_self.df_vedamantra['AnuvakNumber'] == AnuvakNumber) & \ | |
| (_self.df_vedamantra['MantraNumber'] == str(MantraNumber)) | |
| details = _self.df_vedamantra[conditions]['mantra_json'].values | |
| vedamantra_details = json.loads(details[0])['mantraHeader']['language'][1]['kandah']['prapatak']['anuvak'] | |
| else: | |
| conditions = (_self.df_vedamantra['scripture_name'].str.lower() == scripture_name_lower) & \ | |
| (_self.df_vedamantra['AdhyayaNumber'] == AdhyayaNumber) & \ | |
| (_self.df_vedamantra['MantraNumber'] == str(MantraNumber)) | |
| details = _self.df_vedamantra[conditions]['mantra_json'].values | |
| vedamantra_details = json.loads(details[0])['mantraHeader']['language'][1]['adhyaya']['mantra'] | |
| else: | |
| # Handle case when mantraid is provided | |
| details = _self.df_vedamantra[_self.df_vedamantra['mantra_number'] == mantraid]['mantra_json'].values | |
| vedamantra_details = json.loads(details[0])['mantraHeader']['language'][1] | |
| return vedamantra_details | |
| except Exception as e: | |
| return json.dumps({"error": f"Failed to get vedamantra details. {str(e)}"}) | |
| def get_vedamantra_summary(_self, mantraid=None, scripture_name=None, KandahNumber=None, | |
| MandalaNumber=None, ArchikahNumber=None, ShuktaNumber=None, | |
| AnvayaNumber=None, PrapatakNumber=None, MantraNumber=None, | |
| AnuvakNumber=None, AdhyayaNumber=None): | |
| ''' | |
| To obtain the vedamantra summary like anvaya, translation, adhibautic, adhyatmic, adhidaivic meaning of the mantra. | |
| 1. What is the adhibautic meaning of the mantra from AtharvaVeda, first kandah, first shukta, and first mantra? | |
| 2. What is the anvaya of the vedamantra from Rigveda, first mandala, first shukta, and first mantra? | |
| ''' | |
| try: | |
| if mantraid is None: | |
| scripture_name_lower = scripture_name.lower() if scripture_name is not None else False | |
| if scripture_name_lower == 'rigveda': | |
| details = _self.df_vedamantra[ | |
| (_self.df_vedamantra['scripture_name'].str.lower() == scripture_name_lower) | |
| & (_self.df_vedamantra['MandalaNumber'] == MandalaNumber) | |
| & (_self.df_vedamantra['ShuktaNumber'] == ShuktaNumber) | |
| & (_self.df_vedamantra['MantraNumber'] == str(MantraNumber)) | |
| ]['mantra_json'].values | |
| elif scripture_name_lower == 'atharvaveda': | |
| details = _self.df_vedamantra[ | |
| (_self.df_vedamantra['scripture_name'].str.lower() == scripture_name_lower) | |
| & (_self.df_vedamantra['KandahNumber'] == KandahNumber) | |
| & (_self.df_vedamantra['ShuktaNumber'] == ShuktaNumber) | |
| & (_self.df_vedamantra['MantraNumber'] == str(MantraNumber)) | |
| ]['mantra_json'].values | |
| elif scripture_name_lower == 'samaveda': | |
| details = _self.df_vedamantra[ | |
| (_self.df_vedamantra['scripture_name'].str.lower() == scripture_name_lower) | |
| & (_self.df_vedamantra['ArchikahNumber'] == ArchikahNumber) | |
| & (_self.df_vedamantra['ShuktaNumber'] == ShuktaNumber) | |
| & (_self.df_vedamantra['MantraNumber'] == str(MantraNumber)) | |
| ]['mantra_json'].values | |
| elif scripture_name_lower == 'krishnayajurveda': | |
| details = _self.df_vedamantra[ | |
| (_self.df_vedamantra['scripture_name'].str.lower() == scripture_name_lower) | |
| & (_self.df_vedamantra['PrapatakNumber'] == PrapatakNumber) | |
| & (_self.df_vedamantra['AnuvakNumber'] == AnuvakNumber) | |
| & (_self.df_vedamantra['MantraNumber'] == str(MantraNumber)) | |
| ]['mantra_json'].values | |
| else: | |
| details = _self.df_vedamantra[ | |
| (_self.df_vedamantra['scripture_name'].str.lower() == scripture_name_lower) | |
| & (_self.df_vedamantra['AdhyayaNumber'] == AdhyayaNumber) | |
| & (_self.df_vedamantra['MantraNumber'] == str(MantraNumber)) | |
| ]['mantra_json'].values | |
| else: | |
| details = _self.df_vedamantra[_self.df_vedamantra['mantra_number'] == mantraid]['mantra_json'].values | |
| jsonDict = json.loads(details[0]) | |
| mantraSummary = jsonDict['mantraSummary']['language'] | |
| mantraSummary_IAST = jsonDict['mantraSummary']['language'][1] | |
| vedamantra_summary = {"Roman-IAST summary of vedamantra": mantraSummary_IAST} | |
| for item in mantraSummary: | |
| if item['languageName'] == 'English': | |
| vedamantra_summary.update({"English summary of vedamantra": item}) | |
| return vedamantra_summary | |
| except Exception as e: | |
| return json.dumps({"error": f"Failed to get vedamantra summary. {e}"}) | |