Update app.py

app.py

@@ -61,37 +61,7 @@ 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 visualize_subTaraf(taraf_num, hadith_str, yaxis):
-    hadith_list = hadith_str.split(',')
-    hadith_list = [hadith.strip() for hadith in hadith_list]
-    hadiths = np.array([], dtype=int)
-    for hadith in hadith_list:
-        if '-' in hadith:
-            if hadith.count('-') > 1:
-                #print('Please use only one Dash mark!')
-                raise gr.Error('Please use only one Dash mark!')
-            hadith_multi = hadith.strip().split('-')
-            if any([not had.isnumeric() for had in hadith_multi]):
-                #print('Invalid Begining')
-                raise gr.Error('Invalid Begining')
-            elif len(hadith_multi) != 2:
-                #print('Two numbers for a range of Hadith numbers please!')
-                raise gr.Error('Two numbers for a range of Hadith numbers please!')
-            hadith_multi = [int(had) for had in hadith_multi]
-            hadiths = np.append(hadiths, np.arange(hadith_multi[0], hadith_multi[1]))
-        elif hadith.isnumeric():
-            hadiths = np.append(hadiths, int(hadith))
-        else:
-            #print('Invalid Data format!')
-            raise gr.Error("Invalid Data format!")
-            
-    taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
-    num_hadith = taraf.shape[0]
-    taraf['Index'] = np.arange(num_hadith)
-    sub_taraf = taraf[taraf['Index'].isin(hadiths)]
-    hadith_cleaned = isnad_info['Tarafs Cleaned'].apply(lambda x: taraf_num in x)
-    isnad_hadith = isnad_info[hadith_cleaned]
-    return hadiths
+
     
 
 def visualize_isnad(taraf_num, yaxis):
@@ -182,6 +152,96 @@ 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]
+    hadiths = np.array([], dtype=int)
+    for hadith in hadith_list:
+        if '-' in hadith:
+            if hadith.count('-') > 1:
+                #print('Please use only one Dash mark!')
+                raise gr.Error('Please use only one Dash mark!')
+            hadith_multi = hadith.strip().split('-')
+            if any([not had.isnumeric() for had in hadith_multi]):
+                #print('Invalid Begining')
+                raise gr.Error('Invalid Begining')
+            elif len(hadith_multi) != 2:
+                #print('Two numbers for a range of Hadith numbers please!')
+                raise gr.Error('Two numbers for a range of Hadith numbers please!')
+            hadith_multi = [int(had) for had in hadith_multi]
+            hadiths = np.append(hadiths, np.arange(hadith_multi[0], hadith_multi[1]))
+        elif hadith.isnumeric():
+            hadiths = np.append(hadiths, int(hadith))
+        else:
+            #print('Invalid Data format!')
+            raise gr.Error("Invalid Data format!")
+            
+    taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
+    num_hadith = taraf.shape[0]
+    if np.max(hadiths) > num_hadith:
+        raise gr.Error(f'Hadith index outside of range. Total Number of Hadith in this Taraf: {num_hadith}')
+    taraf['Index'] = np.arange(num_hadith)
+    sub_taraf = taraf[taraf['Index'].isin(hadiths)]
+
+    taraf_hadith = sub_taraf['bookid_hadithid'].to_list()
+    taraf_hadith_split = [i.split('_') for i in taraf_hadith]
+
+    hadith_cleaned = isnad_info['Tarafs Cleaned'].apply(lambda x: taraf_num in x)
+    isnad_hadith = isnad_info[hadith_cleaned]
+    subtaraf_filter = isnad_info['Hadiths Cleaned'].apply(lambda x: any(i in x for i in taraf_hadith_split))
+    isnad_hadith = isnad_hadith[subtaraf_filter]
+    
+    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 'فلان')
+
+    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 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 _, 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.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>"""
+
 def taraf_booknum(taraf_num):
     taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
     return taraf[['matn', 'Book_ID', 'Hadith Number', 'Book_Name', 'Author']]
@@ -227,9 +287,6 @@ def visualize_hadith_isnad(df, yaxis):
             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(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():
@@ -256,6 +313,12 @@ with gr.Blocks() as demo:
         taraf_num = gr.Slider(1,taraf_max , value=10000, label="Taraf", info="Choose the Taraf to Input", step = 1)
         btn_num = gr.Button('Retrieve')
         btn_num.click(fn=taraf_booknum, inputs = [taraf_num], outputs= [gr.DataFrame(wrap=True)])
+    with gr.Tab('Sub Taraf Visualizer'):
+        taraf_num = gr.Slider(1,taraf_max , value=10000, label="Taraf", info="Choose the Taraf to Input", step = 1)
+        Yaxis = gr.Dropdown(choices = ['Taraf', 'Hadith', 'Isnad', 'Book'], value = 'Taraf', label = 'Variable to Display', info = 'Choose the variable to visualize.') 
+        hadith_str = gr.Textbox(label='Hadith Selection', info='Choose which range of Hadith you would like visualized from the Taraf')
+        btn_sub = gr.Button('Visualize')
+        btn_sub.click(fn=visualize_subTaraf, inputs = [taraf_num, hadith_str, Yaxis], outputs=[gr.HTML()])
     with gr.Tab('Select Hadith Isnad Visualizer'):
         yyaxis = gr.Dropdown(choices = ['Taraf', 'Hadith', 'Isnad', 'Book'], value = 'Taraf', label = 'Variable to Display', info = 'Choose the variable to visualize.')
         hadith_selection =  gr.Dataframe(