Model prompt change

app.py

@@ -87,6 +87,44 @@ def explain_recycling(class_label):
 
 # ------------------ PIPELINE ------------------
 top_label = None
+waste_analization = {
+    "cardboard": {
+        "Recycling type": "Paper/Cardboard recycling\n",
+        "Disposal": "Flatten the cardboard boxes and remove any non-paper packing materials before placing them in the recycling bin.\n",
+        "Tips": "Keep cardboard dry and clean to ensure it can be recycled efficiently.\n",
+        "Extra": "Avoid using waxed or heavily coated cardboard, as it may not be recyclable.\n"
+    },
+    "glass": {
+        "Recycling type": "Glass recycling\n",
+        "Disposal": "Rinse glass bottles and jars, remove lids, and place them in a dedicated glass recycling container.\n",
+        "Tips": "Avoid breaking the glass; broken pieces can be hazardous and may complicate processing.\n",
+        "Extra": "Colored glass is often recycled separately; check local recycling rules.\n"
+    },
+    "metal": {
+        "Recycling type": "Metal recycling\n",
+        "Disposal": "Rinse aluminum cans and steel containers, remove labels if possible, and put them in the metal recycling bin.\n",
+        "Tips": "Crush cans to save space and facilitate transport.\n",
+        "Extra": "Avoid contaminating with food waste to improve recycling efficiency.\n"
+    },
+    "paper": {
+        "Recycling type": "Paper recycling\n",
+        "Disposal": "Sort newspapers, office paper, and magazines into the paper recycling bin; remove plastic coatings and staples.\n",
+        "Tips": "Shred sensitive documents but keep shredded paper in a bag to avoid scattering.\n",
+        "Extra": "Do not recycle wet or food-stained paper.\n"
+    },
+    "plastic": {
+        "Recycling type": "Plastic recycling\n",
+        "Disposal": "Rinse plastic bottles, containers, and packaging, and place them in the plastic recycling bin.\n",
+        "Tips": "Check recycling codes; some plastics (like #3 PVC) are not widely recyclable.\n",
+        "Extra": "Remove caps and labels when possible to improve sorting efficiency.\n"
+    },
+    "trash": {
+        "Recycling type": "General waste / landfill\n",
+        "Disposal": "Place non-recyclable items in the regular trash bin; avoid mixing with recyclables.\n",
+        "Tips": "Separate hazardous materials like batteries, electronics, or chemicals for special disposal.\n",
+        "Extra": "Try to reduce overall waste by reusing and composting when possible.\n"
+    }
+}
 
 def classify_pipeline(image):
     global top_label
@@ -100,8 +138,10 @@ def analyze_pipeline():
 
     if top_label is None:
         return "Please classify an image first."
+    
     explanation = explain_recycling(top_label)
-    return explanation
+    
+    return waste_analization[top_label]
 
 
 # ------------------ GRADIO UI ------------------