Update app.py

app.py

@@ -13,16 +13,12 @@ def get_price_tool(arg1:str)-> float: #it's import to specify the return type
     """This is a tool which look on a dataset and filter for products and their various atteibutes like prices description
     Args:
         arg1: the category of product
-    """
-    #filter_dataset = set(dataset['train']['Category'])
-    print(type(dataset), type(dataset['Product Name']))
-    
-    responses = dataset[dataset['Product Name'].str.contains(arg1)]
-    print(responses)
+    """    
+    arg1 = arg1.lower()
+    responses =  dataset.filter(lambda example: arg1 in example['Product Name'].lower())
     #filter_dataset_min = filter_dataset['Selling Price'].min()
     #filter_dataset_max = filter_dataset['Selling Price'].min()
     return responses
-    #return (filter_dataset_min + filter_dataset_max )/2
 
 
 #Agent Example