Update app.py

app.py

@@ -32,27 +32,38 @@ def detect_and_draw_image(input_image):
     # تبدیل تصویر PIL به آرایه NumPy
     input_image_np = np.array(input_image)
 
-    # اجرای مدل روی تصویر
-    results = model.predict(source=input_image_np, conf=0.25)
+    # اجرای مدل روی تصویر با تنظیمات اطمینان و iou پایین‌تر
+    results = model.predict(source=input_image_np, conf=0.1, iou=0.4)
     
-    # دریافت تصویر با نتایج رسم‌شده
-    annotated_image = results[0].plot()
-    
-    # تبدیل تصویر به فرمت PIL برای نمایش در Gradio
-    annotated_image_pil = Image.fromarray(annotated_image)
-    
-    # شمارش اشیاء شناسایی‌شده
-    counts = {}
-    for box in results[0].boxes:
-        class_id = int(box.cls[0])
-        label_en, label_fa = class_names.get(class_id, ('unknown', 'ناشناخته'))
-        counts[label_en] = counts.get(label_en, 0) + 1
+    # بررسی وجود اشیاء شناسایی شده
+    if results and results[0].boxes is not None:
+        # دریافت تصویر با نتایج رسم‌شده
+        annotated_image = results[0].plot()
+        
+        # تبدیل تصویر به فرمت PIL برای نمایش در Gradio
+        annotated_image_pil = Image.fromarray(annotated_image)
+
+        # شمارش اشیاء شناسایی‌شده
+        counts = {}
+        for box in results[0].boxes:
+            class_id = int(box.cls[0])
+            label_en, label_fa = class_names.get(class_id, ('unknown', 'ناشناخته'))
+            counts[label_en] = counts.get(label_en, 0) + 1
 
-    df = pd.DataFrame({
-        'Label (English)': list(counts.keys()),
-        'Label (Persian)': [class_names.get(k, ('unknown', 'ناشناخته'))[1] for k in counts.keys()],
-        'Object Count': list(counts.values())
-    })
+        # ساخت DataFrame با نام‌ها و تعداد اشیاء
+        df = pd.DataFrame({
+            'Label (English)': list(counts.keys()),
+            'Label (Persian)': [class_names.get(k, ('unknown', 'ناشناخته'))[1] for k in counts.keys()],
+            'Object Count': list(counts.values())
+        })
+    else:
+        # اگر هیچ شیء شناسایی نشده باشد، بازگرداندن تصویر اصلی و DataFrame خالی
+        annotated_image_pil = input_image
+        df = pd.DataFrame({
+            'Label (English)': [],
+            'Label (Persian)': [],
+            'Object Count': []
+        })
 
     return annotated_image_pil, df