fixed bug with select columns

app.py

@@ -224,7 +224,7 @@ def visualize_subTaraf(taraf_num, hadith_str, yaxis):
     list_of_lists = [hadith_lookup[i] for i in taraf_hadith]
     flattened = list(set([elem for sublist in list_of_lists for elem in sublist]))
 
-    isnad_hadith = isnad_info.iloc[flattened][['Source', 'Destination']]
+    isnad_hadith = isnad_info.iloc[flattened]
     isnad_hadith[['Source', 'Destination']] = isnad_hadith[['Source', 'Destination']].astype(int)
     
     # Merge isnad_hadith with narrator_bios for Teacher and Student
@@ -272,7 +272,7 @@ def visualize_hadith_isnad(df, yaxis):
     list_of_lists = [hadith_lookup[i] for i in taraf_hadith]
     flattened = list(set([elem for sublist in list_of_lists for elem in sublist]))
 
-    isnad_hadith = isnad_info.iloc[flattened][['Source', 'Destination']]
+    isnad_hadith = isnad_info.iloc[flattened]
     isnad_hadith[['Source', 'Destination']] = isnad_hadith[['Source', 'Destination']].astype(int)
     
     # Merge isnad_hadith with narrator_bios for Teacher and Student
@@ -330,7 +330,7 @@ def visualize_narrator_taraf(taraf_num, narrator, yaxis):
 
     # Process each hadith in taraf_hadith_split
     for idx, split_hadith in enumerate(taraf_hadith):
-        isnad_hadith1 = isnad_info.iloc[hadith_lookup[taraf_hadith[i]]][['Source', 'Destination']]
+        isnad_hadith1 = isnad_info.iloc[hadith_lookup[taraf_hadith[i]]]
         G1 = nx.from_pandas_edgelist(isnad_hadith1, source='Source', target='Destination', create_using=nx.DiGraph())
         if narrator in G1.nodes:
             matns_with_narrator.append(taraf_hadith[idx])