Update app.py

app.py

@@ -53,9 +53,9 @@ cmap = plt.colormaps['cool']
 
 books = load_dataset('FDSRashid/Hadith_info', data_files='Books.csv', token = Secret_token)['train'].to_pandas()
 
-matn_info['Book ID'] = matn_info['bookid_hadithid'].apply(lambda x: int(x.split('_')[0]))
+matn_info['Book_ID'] = matn_info['bookid_hadithid'].apply(lambda x: int(x.split('_')[0]))
 matn_info['Hadith Number'] = matn_info['bookid_hadithid'].apply(lambda x: int(x.split('_')[1]))
-matn_info = matn_info.join(books, on='Book ID')
+pd.merge(matn_info, books, on='Book_ID')
 
 def value_to_hex(value):
     rgba_color = cmap(value)
@@ -121,10 +121,10 @@ def visualize_isnad(taraf_num, yaxis):
 
 def taraf_booknum(taraf_num):
     taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
-    return taraf[['matn', 'Book ID', 'Hadith Number', 'Book_Name', 'Author']]
+    return taraf[['matn', 'Book_ID', 'Hadith Number', 'Book_Name', 'Author']]
 
 def visualize_subTaraf(df, yaxis):
-    df['bookid_hadithid'] = df['Book ID'].astype(str) + '_' + df['Hadith Number'].astype(str)
+    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_split = [i.split('_') for i in hadith['bookid_hadithid'].to_list()]
     hadith_cleaned = isnad_info['Hadiths Cleaned'].apply(lambda x: any(i in x for i in taraf_hadith_split))
@@ -176,7 +176,7 @@ with gr.Blocks() as demo:
     with gr.Tab('Select Hadith Isnad Visualizer'):
         yyaxis = gr.Dropdown(choices = ['Taraf', 'Hadith', 'Isnad', 'Book'], value = 'Taraf', label = 'Variable to Display', info = 'Choose the variable to visualize.')
         hadith_selection =  gr.Dataframe(
-            headers=["Book ID", "Hadith Number"],
+            headers=["Book_ID", "Hadith Number"],
             datatype=["number", "number"],
             row_count=5,
             col_count=(2, "fixed"))