Update app.py

app.py

@@ -39,7 +39,7 @@ def detect_fruit(image, weight: float):
             break
 
     if not detected_fruit:
-        return result_np_image, "Фрукт не обнаружен", None, None
+        return result_np_image, None, None, None
     
     fruit_name = fruits_data[detected_fruit]["name"]
     price_per_kg = fruits_data[detected_fruit]["price_per_kg"]
@@ -48,47 +48,50 @@ def detect_fruit(image, weight: float):
     return result_np_image, fruit_name, weight, total_price
 
 # =========================== ЧЕК ============================
-def create_receipt(fruit_name, weight, total_price):
-    receipt_img = Image.new("RGB", (300, 200), color="white")
-    draw = ImageDraw.Draw(receipt_img)
+# def create_receipt(fruit_name, weight, total_price):
+#     receipt_img = Image.new("RGB", (300, 200), color="white")
+#     draw = ImageDraw.Draw(receipt_img)
 
-    try:
-        font = ImageFont.truetype("arial.ttf", 18)
-    except IOError:
-        font = ImageFont.load_default()
+#     try:
+#         font = ImageFont.truetype("arial.ttf", 18)
+#     except IOError:
+#         font = ImageFont.load_default()
 
-    draw.text((10, 10), "Чек", fill="black", font=font)
-    draw.text((10, 50), f"Продукт: {fruit_name}", fill="black", font=font)
-    draw.text((10, 80), f"Вес: {weight} кг", fill="black", font=font)
-    draw.text((10, 110), f"Цена за кг: {fruits_data[fruit_name]['price_per_kg']} руб.", fill="black", font=font)
-    draw.text((10, 140), f"Сумма: {total_price} руб.", fill="black", font=font)
+#     draw.text((10, 10), "Чек", fill="black", font=font)
+#     draw.text((10, 50), f"Продукт: {fruit_name}", fill="black", font=font)
+#     draw.text((10, 80), f"Вес: {weight} кг", fill="black", font=font)
+#     draw.text((10, 110), f"Цена за кг: {fruits_data[fruit_name]['price_per_kg']} руб.", fill="black", font=font)
+#     draw.text((10, 140), f"Сумма: {total_price} руб.", fill="black", font=font)
 
-    with BytesIO() as output:
-        receipt_img.save(output, format="PNG")
-        output.seek(0)
-        return output.getvalue()
+#     with BytesIO() as output:
+#         receipt_img.save(output, format="PNG")
+#         output.seek(0)
+#         return output.getvalue()
 
 # ======================= ИНТЕРФЕЙС ============================
 def gradio_interface(image, weight):
-    if weight <= 0:
-        return image, "Вес должен быть больше 0", None
+    # if weight <= 0:
+    #     return image
 
-    image, fruit_name, weight, total_price = detect_fruit(image, weight)
+    detected_image, fruit_name, weight, total_price = detect_fruit(image, weight)
     if not fruit_name:
-        return image, None  # Вернуть пустой чек
-    
-    receipt = create_receipt(fruit_name, weight, total_price)
-    return image, receipt
+        return image
+        # return image, None  # Вернуть пустой чек
+        
+    return detected_image
+    # receipt = create_receipt(fruit_name, weight, total_price)
+    # return image, receipt
 
 image_input = gr.Image(label="Изображение")
 weight_input = gr.Number(label="Вес (кг)")
 image_output = gr.Image(label="Распознанный фрукт", type="numpy")
-receipt_output = gr.Image(label="Чек", type="numpy")
+# receipt_output = gr.Image(label="Чек", type="numpy")
 
 gr.Interface(
     fn=gradio_interface,
     inputs=[image_input, weight_input],
-    outputs=[image_output, receipt_output],
+    outputs=[image_output],
+    # outputs=[image_output, receipt_output],
     title="Определение плода и создание чека",
     description="Загрузите изображение, введите вес и получите чек"
 ).launch()