testing hierarchal graph format

app.py

@@ -94,59 +94,46 @@ def visualize_isnad(taraf_num, yaxis):
     isnad_hadith['Student'] = isnad_hadith['Student'].apply(lambda x: x[0] if len(x)==1 else 'فلان')
 
     end_nodes = df['End Transmitter ID'].tolist() 
+    G = nx.from_pandas_edgelist(isnad_hadith, source = 'Source', target = 'Destination', create_using = nx.DiGraph())
+    isnad_pos = nx.nx_agraph.graphviz_layout(G, prog='dot')
+    x_stretch = 4
+    y_stretch = 4
     net = Network(directed =True)
-    for _, row in isnad_hadith.iterrows():
-        source = row['Source']
-        target = row['Destination']
-        teacher_info = narrator_bios[narrator_bios['Rawi ID'] == int(row['Source'])]
-        student_info = narrator_bios[narrator_bios['Rawi ID'] == int(row['Destination'])]
-        
-        teacher_narrations = teacher_info['Number of Narrations'].to_list()
-        if len(teacher_narrations):
-            teacher_narrations = teacher_narrations[0]
-        else:
-            teacher_narrations = row['Hadith Count']
-        student_narrations = student_info['Number of Narrations'].to_list()
+
+    for node, pos in isnad_pos.items():
+        node_info = narrator_bios[narrator_bios['Rawi ID'] == int(node)]
+        student_narrations = node_info['Number of Narrations'].to_list()
         if len(student_narrations):
             student_narrations = student_narrations[0]
         else:
-            student_narrations = row['Hadith Count']
-
-        teacher_gen = teacher_info['Generation'].to_list()
-        if len(teacher_gen):
-            teacher_gen = teacher_gen[0]
-        else:
-            teacher_gen = -1
-        student_gen = student_info['Generation'].to_list()
+            student_narrations = 1
+        student_gen = node_info['Generation'].to_list()
         if len(student_gen):
             student_gen = student_gen[0]
         else:
             student_gen = -1
-        teacher_rank = teacher_info["Narrator Rank"].to_list()
-        if len(teacher_rank):
-            teacher_rank = teacher_rank[0]
-        else:
-            teacher_rank = 'فلان'
-            
-        student_rank = student_info["Narrator Rank"].to_list()
+        student_rank = node_info["Narrator Rank"].to_list()
         if len(student_rank):
             student_rank = student_rank[0]
         else:
             student_rank = 'فلان'
-        if row['Source'] == '99999':
-            net.add_node(source, font = {'size':50, 'color': 'Black'}, color = '#000000', label = f'{row["Teacher"]}')
-        elif source in end_nodes:
-            end_matn_info = df[df["End Transmitter ID"] == source]
-            net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{row["Teacher"]} \n {teacher_rank} \n ID: {row["Source"]} - Gen {teacher_gen} \n Hadith {" ".join(end_matn_info["Hadith Number"].tolist())}')    
+        node_name = node_info['Famous Name'].to_list()
+        if len(node_name):
+            node_name = node_name[0]
         else:
-            net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{row["Teacher"]} \n {teacher_rank} \n ID: {row["Source"]} - Gen {teacher_gen}')
-        if target in end_nodes:
-            end_matn_info = df[df["End Transmitter ID"] == target]
-            net.add_node(target, font = {'size': 30, 'color': 'red'}, color = value_to_hex(student_narrations), label = f'{row["Student"]} \n{student_rank} \n ID: {row["Destination"]} - Gen {student_gen} \n Hadith {" ".join(end_matn_info["Hadith Number"].tolist())}')
+            node_name = 'فلان'
+        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 end_nodes:
+            end_matn_info = df[df["End Transmitter ID"] == source]
+            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 Hadith {" ".join(end_matn_info["Hadith Number"].tolist())}', x= pos[0]*x_stretch, y= -1*pos[1]*y_stretch, size= 50)
         else:
-            net.add_node(target, font = {'size': 30, 'color': 'red'}, color = value_to_hex(student_narrations), label = f'{row["Student"]} \n{student_rank} \n ID: {row["Destination"]} - Gen {student_gen}')
+            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 _, row in isnad_hadith.iterrows():
+        source = row['Source']
+        target = row['Destination']
         net.add_edge(source, target, color = value_to_hex(int(row[f'{yaxis} Count'])), label = f"{row[f'{yaxis} Count']}")
-    net.barnes_hut(gravity=-5000, central_gravity=0.3, spring_length=200)
+    net.toggle_physics(False)    
     html = net.generate_html()
     html = html.replace("'", "\"")
     return f"""<iframe style="width: 100%; height: 600px;margin:0 auto" name="result" allow="midi; geolocation; microphone; camera; 
@@ -154,6 +141,66 @@ def visualize_isnad(taraf_num, yaxis):
   allow-scripts allow-same-origin allow-popups 
   allow-top-navigation-by-user-activation allow-downloads" allowfullscreen="" 
   allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>""" , df
+    
+    # for _, row in isnad_hadith.iterrows():
+    #     source = row['Source']
+    #     target = row['Destination']
+    #     teacher_info = narrator_bios[narrator_bios['Rawi ID'] == int(row['Source'])]
+    #     student_info = narrator_bios[narrator_bios['Rawi ID'] == int(row['Destination'])]
+        
+        # teacher_narrations = teacher_info['Number of Narrations'].to_list()
+        # if len(teacher_narrations):
+        #     teacher_narrations = teacher_narrations[0]
+        # else:
+        #     teacher_narrations = row['Hadith Count']
+        # student_narrations = student_info['Number of Narrations'].to_list()
+        # if len(student_narrations):
+        #     student_narrations = student_narrations[0]
+        # else:
+        #     student_narrations = row['Hadith Count']
+
+        # teacher_gen = teacher_info['Generation'].to_list()
+        # if len(teacher_gen):
+        #     teacher_gen = teacher_gen[0]
+        # else:
+        #     teacher_gen = -1
+        # student_gen = student_info['Generation'].to_list()
+        # if len(student_gen):
+        #     student_gen = student_gen[0]
+        # else:
+        #     student_gen = -1
+        # teacher_rank = teacher_info["Narrator Rank"].to_list()
+        # if len(teacher_rank):
+        #     teacher_rank = teacher_rank[0]
+        # else:
+        #     teacher_rank = 'فلان'
+            
+        # student_rank = student_info["Narrator Rank"].to_list()
+  #       if len(student_rank):
+  #           student_rank = student_rank[0]
+  #       else:
+  #           student_rank = 'فلان'
+  #       if row['Source'] == '99999':
+  #           net.add_node(source, font = {'size':50, 'color': 'Black'}, color = '#000000', label = f'{row["Teacher"]}')
+  #       elif source in end_nodes:
+  #           end_matn_info = df[df["End Transmitter ID"] == source]
+  #           net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{row["Teacher"]} \n {teacher_rank} \n ID: {row["Source"]} - Gen {teacher_gen} \n Hadith {" ".join(end_matn_info["Hadith Number"].tolist())}')    
+  #       else:
+  #           net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{row["Teacher"]} \n {teacher_rank} \n ID: {row["Source"]} - Gen {teacher_gen}')
+  #       if target in end_nodes:
+  #           end_matn_info = df[df["End Transmitter ID"] == target]
+  #           net.add_node(target, font = {'size': 30, 'color': 'red'}, color = value_to_hex(student_narrations), label = f'{row["Student"]} \n{student_rank} \n ID: {row["Destination"]} - Gen {student_gen} \n Hadith {" ".join(end_matn_info["Hadith Number"].tolist())}')
+  #       else:
+  #           net.add_node(target, font = {'size': 30, 'color': 'red'}, color = value_to_hex(student_narrations), label = f'{row["Student"]} \n{student_rank} \n ID: {row["Destination"]} - Gen {student_gen}')
+  #       net.add_edge(source, target, color = value_to_hex(int(row[f'{yaxis} Count'])), label = f"{row[f'{yaxis} Count']}")
+  #   net.barnes_hut(gravity=-5000, central_gravity=0.3, spring_length=200)
+  #   html = net.generate_html()
+  #   html = html.replace("'", "\"")
+  #   return f"""<iframe style="width: 100%; height: 600px;margin:0 auto" name="result" allow="midi; geolocation; microphone; camera; 
+  # 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>""" , df
 
 def taraf_booknum(taraf_num):
     taraf = matn_info[matn_info['taraf_ID'] == taraf_num]