tools modified

Tools.py

@@ -54,7 +54,7 @@ class ScriptureDescriptionToolSpec(BaseToolSpec):
 
 class MantraToolSpec(BaseToolSpec):
     '''
-    To obtain the vedamantra details such as vedamantra, padapata, devata, chandah, rishi etc of vedamantras (or mantras or hyms) from all vedas (RigVeda, AtharvaVeda, SamaVeda, KrishnaYajurVeda, and ShuklaYajurVeda) using the function 
+    To obtain the vedamantra details such as vedamantra, padapatha, devata, chandah, rishi etc of vedamantras (or mantras or hyms) from all vedas (RigVeda, AtharvaVeda, SamaVeda, KrishnaYajurVeda, and ShuklaYajurVeda) using the function 
     `get_vedamantra_details`. The mantra summary like anvaya, mantraVishaya, bhavartha/meaning (adhibautic, ahyatmic, adhidaivic), purpose, usage, tippani 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?
@@ -72,7 +72,6 @@ class MantraToolSpec(BaseToolSpec):
     def _get_mantra_details(self, query):
         try:
             details = get_details_mantra_json(query)
-            print(f"The details of the vedamantra are: {details}")
             return details['mantraHeader']['language'][1]
         except Exception as e:
             raise ValueError(f"Failed to get mantra details: {e}")
@@ -80,7 +79,6 @@ class MantraToolSpec(BaseToolSpec):
     def _query_db(self, conditions):
         try:
             result = self.df_vedamantra[conditions]['mantra_number'].values
-            print(f"The result of vedamatra are: {result}")
             if len(result) == 0:
                 raise ValueError("Mantra not found.")
             return result[0]
@@ -146,6 +144,7 @@ class PadaToolSpec(BaseToolSpec):
         self.df_vedic_content = pd.read_csv(VEDAMANTRA_CSV_PATH,encoding = 'utf-8')
 
     def _get_pada_details_by_scripture(self, pada, scripture_name=None, **kwargs):
+        pada = iast_process(pada)
         try:
             condition = (self.df_terms['Pada'] == pada)
             if scripture_name:
@@ -203,6 +202,7 @@ class PadaToolSpec(BaseToolSpec):
             return []
 
     def get_pada_meaning(self, pada):
+        pada=iast_process(pada)
         try:
             pada_details = self.df_terms[self.df_terms['Pada'] == pada]
             meanings_list = []
@@ -217,34 +217,36 @@ class PadaToolSpec(BaseToolSpec):
             return {"error": f"Required meaning associated with pada is not available. {e}"}    
     
 
-    def get_adibauatic_adidaivic_adhyatmic_meaning_of_pada(self, pada, Pada_position=None, mantraid=None, scripture_name=None, **kwargs):
+    def get_adibauatic_adidaivic_adhyatmic_meaning_of_pada(self, pada, mantraid=None, scripture_name=None, **kwargs):
+        pada = iast_process(pada)
         try:
             if mantraid:
                 details = self.df_terms[(self.df_terms['mantra_id'] == mantraid) & (self.df_terms['Pada'] == pada)]
+                
             else:
                 if scripture_name is not None:
                     details = self._get_pada_details_by_scripture(pada, scripture_name, **kwargs)
-            if Pada_position:
-                details = details[details['Pada_position'] == Pada_position]
-            if details.empty:
+        
+            if not details.empty:
+                pada_details = details.iloc[0]
+                #print(pada_details)
+                mantraID = pada_details['mantra_id']
+                meanings = self._get_vedamantra_meaning(mantraID,MahatmaName=kwargs.get('MahatmaName'))
+                if 'error' in meanings:
+                    return meanings
+                ab_term_morph_list = self._get_pada_morphology(pada_details, meanings['adibhautic'])
+                ad_term_morph_list = self._get_pada_morphology(pada_details, meanings['adidaivic'])
+                at_term_morph_list = self._get_pada_morphology(pada_details, meanings['adhyatmic'])
+                return {
+                    f'adibhautic_info_{pada}': ab_term_morph_list,
+                    'vedamantra_adibhautic_meaning': meanings['adibhautic'],
+                    f'adidavic_info_{pada}': ad_term_morph_list,
+                    'vedamantra_adidavic_meaning': meanings['adidaivic'],
+                    f'adhyatmic_info_{pada}': at_term_morph_list,
+                    'vedamantra_adhyatmic_meaning': meanings['adhyatmic']
+                }
+            else:
                 return {"error": f"No details found for pada '{pada}'"}
-            pada_details = details.iloc[0]
-            #print(pada_details)
-            mantraID = pada_details['mantra_id']
-            meanings = self._get_vedamantra_meaning(mantraID,MahatmaName=kwargs.get('MahatmaName'))
-            if 'error' in meanings:
-                return meanings
-            ab_term_morph_list = self._get_pada_morphology(pada_details, meanings['adibhautic'])
-            ad_term_morph_list = self._get_pada_morphology(pada_details, meanings['adidaivic'])
-            at_term_morph_list = self._get_pada_morphology(pada_details, meanings['adhyatmic'])
-            return {
-                f'adibhautic_info_{pada}': ab_term_morph_list,
-                'vedamantra_adibhautic_meaning': meanings['adibhautic'],
-                f'adidavic_info_{pada}': ad_term_morph_list,
-                'vedamantra_adidavic_meaning': meanings['adidaivic'],
-                f'adhyatmic_info_{pada}': at_term_morph_list,
-                'vedamantra_adhyatmic_meaning': meanings['adhyatmic']
-            }
         except Exception as e:
             logging.error(f"Error in get_adibauatic_adidaivic_adhyatmic_meaning_of_pada: {e}")
-            return {"error": f"Failed to get meaning of the word {pada}. {e}"}
+            return {"error": f"Failed to get meaning of the word {pada}. {e}"}