Update app.py

app.py

@@ -34,7 +34,7 @@ def rag_pipeline(query: str, k_retrieved: int = 10, k_reranked: int = 5):
     query_with_context = (
         f"Context: {documents['chunk'].to_list()[:k_reranked]}\n\nQuery: {query}"
     )
-    return generate_response_api(query_with_context), documents
+    return generate_response_api(query_with_context).content, documents
 
 
 with gr.Blocks() as demo:
@@ -70,7 +70,7 @@ with gr.Blocks() as demo:
     )
 
     submit_btn.click(
-        fn=rag_interface,
+        fn=rag_pipeline,
         inputs=[query_input, retrieve_slider, rerank_slider],
         outputs=[response_output, documents_output],
     )