Update app.py

app.py

@@ -61,7 +61,29 @@ def value_to_hex(value):
     return "#{:02X}{:02X}{:02X}".format(int(rgba_color[0] * 255), int(rgba_color[1] * 255), int(rgba_color[2] * 255))
 
 
-
+def get_node_info(node):
+    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 = 1
+    student_gen = node_info['Generation'].to_list()
+    if len(student_gen):
+        student_gen = student_gen[0]
+    else:
+        student_gen = -1
+    student_rank = node_info["Narrator Rank"].to_list()
+    if len(student_rank):
+        student_rank = student_rank[0]
+    else:
+        student_rank = 'فلان'
+    node_name = node_info['Famous Name'].to_list()
+    if len(node_name):
+        node_name = node_name[0]
+    else:
+        node_name = 'فلان'
+    return node_info,student_narrations,student_gen, student_rank, node_name
     
 
 def visualize_isnad(taraf_num, yaxis):
@@ -109,27 +131,7 @@ def visualize_isnad(taraf_num, yaxis):
     net = Network(directed =True, select_menu=True, cdn_resources='remote')
 
     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 = 1
-        student_gen = node_info['Generation'].to_list()
-        if len(student_gen):
-            student_gen = student_gen[0]
-        else:
-            student_gen = -1
-        student_rank = node_info["Narrator Rank"].to_list()
-        if len(student_rank):
-            student_rank = student_rank[0]
-        else:
-            student_rank = 'فلان'
-        node_name = node_info['Famous Name'].to_list()
-        if len(node_name):
-            node_name = node_name[0]
-        else:
-            node_name = 'فلان'
+        node_info,student_narrations,student_gen, student_rank, node_name = get_node_info(node)
         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:
@@ -152,7 +154,6 @@ def visualize_isnad(taraf_num, yaxis):
   allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>""" , df.drop('Hdth Num', axis=1)
     
 
-
 def visualize_subTaraf(taraf_num, hadith_str, yaxis):
     hadith_list = hadith_str.split(',')
     hadith_list = [hadith.strip() for hadith in hadith_list]
@@ -205,27 +206,7 @@ def visualize_subTaraf(taraf_num, hadith_str, yaxis):
     net = Network(directed =True, select_menu=True, cdn_resources='remote')
 
     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 = 1
-        student_gen = node_info['Generation'].to_list()
-        if len(student_gen):
-            student_gen = student_gen[0]
-        else:
-            student_gen = -1
-        student_rank = node_info["Narrator Rank"].to_list()
-        if len(student_rank):
-            student_rank = student_rank[0]
-        else:
-            student_rank = 'فلان'
-        node_name = node_info['Famous Name'].to_list()
-        if len(node_name):
-            node_name = node_name[0]
-        else:
-            node_name = 'فلان'
+        node_info,student_narrations,student_gen, student_rank, node_name = get_node_info(node)
         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)   
         else:
@@ -267,27 +248,7 @@ def visualize_hadith_isnad(df, yaxis):
     net = Network(directed =True, select_menu=True, cdn_resources='remote')
 
     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 = 1
-        student_gen = node_info['Generation'].to_list()
-        if len(student_gen):
-            student_gen = student_gen[0]
-        else:
-            student_gen = -1
-        student_rank = node_info["Narrator Rank"].to_list()
-        if len(student_rank):
-            student_rank = student_rank[0]
-        else:
-            student_rank = 'فلان'
-        node_name = node_info['Famous Name'].to_list()
-        if len(node_name):
-            node_name = node_name[0]
-        else:
-            node_name = 'فلان'
+        node_info,student_narrations,student_gen, student_rank, node_name = get_node_info(node)
         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)   
         else:
@@ -325,11 +286,6 @@ def visualize_narrator_taraf(taraf_num, narrator, yaxis):
     # nodes = list(G.nodes)
     # nodes = [node for node in nodes if node not in list(nx.ancestors(G, narrator)) + list(nx.descendants(G, narrator)) and node !=narrator]
     # [G.remove_nodes_from(nodes)]
-    
-
-
-
-
     matns_with_narrator = []
     end_node = {}
     for i in range(len(taraf_hadith_split)):
@@ -358,32 +314,7 @@ def visualize_narrator_taraf(taraf_num, narrator, yaxis):
     net = Network(directed =True, select_menu=True, cdn_resources='remote')
 
     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 = 1
-            
-        student_gen = node_info['Generation'].to_list()
-        if len(student_gen):
-            student_gen = student_gen[0]
-        else:
-            student_gen = -1
-            
-        student_rank = node_info["Narrator Rank"].to_list()
-        if len(student_rank):
-            student_rank = student_rank[0]
-        else:
-            student_rank = 'فلان'
-            
-        node_name = node_info['Famous Name'].to_list()
-        if len(node_name):
-            node_name = node_name[0]
-        else:
-            node_name = 'فلان'
-            
+        node_info,student_narrations,student_gen, student_rank, node_name = get_node_info(node)
         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()):
@@ -405,7 +336,6 @@ def visualize_narrator_taraf(taraf_num, narrator, yaxis):
   allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>""" , narrator_matn_info[['matn', 'Book_Name', 'Author', 'Book_ID', 'Hadith Number', 'Index', 'Subset Index']]
 
 
-
 with gr.Blocks() as demo:
     with gr.Tab("Whole Taraf Visualizer"):
         Yaxis = gr.Dropdown(choices = ['Taraf', 'Hadith', 'Isnad', 'Book'], value = 'Taraf', label = 'Variable to Display', info = 'Choose the variable to visualize.')