def imageHandler(bot, message, chat_id, local_filename):
bot.sendMessage(chat_id, "Hi, I'm processing your request")
print("Processing request...")
is_good_quality = QualityChecker.is_good_quality(
Indexer.load_image(local_filename,
im_size=config.QUALITYCHECKER_IMSIZE))
if not is_good_quality:
bot.sendMessage(
chat_id,
"Your image is of a poor quality. Please, send me a better one")
print("Message sent: image is of a poor quality.")
else:
is_shoe = ShoeDetector.classify_image(
Indexer.load_image(local_filename,
im_size=config.CLASSIFIER_IM_SIZE))
if not is_shoe:
bot.sendMessage(
chat_id,
"Ops! Something went wrong... Make sure your image contains a shoe"
)
print("Message sent: the photo doesn't contain a shoe.")
else:
try:
most_similar = Matcher.get_most_similar(
Indexer.load_image(local_filename))
retrieved_images = Matcher.retrieve_items(most_similar)
bot.sendMessage(chat_id,
"These are the most similar shoes I've found")
for im in retrieved_images:
bot.sendImage(chat_id, config.DATASET_PATH + im, "")
print("Most similar images sent.")
except FeatureExtractionException:
bot.sendMessage(
chat_id,
"I couldn't process your photo. Please, send me a better one"
)
print("Message sent: the photo can't be processed.")
print("Request processed.")
