setup of multi hadith isnad viewing with specific matn tracking

app.py

@@ -61,13 +61,40 @@ def value_to_hex(value):
     return "#{:02X}{:02X}{:02X}".format(int(rgba_color[0] * 255), int(rgba_color[1] * 255), int(rgba_color[2] * 255))
 
 
+def visualize_subTaraf(taraf_num, hadith_str, yaxis):
+    hadith_list = hadith_str.split(',')
+    hadith_list = [hadith.strip() for hadith in hadith_list]
+    hadiths = np.array([], dtype=int)
+    for hadith in hadith_list:
+        if '-' in hadith:
+            if hadith.count('-') > 1:
+                #print('Please use only one Dash mark!')
+                raise gr.Error('Please use only one Dash mark!')
+            hadith_multi = hadith.strip().split('-')
+            if any([not had.isnumeric() for had in hadith_multi]):
+                #print('Invalid Begining')
+                raise gr.Error('Invalid Begining')
+            elif len(hadith_multi) != 2:
+                #print('Two numbers for a range of Hadith numbers please!')
+                raise gr.Error('Two numbers for a range of Hadith numbers please!')
+            hadith_multi = [int(had) for had in hadith_multi]
+            hadiths = np.append(hadiths, np.arange(hadith_multi[0], hadith_multi[1]))
+        elif hadith.isnumeric():
+            hadiths = np.append(hadiths, int(hadith))
+        else:
+            #print('Invalid Data format!')
+            raise gr.Error("Invalid Data format!")
+    return hadiths
+    
+
 def visualize_isnad(taraf_num, yaxis):
-    taraf_hadith = matn_info[matn_info['taraf_ID'] == taraf_num]['bookid_hadithid'].to_list()
-    taraf_matns = matn_info[matn_info['taraf_ID'] == taraf_num]['matn'].to_list()
+    taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
+    taraf_hadith = taraf['bookid_hadithid'].to_list()
+    taraf_matns = taraf['matn'].to_list()
     taraf_hadith_split = [i.split('_') for i in taraf_hadith]
-    taraf_book = matn_info[matn_info['taraf_ID'] == taraf_num]['Book_Name'].to_list()
-    taraf_author = matn_info[matn_info['taraf_ID'] == taraf_num]['Author'].to_list()
-    taraf_hadith_number = matn_info[matn_info['taraf_ID'] == taraf_num]['Hadith Number'].to_list()
+    taraf_book = taraf['Book_Name'].to_list()
+    taraf_author = taraf['Author'].to_list()
+    taraf_hadith_number = taraf['Hadith Number'].to_list()
     lst_hadith = []
     hadith_cleaned = isnad_info['Tarafs Cleaned'].apply(lambda x: taraf_num in x)
     isnad_hadith = isnad_info[hadith_cleaned]
@@ -152,7 +179,7 @@ def taraf_booknum(taraf_num):
     taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
     return taraf[['matn', 'Book_ID', 'Hadith Number', 'Book_Name', 'Author']]
 
-def visualize_subTaraf(df, yaxis):
+def visualize_hadith_isnad(df, yaxis):
     df['bookid_hadithid'] = df['Book_ID'].astype(str) + '_' + df['Hadith Number'].astype(str)
     hadith = matn_info[matn_info['bookid_hadithid'].isin(df['bookid_hadithid'])]
     taraf_hadith_split = [i.split('_') for i in hadith['bookid_hadithid'].to_list()]
@@ -230,5 +257,5 @@ with gr.Blocks() as demo:
             row_count=5,
             col_count=(2, "fixed"))
         btn_hadith = gr.Button('Visualize')
-        btn_hadith.click(fn=visualize_subTaraf, inputs=[hadith_selection, yyaxis], outputs=[gr.HTML()])
+        btn_hadith.click(fn=visualize_hadith_isnad, inputs=[hadith_selection, yyaxis], outputs=[gr.HTML()])
 demo.launch()