Update app.py

app.py

@@ -74,11 +74,18 @@ def network_narrator(narrator_id, fst_year, lst_year, yaxis):
     html = net.generate_html()
     html = html.replace("'", "\"")
 
+    edge_narrator = edge_info[(edge_info['Teacher_ID'] == narrator_id) | (edge_info['Student_ID'] == narrator_id)]
+    narrator_taraf = taraf_info[taraf_info['ID'].isin(edge_narrator['Edge_ID'].to_numpy())]
+    narrator_taraf = narrator_taraf[(narrator_taraf['Year'] >= fstyear)& (narrator_taraf['Year'] <= lstyear)]
+    merge_edge = narrator_taraf.merge(edges[['Edge_ID', 'Edge_Name']], left_on = 'ID', right_on = 'Edge_ID')
+    merge_edge_group = merge_edge.groupby('Edge_Name').sum().reset_index()[['Taraf', 'Hadith', 'Isnad', 'Edge_Name']]
+    merge_group_formatted = splitIsnad(merge_edge_group)
+
     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>""", edges_prepped[['Teacher', 'Student', 'Tarafs', 'Hadiths', 'Isnads', 'Books']]
+    allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>""", edges_prepped[['Teacher', 'Student', 'Tarafs', 'Hadiths', 'Isnads', 'Books']], merge_group_formatted
 
 def narrator_retriever(name):
         return narrator_bios[(narrator_bios['Official Name'].apply(lambda x: araby.strip_diacritics(x)).str.contains(araby.strip_diacritics(name))) | (narrator_bios['Famous Name'].apply(lambda x: araby.strip_diacritics(x)).str.contains(araby.strip_diacritics(name))) | (narrator_bios['Rawi ID'].astype(str) == name)][['Rawi ID', 'Title Name', 'Official Name', 'Famous Name', 'Number of Narrations', 'Narrator Rank', 'Generation' ]]
@@ -100,7 +107,7 @@ with gr.Blocks() as demo:
             Yaxis = gr.Dropdown(choices = ['Tarafs', 'Hadiths', 'Isnads', 'Books'], value = 'Tarafs', label = 'Variable to Display', info = 'Choose the variable to visualize.')
         image_output = gr.HTML()
         image_button = gr.Button("Visualize!")
-        image_button.click(network_narrator, inputs=[image_input, FirstYear, Last_Year, Yaxis], outputs=[image_output, gr.DataFrame()])
+        image_button.click(network_narrator, inputs=[image_input, FirstYear, Last_Year, Yaxis], outputs=[image_output, gr.DataFrame(), gr.DataFrame()])