Update app.py

app.py

@@ -106,7 +106,7 @@ def visualize_isnad(taraf_num, yaxis):
         else:
             net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{source} \n {teacher_info["Narrator Rank"].to_list()[0]}')
         net.add_node(target, font = {'size': 30, 'color': 'red'}, color = value_to_hex(student_narrations), label = f'{target} \n{student_info["Narrator Rank"].to_list()[0]}')
-        net.add_edge(source, target, color = value_to_hex(int(isnad['Hadith Count'].to_list()[0])), label = f"{isnad['Hadith Count'].to_list()[0]}")
+        net.add_edge(source, target, color = value_to_hex(int(isnad[f'{yaxis} Count'].to_list()[0])), label = f"{isnad[f'{yaxis} Count'].to_list()[0]}")
     net.barnes_hut(gravity=-5000, central_gravity=0.3, spring_length=200)
     html = net.generate_html()
     html = html.replace("'", "\"")
@@ -120,14 +120,46 @@ def taraf_booknum(taraf_num):
     taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
     return taraf[['matn', 'Book ID', 'Hadith Number']]
 
-# def visualize_subTaraf(df):
-#     df['bookid_hadithid']  = df.apply(lambda x : )
-    
+ def visualize_subTaraf(df):
+     df['bookid_hadithid'] = df['Book ID'].astype(str) + '_' + df['Hadith Number'].astype(str)
+     hadith = matn_info[matn_info['bookid_hadithid'].isin(df['bookid_hadithid'])]
+     taraf_hadith_split = [i.split('_') for i in hadith['bookid_hadithid'].to_list()]
+     hadith_cleaned = isnad_info['Hadiths Cleaned'].apply(lambda x: any(i in x for i in taraf_hadith_split))
+     isnad_hadith = isnad_info[hadith_cleaned][['Source', 'Destination']]
+     narrators = isnad_hadith.applymap(lambda x: narrator_bios[narrator_bios['Rawi ID'] == int(x)]['Famous Name'].to_list()).rename(columns={"Source": "Teacher", "Destination": "Student"})
+     isnad_hadith["Student"] = narrators['Student']
+     isnad_hadith["Teacher"] = narrators['Teacher']
+     filtered = isnad_hadith[(isnad_hadith['Teacher'].apply(lambda x: len(x)) == 1) & (isnad_hadith['Student'].apply(lambda x: len(x)) == 1)]
+     filtered['Student'] = filtered['Student'].apply(lambda x: x[0])
+     filtered['Teacher'] = filtered['Teacher'].apply(lambda x: x[0])
+     net = Network(directed =True)
+     for _, row in filtered.iterrows():
+         source = row['Teacher']
+         target = row['Student']
+         teacher_info = narrator_bios[narrator_bios['Rawi ID'] == int(row['Source'])]
+         student_info = narrator_bios[narrator_bios['Rawi ID'] == int(row['Destination'])]
+         isnad = isnad_info[(isnad_info['Source'] == row['Source']) & (isnad_info['Destination'] == row['Destination'])]
+         teacher_narrations = teacher_info['Number of Narrations'].to_list()[0]
+         student_narrations = student_info['Number of Narrations'].to_list()[0]
+         if row['Source'] == '99999':
+             net.add_node(source, font = {'size':50, 'color': 'Black'}, color = '#000000')
+         else:
+             net.add_node(source, font = {'size':30, 'color': 'red'}, color = value_to_hex(teacher_narrations), label = f'{source} \n {teacher_info["Narrator Rank"].to_list()[0]}')
+        net.add_node(target, font = {'size': 30, 'color': 'red'}, color = value_to_hex(student_narrations), label = f'{target} \n{student_info["Narrator Rank"].to_list()[0]}')
+        net.add_edge(source, target, color = value_to_hex(int(isnad[f'{yaxis} Count'].to_list()[0])), label = f"{isnad[f'{yaxis} Count'].to_list()[0]}")
+     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>"""     
 
 
 with gr.Blocks() as demo:
     with gr.Tab("Whole Taraf Visualizer"):
-        Yaxis = gr.Dropdown(choices = ['Tarafs', 'Hadiths', 'Isnads', 'Books'], value = 'Tarafs', label = 'Variable to Display', info = 'Choose the variable to visualize.')  
+        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)
         btn = gr.Button('Submit')
         btn.click(fn = visualize_isnad, inputs = [taraf_number, Yaxis], outputs = [gr.HTML(), gr.DataFrame()])
@@ -135,4 +167,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()])
+    with gr.Tab('Select Hadith Isnad Visualizer'):
+        hadith_selection =  gr.Dataframe(
+            headers=["Book ID", "Hadith Number"],
+            datatype=["number", "number"],
+            row_count=5,
+            col_count=(2, "fixed"))
+        btn_hadith = gr.Button('Search')
+        btn_hadith.click(fn=visualize_isnad, inputs=[hadith_selection], outputs=[gr.HTML()])
 demo.launch()