Update app.py

app.py

@@ -241,7 +241,7 @@ def visualize_subTaraf(taraf_num, hadith_str, yaxis):
   display-capture; encrypted-media;" sandbox="allow-modals allow-forms 
   allow-scripts allow-same-origin allow-popups 
   allow-top-navigation-by-user-activation allow-downloads" allowfullscreen="" 
-  allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>"""
+  allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>""", sub_taraf[['matn', 'Book_Name', 'Author', 'Book_ID', 'Hadith Number']]
 
 def taraf_booknum(taraf_num):
     taraf = matn_info[matn_info['taraf_ID'] == taraf_num]
@@ -399,7 +399,7 @@ with gr.Blocks() as demo:
         Yaxis = gr.Dropdown(choices = ['Taraf', 'Hadith', 'Isnad', 'Book'], value = 'Taraf', label = 'Variable to Display', info = 'Choose the variable to visualize.') 
         hadith_str = gr.Textbox(label='Hadith Selection', info='Choose which range of Hadith you would like visualized from the Taraf (eg "1, 2, 4-7")')
         btn_sub = gr.Button('Visualize')
-        btn_sub.click(fn=visualize_subTaraf, inputs = [taraf_num, hadith_str, Yaxis], outputs=[gr.HTML()])
+        btn_sub.click(fn=visualize_subTaraf, inputs = [taraf_num, hadith_str, Yaxis], outputs=[gr.HTML(), gr.DataFrame(wrap=True)])
     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(