Update app.py

app.py

@@ -71,11 +71,13 @@ def visualize_isnad(taraf_num, yaxis):
     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()
     lst_hadith = []
+    hadith_cleaned = isnad_info['Tarafs Cleaned'].apply(lambda x: taraf_num in x)
+    isnad_hadith = isnad_info[hadith_cleaned]
     for i in range(len(taraf_hadith_split)):
         # This checks each hadith in the Taraf, is that book id hadith id found in each of the edges of isnad_info
         #This loop get the end transmitter of each Hadith in the Taraf
-        isnad_in_hadith1 = isnad_info['Hadiths Cleaned'].apply(lambda x: taraf_hadith_split[i] in x )
-        isnad_hadith1 = isnad_info[isnad_in_hadith1][['Source', 'Destination']]
+        isnad_in_hadith1 = isnad_hadith['Hadiths Cleaned'].apply(lambda x: taraf_hadith_split[i] in x )
+        isnad_hadith1 = isnad_hadith[isnad_in_hadith1][['Source', 'Destination']]
         G = nx.from_pandas_edgelist(isnad_hadith1, source = 'Source', target = 'Destination', create_using = nx.DiGraph())
         node = [int(n) for n, d in G.out_degree() if d == 0][0]
         gen_node = narrator_bios[narrator_bios['Rawi ID']==node]['Generation'].iloc[0]
@@ -84,8 +86,7 @@ def visualize_isnad(taraf_num, yaxis):
     df = pd.DataFrame(lst_hadith, columns = ['Matn', 'Generation', 'Name', 'Book_Name', 'Author', 'Book Hadith Number', 'End Transmitter ID', 'Hadith Number'])
     
     #hadith_cleaned = isnad_info['Hadiths Cleaned'].apply(lambda x: any(i in x for i in taraf_hadith_split) )
-    hadith_cleaned = isnad_info['Tarafs Cleaned'].apply(lambda x: taraf_num in x)
-    isnad_hadith = isnad_info[hadith_cleaned]
+
     isnad_hadith['Teacher'] = isnad_hadith['Source'].apply(lambda x: narrator_bios[narrator_bios['Rawi ID'].astype(int) == int(x)]['Famous Name'].to_list())
     isnad_hadith['Student'] = isnad_hadith['Destination'].apply(lambda x: narrator_bios[narrator_bios['Rawi ID'].astype(int) == int(x)]['Famous Name'].to_list())
     isnad_hadith['Teacher'] = isnad_hadith['Teacher'].apply(lambda x: x[0] if len(x)==1 else 'فلان')