app.py

import gradio as gr
from PIL import Image
import random

# -------------------------------
# Classification logic
# -------------------------------
def classify_item(image, description):
    categories = ["Recyclable", "Compostable", "Trash", "Harmful"]

    if description:
        desc = description.lower()

        # Compostable: fruits & vegetables
        if "banana" in desc or "apple" in desc or "fruit" in desc or "vegetable" in desc or "food" in desc or "peel" in desc or "leaf" in desc:
            category = "Compostable"

        # Harmful: medical + batteries
        elif "syringe" in desc or "needle" in desc or "battery" in desc or