Update app.py

app.py

@@ -263,7 +263,7 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
         answer = json.loads(gen_text(prompt, max_tokens=300, sampling_params=SamplingParams(temperature=0)))['answer']
         answer = f"{answer}\n\n\nSource(s) :\n\n{result_metadatas}"
             
-        if not st.session_state.documents_only:
+        if global_state_documents_only == False:
             answer = f'Documents Based :\n\n{answer}'
     else:
         gen_choice = outlines.generate.choice(llm, choices=['Domain-Specific Question', 'General Question'])
@@ -275,7 +275,7 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
             answer = json.loads(gen_text(prompt, max_tokens=300, sampling_params=SamplingParams(temperature=0.6, top_p=0.9, top_k=10)))['answer']
         else:
             print(f'GLOBAL STATE : {global_state_documents_only}')
-            if global_state_documents_only:
+            if global_state_documents_only == True:
                 prompt = idk(query)
                 answer = json.loads(gen_text(prompt, max_tokens=128, sampling_params=SamplingParams(temperature=0.6, top_p=0.9, top_k=10)))['answer']
             else: