Update app.py

app.py

@@ -120,37 +120,37 @@ 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['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; 
+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=""