some tweaks

app.py

@@ -26,7 +26,9 @@ def validate_api_key(api_key):
 
 # Ask for API key from user
 st.title("Colivara Demo")
-st.write("Welcome to the Colivara Demo.")
+st.write(
+    "Welcome to the Colivara Demo. To reveal the features, please enter your API key."
+)
 st.subheader("API Key")
 api_key = st.text_input("API Key", type="password")
 if not api_key or not validate_api_key(api_key):
@@ -35,12 +37,17 @@ if not api_key or not validate_api_key(api_key):
     st.stop()
 
 client = Colivara(base_url="https://api.colivara.com", api_key=api_key)
+st.divider()
+st.subheader("Features Overview")
+st.write("Each section below is independent, and you can use them separately.")
+
 
+st.divider()
 # Create Collection Section
-st.subheader("Create Collection (Optional)")
+st.subheader("Create Collection")
 collection_name = st.text_input(
     "Collection Name",
-    help="Enter the name of the collection to create, or leave empty if not creating one.",
+    help="Enter the name of the collection to create",
 )
 if st.button("Create Collection"):
     if not collection_name:
@@ -49,7 +56,7 @@ if st.button("Create Collection"):
         with st.spinner("Creating collection..."):
             client.create_collection(collection_name)
             st.success(f"Collection '{collection_name}' successfully created.")
-
+st.divider()
 # List Collections Section
 st.subheader("List Collections")
 if st.button("List Collections"):
@@ -60,7 +67,7 @@ if st.button("List Collections"):
 
 if st.session_state["collections_list"]:
     st.code(st.session_state["collections_list"])
-
+st.divider()
 # Upsert Documents Section
 st.subheader("Upsert Documents")
 uploaded_files = st.file_uploader(
@@ -103,9 +110,15 @@ if st.button("Upsert Files"):
                 st.success(
                     f"{len(new_files)} new files successfully upserted to the collection '{selected_collection}'."
                 )
-
+st.divider()
 # Search and Retrieve Documents Section
 st.subheader("Search and Retrieve Documents")
+
+# Add collection name input for search
+search_collection_name = st.text_input(
+    "Collection Name for Search", help="Enter the collection name to search in."
+)
+
 query = st.text_input(
     "Search Query", help="Enter the search query to retrieve documents."
 )
@@ -119,12 +132,14 @@ top_k = st.slider(
 )
 
 if st.button("Search"):
-    if not query:
+    if not search_collection_name:
+        st.error("Please enter the collection name for the search.")
+    elif not query:
         st.error("Please enter a search query to retrieve documents.")
     else:
         with st.spinner("Searching..."):
             results = client.search(
-                query=query, collection_name=selected_collection, top_k=top_k
+                query=query, collection_name=search_collection_name, top_k=top_k
             )
             st.session_state["search_results"] = results.results