mod app.py

app.py

@@ -151,7 +151,7 @@ def predict(user_input,company):
             ))
             f.write("\n")
 
-    return prediction
+    return (prediction, user_input, context_for_query)
 
 # Set-up the Gradio UI
 # Add text box and radio button to the interface
@@ -172,7 +172,11 @@ inputs = [
     gr.Radio(label="Company", choices=["Amazon", "Google", "Microsoft", "Meta", "IBM"]),
 ]
 
-output = gr.Textbox(label="Answer")
+output = [
+    gr.Textbox(label="Answer"),
+    gr.Textbox(label="query"),
+    gr.Textbox(label="context_for_query")
+]
 
 # Create the interface
 # For the inputs parameter of Interface provide [textbox,company]