Classifier model change

app.py

@@ -52,7 +52,6 @@ def classify_image(image):
     all_prompts = []
     prompt_to_category = {}
 
-    # Build prompt list
     for category, prompts in CLIP_LABELS.items():
         for p in prompts:
             all_prompts.append(p)
@@ -155,121 +154,178 @@ def explain_recycling(class_label):
 
 
 # ------------------ PIPELINE ------------------
-waste_analyzation = {
-    "cardboard": {
-        "Recycling type": "Paper/Cardboard recycling",
-        "Disposal": "Flatten the cardboard boxes and remove any non-paper packing materials before placing them in the recycling bin.",
-        "Tips": "Keep cardboard dry and clean to ensure it can be recycled efficiently.",
-        "Extra": "Avoid using waxed or heavily coated cardboard, as it may not be recyclable."
+waste_analyzation_v1 = {
+    "paper/cardboard": {
+        "Recycling type": "Paper & Cardboard recycling",
+        "Disposal": "Flatten boxes and place clean paper or cardboard into the paper recycling bin.",
+        "Tips": "Remove tape, staples, and plastic covers before recycling.",
+        "Extra": "Wet or food-stained paper should not be recycled."
+    },
+    "plastic": {
+        "Recycling type": "Plastic recycling",
+        "Disposal": "Rinse plastic containers and place accepted plastics into the recycling bin.",
+        "Tips": "Check the recycling code on the plastic item.",
+        "Extra": "Soft plastics may require special drop-off locations."
     },
     "glass": {
         "Recycling type": "Glass recycling",
-        "Disposal": "Rinse glass bottles and jars, remove lids, and place them in a dedicated glass recycling container.",
-        "Tips": "Avoid breaking the glass; broken pieces can be hazardous and may complicate processing.",
-        "Extra": "Colored glass is often recycled separately; check local recycling rules."
+        "Disposal": "Rinse glass bottles and jars and place them in the glass recycling container.",
+        "Tips": "Remove lids and caps before disposal.",
+        "Extra": "Broken glass may not be accepted in curbside programs."
     },
     "metal": {
         "Recycling type": "Metal recycling",
-        "Disposal": "Rinse aluminum cans and steel containers, remove labels if possible, and put them in the metal recycling bin.",
-        "Tips": "Crush cans to save space and facilitate transport.",
-        "Extra": "Avoid contaminating with food waste to improve recycling efficiency."
+        "Disposal": "Clean aluminum cans and metal containers before recycling.",
+        "Tips": "Crushing cans helps save space.",
+        "Extra": "Do not include batteries or electronics."
     },
-    "paper": {
-        "Recycling type": "Paper recycling",
-        "Disposal": "Sort newspapers, office paper, and magazines into the paper recycling bin; remove plastic coatings and staples.",
-        "Tips": "Shred sensitive documents but keep shredded paper in a bag to avoid scattering.",
-        "Extra": "Do not recycle wet or food-stained paper."
+    "organic waste": {
+        "Recycling type": "Organic / compost waste",
+        "Disposal": "Place food scraps and organic matter into compost or organic waste bins.",
+        "Tips": "Avoid mixing organic waste with plastics.",
+        "Extra": "Composting reduces landfill emissions."
     },
-    "plastic": {
-        "Recycling type": "Plastic recycling",
-        "Disposal": "Rinse plastic bottles, containers, and packaging, and place them in the plastic recycling bin.",
-        "Tips": "Check recycling codes; some plastics (like #3 PVC) are not widely recyclable.",
-        "Extra": "Remove caps and labels when possible to improve sorting efficiency."
+    "hazardous waste": {
+        "Recycling type": "Hazardous waste",
+        "Disposal": "Take batteries and chemicals to designated hazardous waste facilities.",
+        "Tips": "Never place hazardous items in regular trash.",
+        "Extra": "Improper disposal can cause fires or pollution."
+    },
+    "electronic waste": {
+        "Recycling type": "Electronic waste recycling",
+        "Disposal": "Drop off electronic devices at certified e-waste centers.",
+        "Tips": "Remove batteries if possible before disposal.",
+        "Extra": "Many electronics contain recyclable metals."
+    },
+    "textile": {
+        "Recycling type": "Textile recycling",
+        "Disposal": "Donate usable clothing or place textiles in textile recycling bins.",
+        "Tips": "Clean items before donation or recycling.",
+        "Extra": "Damaged textiles may be repurposed into insulation."
     },
-    "trash": {
+    "general trash": {
         "Recycling type": "General waste / landfill",
-        "Disposal": "Place non-recyclable items in the regular trash bin; avoid mixing with recyclables.",
-        "Tips": "Separate hazardous materials like batteries, electronics, or chemicals for special disposal.",
-        "Extra": "Try to reduce overall waste by reusing and composting when possible."
+        "Disposal": "Dispose of non-recyclable items in regular trash bins.",
+        "Tips": "Avoid mixing recyclables with trash.",
+        "Extra": "Reducing waste helps the environment."
     }
 }
+
 waste_analyzation_v2 = {
-    "cardboard": {
-        "Recycling type": "Paper/Cardboard recycling",
-        "Disposal": "Fold or cut cardboard into smaller flat pieces so it fits properly in recycling bins.",
-        "Tips": "Keep cardboard free from oils or food stains to avoid contaminating the batch.",
-        "Extra": "Cardboard with shiny or laminated surfaces may need separate processing."
+    "paper/cardboard": {
+        "Recycling type": "Paper recycling",
+        "Disposal": "Place dry paper and cardboard into the paper recycling container.",
+        "Tips": "Keep paper clean and dry.",
+        "Extra": "Pizza boxes with grease stains should go to trash or compost."
+    },
+    "plastic": {
+        "Recycling type": "Plastic recycling",
+        "Disposal": "Sort plastics by type and recycle accepted items.",
+        "Tips": "Remove food residue before recycling.",
+        "Extra": "Some plastics cannot be recycled curbside."
     },
     "glass": {
         "Recycling type": "Glass recycling",
-        "Disposal": "Remove lids and caps, lightly rinse, and place glass containers into the glass-only bin.",
-        "Tips": "Do not place broken drinking glasses or ceramics with recyclable glass—they melt at different temperatures.",
-        "Extra": "Some regions require taking glass to drop-off centers instead of curbside bins."
+        "Disposal": "Recycle bottles and jars after rinsing.",
+        "Tips": "Avoid mixing ceramics with glass.",
+        "Extra": "Glass can be recycled indefinitely."
     },
     "metal": {
         "Recycling type": "Metal recycling",
-        "Disposal": "Empty metal containers and place them in the metal bin; labels may stay on.",
-        "Tips": "Keep metal separate from electronics or batteries, which require special facilities.",
-        "Extra": "Large metal objects like appliances may require bulk recycling pickup."
+        "Disposal": "Recycle metal cans and containers once cleaned.",
+        "Tips": "Steel and aluminum are both recyclable.",
+        "Extra": "Large metal items may need special pickup."
     },
-    "paper": {
-        "Recycling type": "Paper recycling",
-        "Disposal": "Place clean, dry paper products such as envelopes and notebooks into paper bins.",
-        "Tips": "Do not recycle coated paper (e.g., laminated or plastic-lined pages).",
-        "Extra": "Paper with heavy ink coverage might be processed separately."
+    "organic waste": {
+        "Recycling type": "Compostable waste",
+        "Disposal": "Dispose of organic materials in compost bins.",
+        "Tips": "Cut large food scraps into smaller pieces.",
+        "Extra": "Composting improves soil quality."
     },
-    "plastic": {
-        "Recycling type": "Plastic recycling",
-        "Disposal": "Clean and drain plastic bottles or tubs, then place them in the plastics bin.",
-        "Tips": "Flatten plastic bottles to reduce volume unless your local guidelines say otherwise.",
-        "Extra": "Some plastics like polystyrene foam require special drop-off locations."
+    "hazardous waste": {
+        "Recycling type": "Hazardous materials disposal",
+        "Disposal": "Bring hazardous items to approved collection points.",
+        "Tips": "Store batteries safely before disposal.",
+        "Extra": "Never burn or crush batteries."
     },
-    "trash": {
-        "Recycling type": "General waste / landfill",
-        "Disposal": "Dispose of unrecyclable materials in regular waste; secure loose items in bags.",
-        "Tips": "Separate hazardous or toxic waste like paint, solvents, or chemicals.",
-        "Extra": "Reduce landfill impact by choosing reusable products whenever possible."
+    "electronic waste": {
+        "Recycling type": "E-waste recycling",
+        "Disposal": "Recycle electronics through authorized recyclers.",
+        "Tips": "Back up and wipe data before disposal.",
+        "Extra": "E-waste contains valuable recoverable materials."
+    },
+    "textile": {
+        "Recycling type": "Clothing & textile recycling",
+        "Disposal": "Donate or recycle unwanted clothes responsibly.",
+        "Tips": "Separate wearable and damaged items.",
+        "Extra": "Textiles can be reused or downcycled."
+    },
+    "general trash": {
+        "Recycling type": "Landfill waste",
+        "Disposal": "Throw away items that cannot be recycled.",
+        "Tips": "Reduce waste whenever possible.",
+        "Extra": "Landfills should be a last resort."
     }
 }
+
 waste_analyzation_v3 = {
-    "cardboard": {
-        "Recycling type": "Paper/Cardboard recycling",
-        "Disposal": "Remove packing tape when possible and place cardboard in dry storage until pickup day.",
-        "Tips": "Avoid leaving cardboard outdoors where rain could weaken fibers and ruin recyclability.",
-        "Extra": "Food-contaminated cardboard can often be composted instead of recycled."
+    "paper/cardboard": {
+        "Recycling type": "Paper & cardboard recycling",
+        "Disposal": "Recycle clean paper-based materials.",
+        "Tips": "Avoid recycling paper mixed with plastic.",
+        "Extra": "Paper fibers can only be recycled a few times."
+    },
+    "plastic": {
+        "Recycling type": "Plastic recycling",
+        "Disposal": "Recycle only plastics accepted locally.",
+        "Tips": "Flatten bottles to save space.",
+        "Extra": "Plastic recycling rules vary by region."
     },
     "glass": {
         "Recycling type": "Glass recycling",
-        "Disposal": "Sort by color only if required locally; rinse lightly and drop into the proper bin.",
-        "Tips": "Handle carefully to avoid breakage, as shattered glass is often not recyclable in curbside programs.",
-        "Extra": "Glass jars with metal clasps or rubber seals may need partial disassembly before recycling."
+        "Disposal": "Recycle glass containers after rinsing.",
+        "Tips": "Sort by color if required.",
+        "Extra": "Glass recycling saves raw materials."
     },
     "metal": {
         "Recycling type": "Metal recycling",
-        "Disposal": "Ensure metal food cans are clean and empty before placing in the recycling bin.",
-        "Tips": "Rinse cans briefly—no need for perfect cleaning—as this prevents pests and odors.",
-        "Extra": "Metal lids from jars should be recycled separately from the glass container."
+        "Disposal": "Recycle clean metal containers.",
+        "Tips": "Do not include pressurized containers.",
+        "Extra": "Metal recycling saves energy."
     },
-    "paper": {
-        "Recycling type": "Paper recycling",
-        "Disposal": "Place paper in the appropriate bin; keep shredded paper in a paper bag if accepted.",
-        "Tips": "Avoid mixing paper with wet waste like food scraps to maintain recyclability.",
-        "Extra": "Sticky notes and small scraps may or may not be accepted depending on the facility."
+    "organic waste": {
+        "Recycling type": "Organic waste composting",
+        "Disposal": "Place biodegradable waste into compost bins.",
+        "Tips": "Avoid adding oils or meat if not accepted.",
+        "Extra": "Organic waste can be turned into fertilizer."
     },
-    "plastic": {
-        "Recycling type": "Plastic recycling",
-        "Disposal": "Recycle only plastics accepted by your local program; rinse and drain them completely.",
-        "Tips": "Avoid recycling small plastic pieces under 2 inches—they may jam sorting machinery.",
-        "Extra": "Hard-to-recycle plastics may be collected through special community programs."
+    "hazardous waste": {
+        "Recycling type": "Hazardous waste handling",
+        "Disposal": "Dispose of hazardous items at certified facilities.",
+        "Tips": "Handle with care and keep away from heat.",
+        "Extra": "Improper disposal poses health risks."
     },
-    "trash": {
-        "Recycling type": "General waste / landfill",
-        "Disposal": "Place all non-recyclables in the trash bin; tie bags securely to avoid leakage.",
-        "Tips": "Do not place batteries, electronics, or sharp objects directly into household trash.",
-        "Extra": "Consider composting organic waste to reduce household trash volume."
+    "electronic waste": {
+        "Recycling type": "Electronic recycling",
+        "Disposal": "Recycle electronic devices properly.",
+        "Tips": "Check local e-waste programs.",
+        "Extra": "E-waste contains toxic substances."
+    },
+    "textile": {
+        "Recycling type": "Textile reuse & recycling",
+        "Disposal": "Donate or recycle textiles responsibly.",
+        "Tips": "Reuse clothing whenever possible.",
+        "Extra": "Fast fashion contributes to waste."
+    },
+    "general trash": {
+        "Recycling type": "General disposal",
+        "Disposal": "Dispose of non-recyclables in trash.",
+        "Tips": "Sort waste correctly before disposal.",
+        "Extra": "Reducing waste helps the planet."
     }
 }
-analysis = [waste_analyzation_v3, waste_analyzation_v2, waste_analyzation]
+
+analysis = [waste_analyzation_v3, waste_analyzation_v2, waste_analyzation_v1]
 
 top_label = None