Update app.py

app.py

@@ -381,7 +381,6 @@ def get_retriever() -> None:
     Upload document(s), create a vector store, prepare a retriever tool,
     save the tool to the variable st.session_state.retriever_tool
     """
-
     st.write("")
     st.write("**Query Document(s)**")
     uploaded_files = st.file_uploader(
@@ -392,33 +391,44 @@ def get_retriever() -> None:
         key="document_upload_" + str(st.session_state.uploader_key),
     )
 
-    if st.button(label="Create the vector store"):
-        # Create the vector store.
-        vector_store = get_vector_store(uploaded_files)
-
-        if vector_store is not None:
-            uploaded_file_names = [file.name for file in uploaded_files]
-            file_names = ", ".join(uploaded_file_names)
-            st.session_state.vector_store_message = (
-                f"Vector store for :blue[[{file_names}]] is ready!"
-            )
-            retriever = vector_store.as_retriever()
-            st.session_state.retriever_tool = create_retriever_tool(
-                retriever,
-                name="retriever",
-                description=(
-                    "Search for information about the uploaded documents. "
-                    "For any questions about the documents, you must use "
-                    "this tool!"
-                ),
-            )
+    # Only proceed if files are uploaded
+    if uploaded_files:
+        if st.button(label="Create the vector store"):
+            st.info("Creating the vector store and initializing the retriever tool...")
             
-            if st.session_state.retriever_tool:
-                st.success("Retriever tool has been successfully initialized and is ready to use.")
+            # Create the vector store.
+            vector_store = get_vector_store(uploaded_files)
+
+            if vector_store is not None:
+                uploaded_file_names = [file.name for file in uploaded_files]
+                file_names = ", ".join(uploaded_file_names)
+                st.session_state.vector_store_message = (
+                    f"Vector store for :blue[[{file_names}]] is ready!"
+                )
+                retriever = vector_store.as_retriever()
+                st.session_state.retriever_tool = create_retriever_tool(
+                    retriever,
+                    name="retriever",
+                    description=(
+                        "Search for information about the uploaded documents. "
+                        "For any questions about the documents, you must use "
+                        "this tool!"
+                    ),
+                )
+                # Ensure the retrieval tool is included in tool selection
+                st.session_state.tool_names[0].append("Retrieval")
+
+                if st.session_state.retriever_tool:
+                    st.success("Retriever tool has been successfully initialized and is ready to use.")
+                else:
+                    st.error("Failed to initialize the retriever tool. Please try uploading again.")
+                
+                st.session_state.uploader_key += 1
             else:
-                st.error("Failed to initialize the retriever tool. Please try uploading again.")
+                st.error("Could not create vector store. Please check the uploaded files.")
+    else:
+        st.info("Please upload documents to create the vector store.")
 
-            st.session_state.uploader_key += 1
 
 
 def display_text_with_equations(text: str):