Update app.py

app.py

@@ -53,19 +53,23 @@ def shot(input, category, level):
         openai_parsed_response = get_openAI_tags(ast.literal_eval(str(input)))
         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)
-        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'] = {
-            "main": mainColour,
-            "sub": subColour,
-            "score": score
-        }
+        
+        # Commenting out the color 
+        # -------------------------
+        # 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'] = {
+        #     "main": mainColour,
+        #     "sub": subColour,
+        #     "score": score
+        # }
+        # -------------------------
+        
         output_dict['image_mapping'] = openai_parsed_response
         output_dict['face_embeddings'] = face_embeddings
         output_dict['cropped_images'] = cropped_images