Update app.py

app.py

@@ -33,7 +33,11 @@ def splitIsnad(dataframe):
 
     
 def network_visualizer(city, year, num_nodes):
-    edge_15 = splitIsnad(subsetEdges(city, year))[['Teacher', 'Student', 'Hadiths']].groupby(['Teacher', 'Student']).sum().reset_index().sample(num_nodes)
+    edges = splitIsnad(subsetEdges(city, year))[['Teacher', 'Student', 'Hadiths']].groupby(['Teacher', 'Student']).sum().reset_index()
+    if edges.shape[0] > = num_nodes:
+        edge_15 = edge_15.sample(num_nodes)
+    else:
+        edge_15 = edge.copy()
     teacher_hadiths = edge_15[['Teacher', 'Hadiths']].groupby('Teacher').sum().reset_index()
     net = Network()
     for _, row in edge_15.iterrows():