Initial commit

src/tools.py

@@ -163,10 +163,12 @@ def get_image_by_sku(sku: str or int) -> np.array:
 			try:
 				image_name = image_data[image_data['sku'] == str(sku)]['image'].iloc[0]
 				with Image.open(f'data/images/{dir}/{image_name}') as img:
-					return np.array(img)
+					res = np.array(img)
+				return res
 			except:
 				if num == 2:
 					raise Exception
 	except:
 		with Image.open('data/service_images/' + 'no_img.jpg') as img:
-			return np.array(img)
+			res = np.array(img)
+		return res