'mantraID'

Tools.py

@@ -88,13 +88,6 @@ class MantraToolSpec(BaseToolSpec):
         super().__init__()
         self.df_vedamantra = pd.read_csv(VEDAMANTRA_CSV_PATH, encoding='utf-8')
 
-    def _get_mantra_details(self, query):
-        try:
-            details = get_details_mantra_json(query)
-            return details['mantraHeader']['language'][1]
-        except Exception as e:
-            raise ValueError(f"Failed to get mantra details: {e}")
-
     def _get_mantra_details_by_scripture(self, scripture_name=None, KandahNumber=None,MandalaNumber=None, ArchikahNumber=None, 
                                                            ShuktaNumber=None, PrapatakNumber=None, MantraNumber=None,
                                                            AnuvakNumber=None, AdhyayaNumber=None):
@@ -133,15 +126,17 @@ class MantraToolSpec(BaseToolSpec):
                                                            AnuvakNumber=None, AdhyayaNumber=None):
         try:
             if mantraid:
-                query = f"SELECT mantra_json FROM veda_content WHERE mantra_number = '{mantraid}'"
+                MantraID = mantraid
             else:
                 filter_df = self._get_mantra_details_by_scripture(scripture_name=scripture_name, KandahNumber=KandahNumber,MandalaNumber=MandalaNumber, ArchikahNumber=ArchikahNumber, 
                                                            ShuktaNumber=ShuktaNumber, PrapatakNumber=PrapatakNumber, MantraNumber=MantraNumber,
                                                            AnuvakNumber=AnuvakNumber, AdhyayaNumber=AdhyayaNumber)
                 if filter_df is not None:
-                    mantra_id = filter_df.iloc[0]['mantra_number']
-                    query = f"SELECT mantra_json FROM veda_content WHERE mantra_number = '{mantra_id}'"
-            return self._get_mantra_details(query)
+                    MantraID = filter_df.iloc[0]['mantra_number']
+            query = f"SELECT mantra_json FROM veda_content WHERE mantra_number = '{MantraID}'"
+            details = get_details_mantra_json(query)
+            mantra_details = details['mantraHeader']['language'][1]
+            return mantra_details
         except Exception as e:
             return json.dumps({"error": str(e)})
 
@@ -159,16 +154,14 @@ class MantraToolSpec(BaseToolSpec):
         '''
         try:
             if mantraid:
-                query = f"SELECT mantra_json FROM veda_content WHERE mantra_number = '{mantraid}'"
+                MantraID = mantraid
             else:
                 filtered_df = self._get_mantra_details_by_scripture(scripture_name=scripture_name, KandahNumber=KandahNumber,MandalaNumber=MandalaNumber, ArchikahNumber=ArchikahNumber, 
                                                            ShuktaNumber=ShuktaNumber, PrapatakNumber=PrapatakNumber, MantraNumber=MantraNumber,
                                                            AnuvakNumber=AnuvakNumber, AdhyayaNumber=AdhyayaNumber)   
                 if filtered_df is not None:
-                    mantra_id = filtered_df.iloc[0]['mantra_number']
-                    query = f"SELECT mantra_json FROM veda_content WHERE mantra_number = '{mantra_id}'"
-                else:
-                    return None      
+                    MantraID = filtered_df.iloc[0]['mantra_number']
+            query = f"SELECT mantra_json FROM veda_content WHERE mantra_number = '{MantraID}'"
             json_dict = get_details_mantra_json(query)
             mantra_summary = json_dict['mantraSummary']['language']
             summary_dict = {"Roman-IAST summary of vedamantra": json_dict['mantraSummary']['language'][1]}