Create app.py

app.py

@@ -0,0 +1,76 @@
+import gradio as gr
+import cv2
+import numpy as np
+import pandas as pd
+from app_utils import annotate_planogram_compliance, do_sorting, xml_to_csv
+from inference import run
+import json
+import os
+from tempfile import NamedTemporaryFile
+
+
+# Target names list
+target_names = [
+    "Bottle,100PLUS ACTIVE 1.5L",
+    "Bottle,100PLUS ACTIVE 500ML",
+    "Bottle,100PLUS LEMON LIME 1.5L",
+    # Add all other target names here
+]
+
+# Define the function to run planogram compliance check
+def planogram_compliance_check(planogram_image, master_planogram_image, annotation_file):
+    # Convert uploaded images to numpy arrays
+    planogram_img = np.array(planogram_image)
+    master_planogram_img = np.array(master_planogram_image)
+
+    # Perform inference on planogram image
+    result_list = run(
+        weights="base_line_best_model_exp5.pt",
+        source=planogram_img,
+        imgsz=[640, 640],
+        conf_thres=0.6,
+        iou_thres=0.6,
+    )
+
+    # Load annotation file and convert to DataFrame
+    if annotation_file is not None:
+        annotation_df = xml_to_csv(annotation_file)
+        sorted_xml_df = do_sorting(annotation_df)
+    else:
+        sorted_xml_df = None
+
+    # Run planogram compliance check
+    compliance_score, annotated_image = run_compliance_check(
+        planogram_img, master_planogram_img, sorted_xml_df, result_list
+    )
+
+    return compliance_score, annotated_image
+
+# Define the function to run planogram compliance check
+def run_compliance_check(planogram_img, master_planogram_img, sorted_xml_df, result_list):
+    # Perform further processing and calculation of compliance score
+    compliance_score = 0.0  # Placeholder for actual score calculation
+    annotated_image = None  # Placeholder for annotated image
+
+    # Your compliance score calculation and image annotation logic here
+
+    return compliance_score, annotated_image
+
+# Gradio interface
+planogram_check_interface = gr.Interface(
+    fn=planogram_compliance_check,
+    inputs=[
+        gr.inputs.Image(label="Planogram Image"),
+        gr.inputs.Image(label="Master Planogram Image"),
+        gr.inputs.Dataframe(label="Annotation File (XML)")
+    ],
+    outputs=[
+        gr.outputs.Textbox(label="Compliance Score"),
+        gr.outputs.Image(label="Annotated Planogram Image"),
+    ],
+    title="Planogram Compliance Checker",
+    description="Upload planogram image, master planogram image, and annotation file (if available) to check compliance."
+)
+
+# Launch the interface
+planogram_check_interface.launch()