Update app.py

app.py

@@ -844,33 +844,34 @@ if __name__ == '__main__':
             label_visibility='collapsed',
             args=(st.session_state.llm, )
         ):
-            st.chat_message("user").markdown(st.session_state.local_user_input)
-            st.session_state.messages.append({"role": "user", "content": st.session_state.local_user_input})
-
-            ai_response = generate_answer(
-                st.session_state.local_user_input, 
-                st.session_state.client, 
-                st.session_state.collection_name, 
-                st.session_state.model, 
-                st.session_state.dense_model, 
-                st.session_state.sparse_model, 
-                "\n".join([f'{msg["role"]}: {msg["content"]}' for msg in st.session_state.messages]),
-                st.session_state.search_strategy,
-                st.session_state.documents_only
-            )
-
-            with st.chat_message("assistant"):
-                message_placeholder = st.empty()
-                full_response = ""
-                for chunk in re.split(r'(\s+)', ai_response):
-                    full_response += chunk + " "
-                    time.sleep(0.05)
-                    message_placeholder.write(full_response + '▌')
-                message_placeholder.write(re.sub('▌', '', full_response))
+            if prompt != ('Exact Search : ' or 'Explain Further : '):
+                st.chat_message("user").markdown(st.session_state.local_user_input)
+                st.session_state.messages.append({"role": "user", "content": st.session_state.local_user_input})
+                
+                ai_response = generate_answer(
+                    st.session_state.local_user_input, 
+                    st.session_state.client, 
+                    st.session_state.collection_name, 
+                    st.session_state.model, 
+                    st.session_state.dense_model, 
+                    st.session_state.sparse_model, 
+                    "\n".join([f'{msg["role"]}: {msg["content"]}' for msg in st.session_state.messages]),
+                    st.session_state.search_strategy,
+                    st.session_state.documents_only
+                )
+                
+                with st.chat_message("assistant"):
+                    message_placeholder = st.empty()
+                    full_response = ""
+                    for chunk in re.split(r'(\s+)', ai_response):
+                        full_response += chunk + " "
+                        time.sleep(0.05)
+                        message_placeholder.write(full_response + '▌')
+                    message_placeholder.write(re.sub('▌', '', full_response))
                                           
-            st.session_state.messages.append({"role": "assistant", "content": full_response})
+                st.session_state.messages.append({"role": "assistant", "content": full_response})
                 
-            conversations.update({st.session_state.id_chat: st.session_state.messages})
-            with open(conversations_path, 'wb') as fp:
-                packed_bytes = msgpack.packb(conversations, use_bin_type=True)
-                fp.write(packed_bytes)
+                conversations.update({st.session_state.id_chat: st.session_state.messages})
+                with open(conversations_path, 'wb') as fp:
+                    packed_bytes = msgpack.packb(conversations, use_bin_type=True)
+                    fp.write(packed_bytes)