Update app.py

app.py

@@ -54,7 +54,11 @@ def shot(input, category, level):
         face_embeddings = get_face_embeddings(ast.literal_eval(str(input)))
         cropped_images, product_crops = get_cropped_images(ast.literal_eval(str(input)), category)
         print(product_crops)
-        subColour, mainColour, score = get_colour(product_crops, category)    
+        if len(product_crops) == 0:
+            print("No product crops found. Using image urls instead.")
+            subColour, mainColour, score = get_colour(ast.literal_eval(str(input)), category)    
+        else:
+            subColour, mainColour, score = get_colour(product_crops, category)    
 
         # Ensure all outputs are JSON serializable
         output_dict['colors'] = {