from marqo import Client

# Initialize the Marqo client with your API key
api_key = "your_api_key_here"
mq = Client("https://api.marqo.ai", api_key=api_key)

# Set your API key as an environment variable
export MARQO_API_KEY="your_api_key_here"

# Define your ecommerce index name
index_name = "ecommerce-product-search"

# Configure settings optimized for ecommerce
settings = {
    "treatUrlsAndPointersAsImages": True,
    "model": "Marqo/marqo-ecommerce-embeddings-L",
    "numberOfShards": 1,
    "numberOfReplicas": 0,
    "inferenceType": "marqo.GPU",
    "storageClass": "marqo.basic",
    "numberOfInferences": 1,
}

# Create the index
mq.create_index(index_name, settings_dict=settings)

curl -X POST 'https://api.marqo.ai/api/v2/indexes/ecommerce-product-search' \
-H "x-api-key: $MARQO_API_KEY" \
-H 'Content-type: application/json' \
-d '{
  "treatUrlsAndPointersAsImages": true,
  "model": "Marqo/marqo-ecommerce-embeddings-L",
  "numberOfShards": 1,
  "numberOfReplicas": 0,
  "inferenceType": "marqo.GPU",
  "storageClass": "marqo.basic",
  "numberOfInferences": 1
}'

# Sample ecommerce product data
products = [
    {
        "product_id": "SKU001",
        "title": "Premium Wireless Bluetooth Headphones",
        "description": "High-quality over-ear headphones with noise cancellation and 30-hour battery life",
        "category": "Electronics",
        "brand": "AudioTech",
        "price": 199.99,
        "image_url": "https://your-cdn.com/products/headphones-001.jpg",
        "tags": ["wireless", "bluetooth", "noise-cancelling", "premium"],
    },
    {
        "product_id": "SKU002",
        "title": "Organic Cotton T-Shirt",
        "description": "Comfortable everyday t-shirt made from 100% organic cotton in navy blue",
        "category": "Clothing",
        "brand": "EcoWear",
        "price": 29.99,
        "image_url": "https://your-cdn.com/products/tshirt-navy-002.jpg",
        "tags": ["organic", "cotton", "casual", "navy"],
    },
    {
        "product_id": "SKU003",
        "title": "Stainless Steel Water Bottle",
        "description": "Insulated water bottle keeps drinks cold for 24 hours, hot for 12 hours",
        "category": "Home & Kitchen",
        "brand": "HydroLife",
        "price": 34.99,
        "image_url": "https://your-cdn.com/products/bottle-steel-003.jpg",
        "tags": ["insulated", "stainless-steel", "eco-friendly", "travel"],
    },
]

# Index products with multimodal mappings
response = mq.index(index_name).add_documents(
    products,
    client_batch_size=50,  # Batch size for efficient indexing
    mappings={
        "product_multimodal": {
            "type": "multimodal_combination",
            "weights": {
                "title": 0.1,  # Product title importance
                "image_url": 0.9,  # Image visual features (highest weight)
            },
        }
    },
    tensor_fields=["product_multimodal"],  # Generate embeddings for this combined field
)

print(f"Indexed {len(products)} products successfully")

curl -X POST 'https://api.marqo.ai/api/v2/indexes/ecommerce-product-search/documents' \
-H "x-api-key: $MARQO_API_KEY" \
-H 'Content-type: application/json' \
-d '{
  "documents": [
    {
      "product_id": "SKU001",
      "title": "Premium Wireless Bluetooth Headphones",
      "description": "High-quality over-ear headphones with noise cancellation and 30-hour battery life",
      "category": "Electronics",
      "brand": "AudioTech", 
      "price": 199.99,
      "image_url": "https://your-cdn.com/products/headphones-001.jpg",
      "tags": ["wireless", "bluetooth", "noise-cancelling", "premium"]
    }
  ],
  "mappings": {
    "product_multimodal": {
      "type": "multimodal_combination",
      "weights": {
        "title": 0.1,
        "image_url": 0.9,
      }
    }
  },
  "tensorFields": ["product_multimodal"]
}'

# Search for products using natural language
query = "comfortable headphones for music"

search_results = mq.index(index_name).search(
    q=query,
    limit=10,  # Number of results to return
    searchable_attributes=["product_multimodal"],  # Search in multimodal field
)

# Process results
for hit in search_results["hits"]:
    print(f"Product: {hit['title']}")
    print(f"Score: {hit['_score']:.3f}")
    print(f"Price: ${hit['price']}")
    print("---")

# Advanced search with hybrid parameters and filters
search_results = mq.index(index_name).search(
    q="wireless audio device",
    searchMethod="HYBRID",  # Use hybrid search (semantic + lexical)
    hybridParameters={
        "alpha": 0.7,  # Weight between semantic (1.0) and lexical (0.0) search
        "searchableAttributes": ["product_multimodal"],
        "lexicalSearchParameters": {
            "searchableAttributes": ["title", "description", "tags"]
        },
    },
    filter="category:(Electronics) AND price:[50 TO 300]",  # Filter by category and price range
    limit=5,
    showHighlights=True,  # Show matching text snippets
)

# Display enhanced results
for hit in search_results["hits"]:
    print(f"Product: {hit['title']}")
    print(f"Brand: {hit['brand']}")
    print(f"Price: ${hit['price']}")
    print(f"Relevance Score: {hit['_score']:.3f}")

    # Show highlighted matches if available
    if "_highlights" in hit:
        for field, highlights in hit["_highlights"].items():
            print(f"Matches in {field}: {highlights}")
    print("---")

curl -X POST 'https://api.marqo.ai/api/v2/indexes/ecommerce-product-search/search' \
-H "x-api-key: $MARQO_API_KEY" \
-H 'Content-type: application/json' \
-d '{
  "q": "wireless audio device",
  "searchMethod": "HYBRID",
  "hybridParameters": {
    "alpha": 0.7,
    "searchableAttributes": ["product_multimodal"],
    "lexicalSearchParameters": {
      "searchableAttributes": ["title", "description", "tags"]
    }
  },
  "filter": "category:(Electronics) AND price:[50 TO 300]",
  "limit": 5,
  "showHighlights": true
}'

# Search using an image URL
image_query = "https://customer-uploads.com/search-image.jpg"

visual_search_results = mq.index(index_name).search(
    q=image_query,
    searchMethod="TENSOR",  # Use tensor search for image queries
    searchableAttributes=["product_multimodal"],
    limit=8,
)

print("Visually similar products:")
for hit in visual_search_results["hits"]:
    print(f"Product: {hit['title']}")
    print(f"Visual similarity: {hit['_score']:.3f}")
    print(f"Category: {hit['category']}")
    print("---")

# Add a new product
new_product = {
    "product_id": "SKU004",
    "title": "Smart Fitness Watch",
    "description": "Track your workouts, heart rate, and sleep with GPS connectivity",
    "category": "Electronics",
    "brand": "FitTech",
    "price": 299.99,
    "image_url": "https://your-cdn.com/products/smartwatch-004.jpg",
    "tags": ["smartwatch", "fitness", "GPS", "health"],
}

mq.index(index_name).add_documents([new_product])

# Update product information
updated_product = {
    "product_id": "SKU001",
    "price": 179.99,  # New sale price
    "tags": ["wireless", "bluetooth", "noise-cancelling", "premium", "sale"],
}

mq.index(index_name).add_documents([updated_product])