Update app.py

app.py

@@ -2,21 +2,40 @@ import gradio as gr
 from ultralytics import YOLO
 from PIL import Image
 
-# Pretrained YOLOv8n моделін жүктеу (ең жеңіл нұсқа)
+# Pretrained YOLOv8n моделін жүктеу
 model = YOLO('yolov8n.pt')
 
-# Қоқыс жәшігін тексеру функциясы
+# Қоқыс жәшігін тексеру және дабыл шығару функциясы
 def check_trash(img):
     results = model(img)
     boxes = results[0].boxes
-    output = []
+    detected_bins = []
+    detected_trash = []
+
+    # detected object-терді тексеру
     for box in boxes:
-        label = box.cls.item()
+        label = results[0].names[int(box.cls.item())]
         conf = box.conf.item()
-        if conf > 0.5:  # сенімділік threshold
-            output.append(f"{results[0].names[int(label)]}: {conf*100:.1f}%")
-    if not output:
-        return "Қоқыс табылмады"
+        if conf < 0.5:
+            continue
+        if 'bin' in label or 'container' in label:  # жәшік
+            detected_bins.append(label)
+        else:  # басқа detected object-терді қоқыс деп қарау
+            detected_trash.append(label)
+
+    # Нәтижені шығару
+    output = []
+    if detected_bins:
+        if detected_trash:
+            output.append("Жәшік: Толы ✅")
+        else:
+            output.append("Жәшік: Бос ⚪")
+    else:
+        output.append("Жәшік табылмады ❌")
+
+    if detected_trash:
+        output.append(f"Сыртқа шыққан қоқыс табылды 🚨: {', '.join(detected_trash)}")
+
     return "\n".join(output)
 
 # Gradio интерфейсі
@@ -25,7 +44,7 @@ app = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs=gr.Textbox(),
     title="SmartCity Waste Detection",
-    description="Қоқыс жәшігін, сыртқа төгілген қоқысты және түрін анықтайды"
+    description="Қоқыс жәшігін, сыртқа төгілген қоқысты және толық/бос жәшікті анықтайды"
 )
 
 app.launch()