Update app.py

app.py

@@ -78,7 +78,7 @@ def network_narrator(narrator_id, fst_year, lst_year, 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>""", edges_prepped[['Teacher', 'Student']]
 
 def narrator_retriever(name):
         return narrator_bios[(narrator_bios['Official Name'].str.contains(name)) | (narrator_bios['Famous Name'].str.contains(name)) | (narrator_bios['Rawi ID'].astype(str) == name)][['Rawi ID', 'Title Name', 'Official Name', 'Famous Name', 'Number of Narrations', 'Narrator Rank', 'Generation' ]]
@@ -101,7 +101,7 @@ with gr.Blocks() as demo:
         image_button = gr.Button("Visualize!")
 
     text_button.click(narrator_retriever, inputs=text_input, outputs=text_output)
-    image_button.click(network_narrator, inputs=[image_input, FirstYear, Last_Year, Yaxis], outputs=image_output)
+    image_button.click(network_narrator, inputs=[image_input, FirstYear, Last_Year, Yaxis], outputs=[image_output, gr.DataFrame()])
 
 demo.launch()