added tool to view isnad of taraf that involves a narrator

app.py

@@ -304,6 +304,85 @@ def visualize_hadith_isnad(df, yaxis):
   allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>"""  
 
 
+def visualize_narrator_taraf(taraf_num, narrator, Yaxis):
+    taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
+    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())
+    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 'فلان')
+    taraf_hadith = taraf['bookid_hadithid'].to_list()
+    taraf_hadith_split = [i.split('_') for i in taraf_hadith]
+
+    G = nx.from_pandas_edgelist(isnad_hadith, source = 'Source', target = 'Destination', create_using = nx.DiGraph())
+    if narrator not in G.nodes():
+        raise gr.Error('Narrator not in Isnad of Taraf!')
+    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]
+    isnad_pos = nx.nx_agraph.graphviz_layout(G, prog='dot')
+    
+    matns_with_narrator = []
+    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]
+        G = nx.from_pandas_edgelist(isnad_hadith1, source = 'Source', target = 'Destination', create_using = nx.DiGraph())
+        if narrator in list(G.nodes):
+            matns_with_narrator.append(taraf_hadith[i])
+
+    narrator_matn_info = taraf[taraf['bookid_hadithid'].isin(matns_with_narrator)]
+    
+    x_stretch = 4
+    y_stretch = 4
+    net = Network(directed =True)
+
+    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 = 'فلان'
+            
+        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:
+            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:
+        row = isnad_hadith[(isnad_hadith['Source'] == edge[0]) & (isnad_hadith['Destination'] == edge[1])]
+        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.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>""" , narrator_matn_info[['matn', 'Book_Name', 'Author', 'Book_ID', 'Hadith Number']]
+
+
+
 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.')  
@@ -329,4 +408,10 @@ with gr.Blocks() as demo:
             col_count=(2, "fixed"))
         btn_hadith = gr.Button('Visualize')
         btn_hadith.click(fn=visualize_hadith_isnad, inputs=[hadith_selection, yyaxis], outputs=[gr.HTML()])
+    with gr.Tab('Taraf Narrator Isnad Visualizer'):
+        Yaxis = gr.Dropdown(choices = ['Taraf', 'Hadith', 'Isnad', 'Book'], value = 'Taraf', label = 'Variable to Display', info = 'Choose the variable to visualize.')  
+        taraf_number = gr.Slider(1,taraf_max , value=10000, label="Taraf", info="Choose the Taraf to Input", step = 1)
+        narr = gr.Textbox(label='Narrator', info='Choose a Narrator (Refer to full isnad from previous tab)')
+        btn_narr = gr.Button('Visualize')
+        btn_narr.click(fn=visualize_narrator_taraf, inputs=[taraf_number, narr, Yaxis], outputs=[gr.HTML(), gr.DataFrame(wrap=True)])
 demo.launch()