Update app.py

app.py

@@ -25,27 +25,28 @@ def detect_fruit(image, weight: float):
 
     # Проверка на наличие детекций
     if len(detections) == 0:
-        return image, "Фрукт не обнаружен", None, None
+        return image
 
     result_np_image = detections[0].plot()
     result_np_image = cv2.cvtColor(result_np_image, cv2.COLOR_BGR2RGB)
 
-    detected_fruit = None
+    # detected_fruit = None
     
-    for det in detections.xyxy[0]:  # Предполагается, что вы используете ультралайтики и доступ к детекциям
-        label = model.names[int(det[5])]  # название фрукта
-        if label in fruits_data:
-            detected_fruit = label
-            break
+    # for det in detections.xyxy[0]:  # Предполагается, что вы используете ультралайтики и доступ к детекциям
+    #     label = model.names[int(det[5])]  # название фрукта
+    #     if label in fruits_data:
+    #         detected_fruit = label
+    #         break
 
-    if not detected_fruit:
-        return result_np_image, None, None, None
+    # if not detected_fruit:
+    #     return result_np_image, None, 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)
+    # 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 result_np_image
     
-    return result_np_image, fruit_name, weight, total_price
+    # return result_np_image, fruit_name, weight, total_price
 
 # =========================== ЧЕК ============================
 # def create_receipt(fruit_name, weight, total_price):
@@ -89,7 +90,7 @@ image_output = gr.Image(label="Распознанный фрукт", type="numpy
 
 gr.Interface(
     fn=gradio_interface,
-    inputs=[image_input, weight_input],
+    inputs=[image_input],
     outputs=[image_output],
     # outputs=[image_output, receipt_output],
     title="Определение плода и создание чека",