Update app.py

app.py

@@ -308,6 +308,8 @@ def visualize_hadith_isnad(df, yaxis):
 
 def visualize_narrator_taraf(taraf_num, narrator, yaxis):
     taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
+    num_hadith = taraf.shape[0]
+    taraf['Index'] = np.arange(num_hadith)
     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())
@@ -326,12 +328,21 @@ def visualize_narrator_taraf(taraf_num, narrator, yaxis):
     isnad_pos = nx.nx_agraph.graphviz_layout(G, prog='dot')
     
     matns_with_narrator = []
+    end_node = {}
     for i in range(len(taraf_hadith_split)):
         isnad_in_hadith1 = isnad_hadith['Hadiths Cleaned'].apply(lambda x: taraf_hadith_split[i] in x )
         isnad_hadith1 = isnad_hadith[isnad_in_hadith1]
         G1 = 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]
         if narrator in list(G1.nodes):
             matns_with_narrator.append(taraf_hadith[i])
+            for n in node:
+                if n not in list(end_node.keys()):
+                    end_node[n] = []
+                end_node[n].append(i)
+            
+
+                
 
     narrator_matn_info = taraf[taraf['bookid_hadithid'].isin(matns_with_narrator)]
     
@@ -368,6 +379,8 @@ def visualize_narrator_taraf(taraf_num, narrator, yaxis):
             
         if node == '99999':
             net.add_node(node, font = {'size':50, 'color': 'black'}, color = '#000000', label = f'{node_name} \n ID: {node} - Gen {student_gen}', x= pos[0]*x_stretch, y= -1*pos[1]*y_stretch, size= 70)   
+        elif node in list(end_node.keys()):
+            net.add_node(node, font = {'size':30, 'color': 'red'}, color = value_to_hex(student_narrations), label = f'{node_name} \n {student_rank} \n ID: {node} - Gen {student_gen} \n Hadiths {", ".join(end_node[node])}', x= pos[0]*x_stretch, y= -1*pos[1]*y_stretch, size= 50)
         else:
             net.add_node(node, font = {'size':30, 'color': 'red'}, color = value_to_hex(student_narrations), label = f'{node_name} \n {student_rank} \n ID: {node} - Gen {student_gen}', x= pos[0]*x_stretch, y= -1*pos[1]*y_stretch, size= 50)
     for edge in G.edges:
@@ -382,7 +395,7 @@ def visualize_narrator_taraf(taraf_num, narrator, yaxis):
   display-capture; encrypted-media;" sandbox="allow-modals allow-forms 
   allow-scripts allow-same-origin allow-popups 
   allow-top-navigation-by-user-activation allow-downloads" allowfullscreen="" 
-  allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>""" , narrator_matn_info[['matn', 'Book_Name', 'Author', 'Book_ID', 'Hadith Number']]
+  allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>""" , narrator_matn_info[['matn', 'Book_Name', 'Author', 'Book_ID', 'Hadith Number', 'Index']]