gradio

app.py

@@ -1,46 +1,24 @@
-def classify_item(image, description):
-    categories = ["Recyclable", "Compostable", "Trash", "Harmful"]
+import gradio as gr
 
-    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 "vegetables" in desc or "veg" in desc or 
-            "veg waste" 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 "cell" in desc:
-            category = "Harmful"
-
-        # Recyclable
-        elif "plastic" in desc or "bottle" in desc or "can" in desc or "metal" in desc:
-            category = "Recyclable"
-        elif "paper" in desc and "greasy" not in desc:
-            category = "Recyclable"
-
-        # Trash
-        elif "pizza box" in desc or "styrofoam" in desc or "chip bag" in desc:
-            category = "Trash"
-
-        # Fallback
-        else:
-            category = random.choice(categories)
-
-    elif image:
-        # Placeholder – replace with ML image model later
-        category = random.choice(categories)
+def classify_waste(waste_item):
+    compostable = ["fruit", "vegetable", "vegetables", "food", "paper", "leaves"]
+    harmful = ["syringe", "battery", "chemical", "plastic"]
 
+    if waste_item.lower() in compostable:
+        return "✅ Compostable Waste"
+    elif waste_item.lower() in harmful:
+        return "⚠️ Harmful & Non-Compostable"
     else:
-        return "No input provided", "⚠️ Please upload an image or type a description."
+        return "♻️ General Waste"
+
+iface = gr.Interface(
+    fn=classify_waste,
+    inputs=gr.Textbox(label="Enter Waste Item"),
+    outputs=gr.Label(label="Waste Category"),
+    title="Waste Classification App",
+    description="Type a waste item (e.g., 'fruit', 'vegetable', 'battery', 'syringe') and get its category."
+)
 
-    # Eco tips
-    tips = {
-        "Recyclable": "♻️ Rinse before recycling. Check local rules for plastics.",
-        "Compostable": "🌱 Add to compost bin or green waste collection.",
-        "Trash": "🗑️ Not recyclable. Consider reusable alternatives.",
-        "Harmful": "⚠️ Special disposal needed. Take syringes, needles, and batteries to hazardous waste collection centers."
-    }
+if __name__ == "__main__":
+    iface.launch()
 
-    return category, tips.get(category, "Check local disposal guidelines.")