Update app.py

app.py

@@ -168,6 +168,8 @@ llm1 = st.sidebar.selectbox("Model Selection", ['GPT-4', 'GPT3.5'], key='persona
 temp1 = st.sidebar.slider("Temperature", min_value=0.0, max_value=1.0, step=0.1, value=0.6, key='persona1_temp')
 tokens1 = st.sidebar.slider("Tokens", min_value=0, max_value=4000, step=100, value=500, key='persona1_tokens')
 
+k = st.sidebar.slider("Returned Docs", min_value=1, max_value=10, step=1, value=3, key='k')
+
 st.sidebar.caption(f"Session ID: {genuuid()}")
 
 # Main chat interface
@@ -217,9 +219,10 @@ else:
             llm2="GPT3.5",
             tokens2=1000,
             temperature2=0.2
+        
         )
 
-        response, retrival = call_chat_api(data)
+        response, retrival = call_chat_api(data, k)
         agent_message = response.get("content", "No response received from the agent.")
         elapsed_time = response.get("elapsed_time", 0)
         st.session_state.messages.append({"role": "assistant", "content": agent_message})