import json
from pydantic import BaseModel, Field
from APIs.aichat import generate
import shared.config as config

class DishSafetyResult(BaseModel):
    visible_objects: list[str] = Field(
        description="List EVERY distinct item visible in the frame (e.g., ['ceramic bowl', 'rice', 'metal spoon handle', 'stew'])."
    )
    is_safe: bool = Field(
        description="Is the microwave dish safe to microwave? True if safe, False if unsafe."
    )
    detected_hazards: list[str] = Field(
        description="List ONLY physical hazard items found. Empty list [] if safe."
    )
    warning_message: str = Field(
        description="Warning statement if unsafe, otherwise empty string ''."
    )

def check_dish_safety(image_path: str) -> dict:
    prompt = (
        "You are an expert microwave safety quality inspector analyzing a top-down camera frame.\n\n"
        "INSPECTION STEPS:\n"
        "Describe what you see in the image, identify any potential hazards, and determine if the dish is safe to microwave.\n\n"
        "Alert only if the cooking of the dish will cause damage to the microwave or the dish itself."
    )

    response_raw = generate(
        prompt=prompt,
        images=[image_path],
        output_format=DishSafetyResult,
        should_think=False,
    )
    print(f"[Debug] Raw response from AI generator: {response_raw}")

    if isinstance(response_raw, str):
        try:
            validated_result = DishSafetyResult.model_validate_json(response_raw)
            return validated_result.model_dump()
        except Exception:
            return json.loads(response_raw)

    if isinstance(response_raw, DishSafetyResult):
        return response_raw.model_dump()

    if isinstance(response_raw, dict):
        return response_raw

    raise ValueError(f"Unexpected response type from AI generator: {type(response_raw)}")