diff --git a/cloud/safety_checker.py b/cloud/safety_checker.py index 432f6c1..2458136 100644 --- a/cloud/safety_checker.py +++ b/cloud/safety_checker.py @@ -2,7 +2,6 @@ import json from pydantic import BaseModel, Field from APIs.aichat import generate import shared.config as config -from PIL import Image, ImageEnhance class UtensilMaterialCheck(BaseModel): object_name: str = Field( @@ -29,22 +28,6 @@ class DishSafetyResult(BaseModel): description="If is_safe is False, set to 'REMOVE METAL UTENSIL OR FOIL BEFORE MICROWAVING'. Otherwise empty ''." ) -def preprocess_dish_image(image_path: str) -> str: - """Brightens dark areas and sharpens food texture for small vision models.""" - img = Image.open(image_path) - - # 1. Boost brightness slightly - enhancer = ImageEnhance.Brightness(img) - img = enhancer.enhance(1.4) - - # 2. Boost contrast to separate food shapes from dark shadows - enhancer = ImageEnhance.Contrast(img) - img = enhancer.enhance(1.3) - - processed_path = "/tmp/processed_dish.jpg" - img.save(processed_path, quality=85) - return processed_path - def check_dish_safety(image_path: str) -> dict: prompt = ( "Analyze this microwave dish image strictly for EQUIPMENT DAMAGE HAZARDS (Metal, Steel, Aluminum Foil).\n\n"