Update app.py

app.py

@@ -99,7 +99,7 @@ def lookup_hadith(taraf_hadith, hadith_lookup):
         hadith_lookup (defaultdict): A defaultdict containing the hadith data.
 
     Returns:
-        list: A list of unique elements from the lookup results.
+        list: A list of the unique indices for isnad_info. so which edges for that matn
     """
     # Ensure taraf_hadith is always a list
     if isinstance(taraf_hadith, str):
@@ -350,12 +350,8 @@ def visualize_narrator_taraf(taraf_num, narrator, yaxis):
             matns_with_narrator.append(taraf_hadith[idx])
             for node in (n for n, d in G1.out_degree() if d == 0):
                 end_node.setdefault(node, []).append(str(idx))
-                
-    # Update the graph
-    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']]
+    isnad_hadith = isnad_info.iloc[lookup_hadith(matns_with_narrator, HADITH_LOOKUP)][['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')