Update app.py

app.py

@@ -104,14 +104,19 @@ def process_input(query, slider_value):
 
         ###########
         # exact_match = filtered_df.filter(lambda x: x['productDisplayName'] == query)[0]['productDisplayName']
-        # exact_img = filtered_df.filter(lambda x: x['productDisplayName'] == query)[0]['image']
-        exact_match = filtered_df.loc[filtered_df['productDisplayName']==query, 'productDisplayName'].iat[0]
-        print(f"Exact match: {exact_match}")
-        exact_img = filtered_df.loc[filtered_df['productDisplayName']==query, 'image'].iat[0]
-
-        if exact_match:
+        # exact_img = filtered_df.filter(lambda x: x['productDisplayName'] == query)[0]['image
+        
+        # exact_match = filtered_df.loc[filtered_df['productDisplayName']==query, 'productDisplayName'].iat[0]
+        # print(f"Exact match: {exact_match}")
+        # exact_img = filtered_df.loc[filtered_df['productDisplayName']==query, 'image'].iat[0]
+
+        if query in matches:
+            exact_img = filtered_df.loc[filtered_df['productDisplayName']==query, 'image'].iat[0]
             imgs.insert(0, exact_img)
-            matches.insert(0, exact_match)
+            matches.insert(0, query)
+        # if exact_match:
+        #     imgs.insert(0, exact_img)
+        #     matches.insert(0, exact_match)
         ##########
 
         print(f"No. of matching images: {len(imgs)}")