bug with numpy reshaping

app.py

@@ -221,7 +221,10 @@ def visualize_subTaraf(taraf_num, hadith_str, yaxis):
     sub_taraf = taraf[taraf['Index'].isin(hadiths)]
 
     taraf_hadith = sub_taraf['bookid_hadithid'].to_list()
-    isnad_hadith = isnad_info.iloc[np.array([hadith_lookup[i] for i in taraf_hadith]).reshape(-1)][['Source', 'Destination']]
+    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[['Source', 'Destination']] = isnad_hadith[['Source', 'Destination']].astype(int)
     
     # Merge isnad_hadith with narrator_bios for Teacher and Student
@@ -266,7 +269,10 @@ def visualize_hadith_isnad(df, yaxis):
     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 = df['bookid_hadithid'].to_list()
-    isnad_hadith = isnad_info.iloc[np.array([hadith_lookup[i] for i in taraf_hadith]).reshape(-1)][['Source', 'Destination']]
+    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[['Source', 'Destination']] = isnad_hadith[['Source', 'Destination']].astype(int)
     
     # Merge isnad_hadith with narrator_bios for Teacher and Student
@@ -332,8 +338,10 @@ def visualize_narrator_taraf(taraf_num, narrator, yaxis):
                 end_node.setdefault(node, []).append(str(idx))
                 
     # Update the graph
-    hadith_cleaned = isnad_info.iloc[np.array([hadith_lookup[i] for i in matns_with_narrator]).reshape(-1)][['Source', 'Destination']]
-    isnad_hadith = isnad_hadith[hadith_cleaned]
+    list_of_lists = [hadith_lookup[i] for i in matns_with_narrator]
+    flattened = list(set([elem for sublist in list_of_lists for elem in sublist]))
+
+    isnad_hadith = isnad_info.iloc[flattened][['Source', 'Destination']]
     G = nx.from_pandas_edgelist(isnad_hadith, source = 'Source', target = 'Destination', create_using = nx.DiGraph())
     isnad_pos = nx.nx_agraph.graphviz_layout(G, prog='dot')