Update app.py

app.py

@@ -358,20 +358,6 @@ if __name__ == '__main__':
     )
     st.title('Multipurpose AI Agent')
     #st.markdown("<h1 style='position: fixed; top: 0; left: 0; width: 100%; padding: 10px; text-align: left; color: black;'>Multipurpose AI Agent</h1>", unsafe_allow_html=True)
-
-    if "qa_prompt" not in st.session_state:
-        st.session_state.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, reply with 'no_answer'. Use three sentences maximum and keep the answer concise.
-            
-            Question: {query} 
-              
-            Context: {context} 
-              
-            Answer:"""
-        )
-
-    if "tooltip" not in st.session_state:
-        st.session_state.tooltip = 'The AI answer your questions only considering the documents provided'
     
     client, collection_name, template, model, llm, dense_model, sparse_model = load_models_and_documents()
 
@@ -482,6 +468,14 @@ if __name__ == '__main__':
 
         with st.sidebar:
             st.divider()
+            
+            if 'documents_only' not in st.session_state:
+                st.session_state.documents_only = True
+                on_change_documents_only()
+                
+            if "tooltip" not in st.session_state:
+                st.session_state.tooltip = 'The AI answer your questions only considering the documents provided'
+                
             tooltip = 'The AI answer your questions only considering the documents provided'
             
             st.toggle(