Visual Similarity Search

from mixpeek import Mixpeek

client = Mixpeek(api_key="YOUR_API_KEY")

# Create collections for images and videos
image_col = client.collections.create(
    namespace_id="ns_your_namespace",
    name="image_assets",
    extractors=["multimodal-extractor"]
)

video_col = client.collections.create(
    namespace_id="ns_your_namespace",
    name="video_assets",
    extractors=["multimodal-extractor"]
)

# Build retriever spanning both collections
retriever = client.retrievers.create(
    namespace_id="ns_your_namespace",
    name="visual_similarity",
    collection_ids=["col_image_assets", "col_video_assets"],
    stages=[
        {"type": "feature_search", "top_k": 50},
        {"type": "mmr", "lambda": 0.7, "top_k": 20}
    ]
)

# Search with an image
results = client.retrievers.execute(
    retriever_id=retriever["retriever_id"],
    query={"image_url": "https://example.com/reference.jpg"}
)

for doc in results["results"]:
    print(f"{doc['document_id']} ({doc['modality']}): {doc['score']:.3f}")