Update app.py

app.py

@@ -32,27 +32,26 @@ def splitIsnad(dataframe):
   return dataframe
 
     
-def network_visualizer(city, year, num_nodes):
-    edges = splitIsnad(subsetEdges(city, year))[['Teacher', 'Student', 'Hadiths']].groupby(['Teacher', 'Student']).sum().reset_index()
+def network_visualizer(yaxis, city, year, num_nodes):
+    edges = splitIsnad(subsetEdges(city, year))[['Teacher', 'Student', yaxis]].reset_index()
+    #.groupby(['Teacher', 'Student']).sum()
     if edges.shape[0] > num_nodes:
         edge_15 = edges.sample(num_nodes)
     else:
         edge_15 = edges.copy()
-    teacher_hadiths = edge_15[['Teacher', 'Hadiths']].groupby('Teacher').sum().reset_index()
-    student_hadiths = edge_15[['Student', 'Hadiths']].groupby('Student').sum().reset_index()
+    teacher_hadiths = edge_15[['Teacher', yaxis]].groupby('Teacher').sum().reset_index()
+    student_hadiths = edge_15[['Student', yaxis]].groupby('Student').sum().reset_index()
     net = Network()
-    # Create dictionaries to store node roles and colors based on attribute values
-    node_roles = {}
-    node_colors = {}
+
 
     
     for _, row in edge_15.iterrows():
         source = row['Teacher']
         target = row['Student']
-        attribute_value = row['Hadiths']
+        attribute_value = row[yaxis]
         edge_color = value_to_hex(attribute_value)
-        hadith_count = teacher_hadiths[teacher_hadiths['Teacher'] == source]['Hadiths'].to_list()[0]
-        student_count = student_hadiths[student_hadiths['Student'] == target]['Hadiths'].to_list()[0]
+        hadith_count = teacher_hadiths[teacher_hadiths['Teacher'] == source][yaxis].to_list()[0]
+        student_count = student_hadiths[student_hadiths['Student'] == target][yaxis].to_list()[0]
         edge_color = value_to_hex(attribute_value)
         
         net.add_node(source, color=value_to_hex(hadith_count), font = {'size':30, 'color': 'orange'}, label = f"{source}\nHadiths: {hadith_count}")
@@ -73,10 +72,11 @@ def network_visualizer(city, year, num_nodes):
 
 
 with gr.Blocks() as demo:
+  Yaxis = gr.Dropdown(choices = ['Taraf', 'Hadith', 'Isnad'], value = 'Taraf', label = 'Variable to Display', info = 'Choose the variable to visualize.')  
   Places = gr.Dropdown(choices = cities, value = 'المدينه', label = 'Location')
   Last_Year = gr.Slider(min_year, max_year, value = 9, label = 'End', info = 'Choose the  year to display Narrators')
   num_narrators = gr.Slider(0, 700, value = 400, label = 'Narrators', info = 'Choose the  number of Narrators to display')  
     
   btn = gr.Button('Submit')
-  btn.click(fn = network_visualizer, inputs = [Places, Last_Year, num_narrators], outputs = gr.HTML())
+  btn.click(fn = network_visualizer, inputs = [Yaxis, Places, Last_Year, num_narrators], outputs = gr.HTML())
   demo.launch()