gradio





app.py

@@ -1,15 +1,16 @@
 import gradio as gr
+import os
 
 # Categories
 compostable = ["vegetable", "fruit", "food", "leaves"]
 recyclable = ["plastic", "paper", "glass", "metal", "bottle", "can"]
 harmful = ["syringe", "battery", "medical", "chemical"]
 
-# Bin images
+# Bin images stored in 'bins/' folder
 bin_images = {
-    "Compostable": "stock-vector-bin-recycle-plastic-green-wheelie-bin-for-waste-isolated-on-white-background-green-bin-with-1306559440.jpg",
-    "Recyclable": "blue-recycling-bin-with-recycle-symbol-for-waste-management-and-sustainability-png.png",
-    "Harmful": "medical-waste-bin.jpeg"
+    "Compostable": "bins/stock-vector-bin-recycle-plastic-green-wheelie-bin-for-waste-isolated-on-white-background-green-bin-with-1306559440.jpg",
+    "Recyclable": "bins/blue-recycling-bin-with-recycle-symbol-for-waste-management-and-sustainability-png.png",
+    "Harmful": "bins/medical-waste-bin.jpeg"
 }
 
 def classify_waste(item):
@@ -21,14 +22,14 @@ def classify_waste(item):
     elif item in harmful:
         return "⚠️ Harmful/Non-Decomposable Waste", bin_images["Harmful"]
     else:
-        # Anything unknown defaults to Compostable
+        # Unknown items treated as compostable
         return "✅ Compostable Waste", bin_images["Compostable"]
 
 # Gradio UI
 iface = gr.Interface(
     fn=classify_waste,
     inputs="text",
-    outputs=["text", "image"],
+    outputs=["text", gr.Image(type="auto")],  # Make sure output type is an image
     title="♻️ Eco-Friendly Waste Classifier",
     description="Enter a waste item (e.g., vegetable, plastic, syringe) and see which bin it belongs to!"
 )
@@ -36,3 +37,4 @@ iface = gr.Interface(
 if __name__ == "__main__":
     iface.launch()
 
+