Update app.py

app.py

@@ -9,7 +9,6 @@ from datasets import Features
 from datasets import Value
 from datasets import Dataset
 import matplotlib.pyplot as plt
-
 import re
 
 pattern = r'"(.*?)"'
@@ -61,7 +60,6 @@ def value_to_hex(value):
     rgba_color = cmap(value)
     return "#{:02X}{:02X}{:02X}".format(int(rgba_color[0] * 255), int(rgba_color[1] * 255), int(rgba_color[2] * 255))
 
-#edge_info, matn_info, narrator_bios, isnad_info
 
 def visualize_isnad(taraf_num, yaxis):
     taraf_hadith = matn_info[matn_info['taraf_ID'] == taraf_num]['bookid_hadithid'].to_list()
@@ -86,8 +84,6 @@ def visualize_isnad(taraf_num, yaxis):
             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'])
     
-    #hadith_cleaned = isnad_info['Hadiths Cleaned'].apply(lambda x: any(i in x for i in taraf_hadith_split) )
-
     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 'فلان')
@@ -142,65 +138,6 @@ def visualize_isnad(taraf_num, yaxis):
   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]
@@ -216,59 +153,55 @@ def visualize_subTaraf(df, yaxis):
     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 'فلان')
     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"]}')
+        node_name = node_info['Famous Name'].to_list()
+        if len(node_name):
+            node_name = node_name[0]
+        else:
+            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"] == node]
+            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(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}')
-        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; 
-    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>"""     
+  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>"""  
 
 
 with gr.Blocks() as demo: