Update app.py

app.py

@@ -79,8 +79,16 @@ def visualize_isnad(taraf_num, yaxis):
         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]
         for n in node:
-            gen_node = narrator_bios[narrator_bios['Rawi ID']==n]['Generation'].iloc[0]
-            name_node = narrator_bios[narrator_bios['Rawi ID']==n]['Famous Name'].iloc[0]
+            gen_node = narrator_bios[narrator_bios['Rawi ID']==n]['Generation'].to_list()
+            if len(gen_node):
+                gen_node = gen_node[0]
+            else:
+                gen_node = -1
+            name_node = narrator_bios[narrator_bios['Rawi ID']==n]['Famous Name'].to_list()
+            if len(name_node):
+                name_node = name_node[0]
+            else:
+                name_node = 'فلان'
             lst_hadith.append([taraf_matns[i], gen_node, name_node, taraf_book[i], taraf_author[i], taraf_hadith_number[i], str(n), str(i)])
     df = pd.DataFrame(lst_hadith, columns = ['Matn', 'Generation', 'Name', 'Book_Name', 'Author', 'Book Hadith Number', 'End Transmitter ID', 'Hadith Number'])