Update app.py

app.py

@@ -120,7 +120,7 @@ def taraf_booknum(taraf_num):
     taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
     return taraf[['matn', 'Book ID', 'Hadith Number']]
 
-def visualize_subTaraf(df):
+def visualize_subTaraf(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_split = [i.split('_') for i in hadith['bookid_hadithid'].to_list()]
@@ -168,11 +168,12 @@ with gr.Blocks() as demo:
         btn_num = gr.Button('Retrieve')
         btn_num.click(fn=taraf_booknum, inputs = [taraf_num], outputs= [gr.DataFrame()])
     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"],
             datatype=["number", "number"],
             row_count=5,
             col_count=(2, "fixed"))
         btn_hadith = gr.Button('Search')
-        btn_hadith.click(fn=visualize_subTaraf, inputs=[hadith_selection], outputs=[gr.HTML()])
+        btn_hadith.click(fn=visualize_subTaraf, inputs=[hadith_selection, yyaxis], outputs=[gr.HTML()])
 demo.launch()