Inialize Colivara Demo

.gitignore

@@ -0,0 +1 @@
+venv/

app.py

@@ -0,0 +1,136 @@
+# import libraries
+import streamlit as st
+from colivara_py import Colivara
+import base64
+from PIL import Image
+from io import BytesIO
+
+
+def validate_api_key(api_key):
+    # check if the api key is valid
+    try:
+        client = Colivara(base_url="https://api.colivara.com", api_key=api_key)
+        client.list_collections()
+        return True
+    except Exception:
+        return False
+
+
+# ask for api key from user in professional way
+st.title("Colivara Demo")
+st.write("Welcome to the Colivara Demo.")
+st.subheader("API Key")
+api_key = st.text_input("API Key", type="password")
+if not api_key or not validate_api_key(api_key):
+    st.write("Please enter your API key to proceed.")
+    st.markdown("If you don't have one, get yours [here](https://colivara.com).")
+    st.stop()
+
+client = Colivara(base_url="https://api.colivara.com", api_key=api_key)
+
+st.subheader("Create Collection (Optional)")
+collection_name = st.text_input(
+    "Collection Name",
+    help="Enter the name of the collection to create, or leave empty if not creating one.",
+)
+if st.button("Create Collection"):
+    if not collection_name:
+        st.error("Please enter a collection name to create the collection.")
+    else:
+        with st.spinner("Creating collection..."):
+            client.create_collection(collection_name)
+            st.success(f"Collection '{collection_name}' successfully created.")
+
+
+st.subheader("List Collections")
+if st.button("List Collections"):
+    with st.spinner("Fetching collections..."):
+        collections = client.list_collections()
+        collections_text = "\n".join([collection.name for collection in collections])
+        st.text_area(
+            "Existing Collections:",
+            value=collections_text,
+            height=200,
+            help="List of existing collections.",
+        )
+
+st.subheader("Upsert Documents")
+uploaded_files = st.file_uploader(
+    "Upload Docoments",
+    accept_multiple_files=True,
+    help="Upload your documents here, most extensions are supported (pdf, docx, png, jpg, xlsx, etc.)",
+)
+selected_collection = st.text_input(
+    "Collection Name", help="Enter the collection name to store the documents in."
+)
+if not uploaded_files or not selected_collection:
+    st.write("Please upload your documents and enter the collection name to proceed.")
+    st.stop()
+
+uploaded_file_names = set()
+
+if st.button("Upsert Files"):
+    new_files = [
+        file for file in uploaded_files if file.name not in uploaded_file_names
+    ]
+    if not new_files:
+        st.info("No new files to upsert. Please upload new files to proceed.")
+    else:
+        with st.spinner("Upserting Files..."):
+            progress_bar = st.progress(0)
+            total_files = len(new_files)
+            for idx, file in enumerate(new_files):
+                uploaded_file_names.add(file.name)
+                encoded_file = base64.b64encode(file.read()).decode("utf-8")
+                client.upsert_document(
+                    name=file.name,
+                    document_base64=encoded_file,
+                    collection_name=selected_collection,
+                    wait=True,
+                )
+                progress_bar.progress((idx + 1) / total_files)
+            st.success(
+                f"{len(new_files)} new files successfully upserted to the collection '{selected_collection}'."
+            )
+
+st.subheader("Search and Retrieve Documents")
+query = st.text_input(
+    "Search Query", help="Enter the search query to retrieve documents."
+)
+
+top_k = st.slider(
+    "Number of Top Documents to Retrieve",
+    min_value=1,
+    max_value=10,
+    value=3,
+    help="Select the number of top documents to retrieve.",
+)
+
+
+if st.button("Search"):
+    if 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
+            )
+            results = results.results
+
+            if not results:
+                st.info("No documents found for the given query.")
+
+            else:
+                for idx, result in enumerate(results):
+                    img_base64 = result.img_base64
+                    page_number = result.page_number
+                    document_name = result.document_name
+                    img = Image.open(BytesIO(base64.b64decode(img_base64)))
+                    st.image(
+                        img, caption=f"Document: {document_name}, Page: {page_number}"
+                    )
+                    st.success("Search completed.")
+
+
+st.markdown("----")
+st.markdown("Developed by Abdulhaleem from TJM Labs")

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+colivara-py
+Pillow