Update app.py

app.py

@@ -120,7 +120,10 @@ def generate_response(query, history, model, temperature, max_tokens, top_p, see
     retriever = vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 16})
     llm = ChatGroq(groq_api_key=os.environ.get("GROQ_API_KEY"), model=model)
     custom_rag_prompt = PromptTemplate.from_template(template)
-   
+
+    docs = retriever.invoke(query)
+    relevant_info = format_docs(docs)    
+    
     rag_chain = (
         {"context": retriever | format_docs, "question": RunnablePassthrough()}
         | custom_rag_prompt
@@ -130,7 +133,7 @@ def generate_response(query, history, model, temperature, max_tokens, top_p, see
 
     response = rag_chain.invoke(query)
    
-    return response
+    return response, relevant_info
 
 
 
@@ -159,15 +162,16 @@ with gr.Blocks(theme="Nymbo/Alyx_Theme") as demo:
     
     with gr.Tab("Chatbot"):
         with gr.Row():
-            with gr.Column(scale=2):
+            with gr.Column():
                 gr.ChatInterface(
                     fn=generate_response, 
                     chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
                     examples=examples_questions,
                     additional_inputs=additional_inputs,
+                    additional_outputs=[relevant_info],
                     cache_examples=False,
                 )
-            with gr.Column(scale=2):
+            with gr.Column():
                 relevant_info = gr.Textbox(
                     label="Retrieved Information",
                     interactive=False,