Upload 2 files

classifier.py

@@ -126,7 +126,7 @@ class GarbageClassifier:
                         {"type": "image", "image": processed_image},
                         {
                             "type": "text",
-                            "text": "Please classify the garbage in this image and explain your reasoning.",
+                            "text": "Please classify what you see in this image. If it shows garbage/waste items, classify them according to the garbage classification standards. If it shows people, living things, or other non-waste items, classify it as 'Unable to classify' and explain why it's not garbage.",
                         },
                     ],
                 },
@@ -174,8 +174,33 @@ class GarbageClassifier:
         # Convert response to lowercase for matching
         response_lower = response.lower()
 
-        # Look for exact category matches first
-        for category in categories:
+        # First check for "Unable to classify" indicators
+        unable_indicators = [
+            "unable to classify",
+            "cannot classify",
+            "not garbage",
+            "not waste",
+            "person",
+            "people",
+            "human",
+            "face",
+            "living",
+            "alive",
+            "animal",
+            "functioning",
+            "in use",
+            "working",
+            "furniture",
+            "appliance",
+            "electronic device",
+        ]
+
+        if any(indicator in response_lower for indicator in unable_indicators):
+            return "Unable to classify"
+
+        # Look for exact category matches (excluding "Unable to classify" since we handled it above)
+        waste_categories = [cat for cat in categories if cat != "Unable to classify"]
+        for category in waste_categories:
             if category.lower() in response_lower:
                 return category
 
@@ -231,6 +256,7 @@ class GarbageClassifier:
             if any(keyword in response_lower for keyword in keywords):
                 return category
 
+        # If no clear classification found, default to "Unable to classify"
         return "Unable to classify"
 
     def _format_response(self, classification: str, full_response: str) -> str:

knowledge_base.py

@@ -3,6 +3,8 @@ class GarbageClassificationKnowledge:
     def get_system_prompt():
         return """You are a professional garbage classification expert. You need to carefully observe the items in the picture, analyze their materials, properties and uses, and then make accurate judgments according to garbage classification standards.
 
+IMPORTANT: You should ONLY classify items that are actually garbage/waste. If the image contains people, living things, furniture, electronics in use, or other non-waste items, you should classify it as "Unable to classify" and explain that it's not garbage.
+
 Garbage classification standards:
 
 **Recyclable Waste**:
@@ -31,10 +33,19 @@ Garbage classification standards:
 - Large bones, hard shells, hard fruit pits (coconut shells, durian shells, walnut shells, corn cobs, etc.)
 - Hair, pet waste, cat litter, etc.
 
-Please observe the items in the image carefully according to the above classification standards, provide accurate garbage classification results, and briefly explain the classification reasoning. Format your response as:
+**Unable to classify**:
+- People, human faces, human body parts
+- Living animals, pets
+- Furniture, appliances, electronics in normal use
+- Buildings, landscapes, vehicles
+- Any item that is not intended to be discarded as waste
+
+Please observe the items in the image carefully according to the above classification standards. If the image shows garbage/waste items, provide accurate garbage classification results. If the image does NOT show garbage/waste (e.g., people, living things, functioning items), classify it as "Unable to classify" and explain why it's not garbage.
+
+Format your response as:
 
-**Classification**: [Category Name]
-**Reasoning**: [Brief explanation of why this item belongs to this category]"""
+**Classification**: [Category Name or "Unable to classify"]
+**Reasoning**: [Brief explanation of why this item belongs to this category, or why it cannot be classified as garbage]"""
 
     @staticmethod
     def get_categories():
@@ -43,6 +54,7 @@ Please observe the items in the image carefully according to the above classific
             "Food/Kitchen Waste",
             "Hazardous Waste",
             "Other Waste",
+            "Unable to classify",
         ]
 
     @staticmethod
@@ -52,4 +64,5 @@ Please observe the items in the image carefully according to the above classific
             "Food/Kitchen Waste": "Organic waste from food preparation and consumption",
             "Hazardous Waste": "Items containing harmful substances that require special disposal",
             "Other Waste": "Items that don't fit into other categories and go to general waste",
+            "Unable to classify": "Items that are not garbage/waste, such as people, living things, or functioning objects",
         }