Update app.py

app.py

@@ -34,9 +34,9 @@ def retrieve_answer(question, goals):
                    model_pipeline=pipelines, 
                    question=question, 
                    goals=goals) as q:
-        f = q.run()
+        f, c = q.run()
 
-    return f
+    return f, c
 
 
 ##################### Process #####################
@@ -68,10 +68,12 @@ def main(conf):
             question = gr.Textbox(label="Ask a Question",
                                   value=conf["defaults"]["question"])
             ask_button = gr.Button("Ask!")
-            model_output = gr.Markdown(label="Answer")
+            model_output = gr.Markdown("### Answer")
+            context_output = gr.components.Textbox(label="Retrieved Context")
+            
             ask_button.click(fn=retrieve_answer,
                              inputs=[question, goals],
-                             outputs=model_output)
+                             outputs=[model_output,context_output])
     
         demo.launch()