Update app.py

app.py

@@ -8,7 +8,7 @@ def call_gradio_api(user_input):
     
     client = Client("https://traversaal-internal-rag-faiss-gpt3-5.hf.space/", hf_token=access_token)
     result = client.predict(user_input, api_name="/predict")
-    return result[0], result[1], result[2]
+    return result[0], result[1], result[2], result[3]
     
 # Interface for the Gradio app
 
@@ -16,8 +16,12 @@ new_prompt = gr.Textbox(
                 label="Augmented Prompt", placeholder="Augmented Prompt will appear here"
             )
 
-chat_response = gr.Textbox(
-                label="Response", placeholder="Response of the prompt will appear here"
+chat_response_1 = gr.Textbox(
+                label="Response based on augmented prompt", placeholder="Response of the prompt will appear here"
+            )
+
+chat_response_2 = gr.Textbox(
+                label="Response based on original prompt", placeholder="Response of the prompt will appear here"
             )
 
 source_list = gr.Textbox(
@@ -27,7 +31,7 @@ source_list = gr.Textbox(
 iface = gr.Interface(
     fn=call_gradio_api, 
     inputs="text",
-    outputs = [new_prompt, chat_response, source_list],
+    outputs = [new_prompt, chat_response_1, chat_response_2, source_list],
     examples=["Make me a 4-hour workshop agenda for handling conflict", "Tell me about the different skills from middle management to executive leadership", "What are some of the major debates among scholars regarding the trait versus process theories of leadership? How have these perspectives evolved over time?"],
     layout="horizontal",
     title="CCL Playground",