Update services/elasticsearch.py

services/elasticsearch.py

@@ -29,21 +29,22 @@ def search_elasticsearch(index, query):
 
 def search_products_by_keywords(encoded_query):
     body_query = {
-        "_source": ["ProductName", "Description", "Gender", "Price (INR)", "PrimaryColor"],
+        "_source": ['Gender', 'Category', 'SubCategory', 'ProductType',
+       'Colour', 'Usage', 'ProductTitle', 'ImageURL'],
         "knn": {
-            "field": "DescriptionVector",
+            "field": "ImageEmbedding",
             "query_vector": encoded_query,
             "k": 4,
             "num_candidates": 500
         }
     }
 
-    response = es.search(index='products', body=body_query)
+    response = es.search(index='fashion', body=body_query)
     return response['hits']['hits']
 
 
 def search_products_by_filters(filters):
-    response = es.search(index='products', body={
+    response = es.search(index='fashion', body={
         "query": {
             "bool": {
                 "filter": filters
@@ -54,12 +55,12 @@ def search_products_by_filters(filters):
 
 
 def get_product_details(product_id):
-    response = es.get(index='products', id=product_id)
+    response = es.get(index='fashion', id=product_id)
     return response['_source']
 
 
 def index_product(product: Product):
-    es.index(index="products", id=product.product_id, body=product.dict())
+    es.index(index="fashion", id=product.product_id, body=product.dict())
 
 
 def index_customer(customer: Customer):
@@ -134,4 +135,4 @@ def checkout(customer_id: str):
     }
     es.index(index="orders", body=order)
     es.delete(index="carts", id=customer_id)
-    return order
+    return order