Update app.py

app.py

@@ -99,7 +99,7 @@ def rrf(rank_lists, alpha=60, default_rank=1000):
     return sorted_items
 
 
-def main(query: str, client: QdrantClient, collection_name: str, llm, dense_model, sparse_model):
+def main(query: str, client: QdrantClient, collection_name: str, llm, dense_model, sparse_model, qa_prompt):
     dense_query = list(dense_model.embed_query(query, 32))
     sparse_query = list(sparse_model.embed(query, 32))
 
@@ -358,7 +358,7 @@ def chunk_documents(texts, metadatas, dense_model, sparse_model):
     
     return documents, metadatas_docs, dense_embeddings, sparse_embeddings
 
-def on_change_documents_only(qa_prompt):
+def on_change_documents_only(qa_prompt, tooltip):
     qa_prompt = lambda query, context: (
         f"""You are an assistant for question-answering tasks. Use the following pieces of retrieved context to answer the question. If you don't know the answer, just say that you don't know. Use three sentences maximum and keep the answer concise.
         
@@ -374,7 +374,9 @@ def on_change_documents_only(qa_prompt):
         {context}
 
         Question: {query}"""
-)
+    )
+    
+    tooltip = 'The AI answer your questions only considering the documents provided' if st.session_state.documents_only else """The AI answer your questions considering the documents provided, and if it doesn't found the answer in them, try to find in its on own internal knowledge"""
     
     
 if __name__ == '__main__':
@@ -398,7 +400,7 @@ if __name__ == '__main__':
         st.chat_message("user").markdown(prompt)
         st.session_state.messages.append({"role": "user", "content": prompt})
 
-        ai_response, ai_response_with_metadatas = main(prompt, client, collection_name, llm, dense_model, sparse_model, IsDocumentsOnly)
+        ai_response, ai_response_with_metadatas = main(prompt, client, collection_name, llm, dense_model, sparse_model, qa_prompt)
         with st.chat_message("assistant"):
             message_placeholder = st.empty()
             full_response = ""
@@ -466,15 +468,13 @@ if __name__ == '__main__':
         )
         st.sidebar.success("Document content uploaded and ready!")
 
+    tooltip = 'The AI answer your questions only considering the documents provided'
+
     IsDocumentsOnly = st.sidebar.toggle(
         label="""Enable 'Documents-Only' Mode""", 
         value=True, 
         on_change=on_change_documents_only, 
         key="documents_only",
-        args=(qa_prompt, )
-    )
-
-    if IsDocumentsOnly:
-        st.sidebar.write('The AI answer your questions only considering the documents provided')
-    else:
-        st.sidebar.write("""The AI answer your questions considering the documents provided, and if it doesn't found the answer in them, try to find in its on own internal knowledge""")
+        args=(qa_prompt, tooltip),
+        help=tooltip
+    )