import gradio as gr

# Base de données simplifiée
recyclage_dict = {
    "Bouteille en plastique": ("Plastique", "Poubelle jaune"),
    "Boîte de conserve": ("Métal", "Poubelle jaune"),
    "Peau de banane": ("Déchet organique", "Compost"),
    "Papier journal": ("Papier", "Poubelle bleue"),
    "Canette": ("Aluminium", "Poubelle jaune"),
    "Assiette cassée": ("Non recyclable", "Déchet ordinaire"),
    "Bouteille en verre": ("Verre", "Benne à verre"),
}

# Fonction principale
def identifier_dechet(objet):
    if objet in recyclage_dict:
        matiere, consigne = recyclage_dict[objet]
        return f"🧾 Matière : {matiere}\n🗑️ À jeter dans : {consigne}"
    else:
        return "Objet non reconnu. Essayez avec un autre."

# Interface Gradio
gr.Interface(
    fn=identifier_dechet,
    inputs=gr.Dropdown(choices=list(recyclage_dict.keys()), label="Choisissez un objet à recycler"),
    outputs="text",
    title="♻️ RecycleApp - Démo",
    description="Découvrez comment recycler vos déchets du quotidien. Sélectionnez un objet pour voir où le jeter !"
).launch()