Update app.py

app.py

@@ -71,6 +71,8 @@ from qdrant_client.models import (
     MultiVectorConfig
 )
 
+global_state_documents_only = None
+
 class Question(BaseModel):
     answer: str
 schema = pyjson.dumps(Question.model_json_schema())
@@ -252,7 +254,8 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
             prompt = open_query_prompt(past_messages, query)
             answer = gen_text(prompt, max_tokens=300, sampling_params=SamplingParams(temperature=0.6, top_p=0.9, top_k=10))
         else:
-            if st.session_state.documents_only:
+            print(f'GLOBAL STATE : {global_state_documents_only}')
+            if global_state_documents_only:
                 prompt = idk(query)
                 answer = gen_text(prompt, max_tokens=128, sampling_params=SamplingParams(temperature=0.6, top_p=0.9, top_k=10))
                 print(f'TYPE: {type(answer)}')
@@ -552,11 +555,13 @@ def load_models_and_documents():
 
 def on_change_documents_only():
     if st.session_state.documents_only:
+        global_state_documents_only = True
         st.session_state.toggle_docs = {
             'tooltip': 'The AI answer your questions only considering the documents provided',
             'display': True
         }
     else:
+        global_state_documents_only = False
         st.session_state.toggle_docs = {
             'tooltip': """The AI answer your questions considering the documents provided, and if it doesn't found the answer in them, try to find in its own internal knowledge""",
             'display': False