Update app.py

app.py

@@ -24,11 +24,14 @@ detections = []
 # =========================== ДЕТЕКЦИЯ ПЛОДА ============================
 
 def detect_image(image_path: str, weight: float):
+    # Считываем изображение
     image = cv2.imread(image_path)
+
+    # Выполняем детекцию
     detections = model.predict(source=image_path, conf=0.5)
     result_np_image = detections[0].plot()
     result_np_image = cv2.cvtColor(result_np_image, cv2.COLOR_BGR2RGB)
-    return result_np_image
+
     detected_fruit = None
     
     for det in detections:
@@ -38,13 +41,13 @@ def detect_image(image_path: str, weight: float):
             break
 
     if not detected_fruit:
-        return image, "Фрукт не обнаружен", None, None
+        return result_np_image, "Фрукт не обнаружен", None, None
     
     fruit_name = fruits_data[detected_fruit]["name"]
     price_per_kg = fruits_data[detected_fruit]["price_per_kg"]
     total_price = round(price_per_kg * weight, 2)
     
-    return image, fruit_name, weight, total_price
+    return result_np_image, fruit_name, weight, total_price
 
 
 # =========================== ЧЕК ============================