Update app.py

app.py

@@ -38,12 +38,18 @@ matn_info['taraf_ID'] = matn_info['taraf_ID'].replace('KeyAbsent', -1)
 
 
 matn_info['taraf_ID'] = matn_info['taraf_ID'].astype(int)
+matn_info = matn_info.sort_values('taraf_ID')
+tarafs = matn_info['taraf_ID'].unique()
+for i, taraf in enumerate(tarafs):
+  matn_info.loc[matn_info['taraf_ID'] == taraf, 'taraf_ID_New'] = i + 1 # Replace 'a' with 'e' in column 'C' where the condition is met
+matn_info['taraf_ID_New'] = matn_info['taraf_ID_New'].astype(int)
+
 
 
 isnad_info = load_dataset('FDSRashid/hadith_info',token = Secret_token, data_files = 'isnad_info.csv', split = 'train').to_pandas()
 isnad_info['Hadiths Cleaned'] = isnad_info['Hadiths'].apply(lambda x: [re.findall(pattern, string)[0].split("_") for string in x[1:-1].split(',')])
 
-tarafs = np.max(matn_info['taraf_ID'].unique())
+taraf_max = np.max(matn_info['taraf_ID_New'].unique())
 
 cmap = plt.colormaps['cool']
 def value_to_hex(value):
@@ -53,7 +59,7 @@ def value_to_hex(value):
 #edge_info, matn_info, narrator_bios, isnad_info
 
 def visualize_isnad(taraf_num, yaxis):
-  taraf_hadith = matn_info[matn_info['taraf_ID'] == taraf_num]['bookid_hadithid'].to_list()
+  taraf_hadith = matn_info[matn_info['taraf_ID_New'] == taraf_num]['bookid_hadithid'].to_list()
   taraf_hadith_split = [i.split('_') for i in taraf_hadith]
   hadith_cleaned = isnad_info['Hadiths Cleaned'].apply(lambda x: any(i in x for i in taraf_hadith_split) )
   isnad_hadith = isnad_info[hadith_cleaned][['Source', 'Destination']]
@@ -92,7 +98,7 @@ def visualize_isnad(taraf_num, yaxis):
 
 with gr.Blocks() as demo:
   Yaxis = gr.Dropdown(choices = ['Tarafs', 'Hadiths', 'Isnads', 'Books'], value = 'Tarafs', label = 'Variable to Display', info = 'Choose the variable to visualize.')  
-  taraf_number = gr.Slider(1,tarafs , value=10000, label="Taraf", info="Choose the Taraf to Input", step = 1)
+  taraf_number = gr.Slider(1,taraf_max , value=10000, label="Taraf", info="Choose the Taraf to Input", step = 1)
   btn = gr.Button('Submit')
   btn.click(fn = visualize_isnad, inputs = [taraf_number, Yaxis], outputs = gr.HTML())
   demo.launch()