Update app.py

app.py

@@ -1,17 +1,20 @@
-from pyzbar import pyzbar
+import cv2
+import pylibdmtx.pylibdmtx as pylibdmtx
 import gradio as gr
-import numpy as np
 
 def read_barcodes(image):
-    # Find and decode barcodes and QR codes
-    barcodes = pyzbar.decode(image)
+    # Load the image
+    img = cv2.imdecode(image, cv2.IMREAD_GRAYSCALE)
 
-    # Process each barcode/QR code found
+    # Decode DataMatrix barcodes
+    barcodes = pylibdmtx.decode(img)
+
+    # Process each barcode found
     results = []
     for barcode in barcodes:
         # Extract the data (text) from the barcode
         barcode_data = barcode.data.decode("utf-8")
-        barcode_type = barcode.type
+        barcode_type = "DataMatrix"
 
         # Store the barcode type and data
         results.append((barcode_type, barcode_data))
@@ -20,13 +23,13 @@ def read_barcodes(image):
 
 def process_image(image):
     # Convert the image data to a numpy array
-    img = np.array(image, dtype=np.uint8)
+    img = image.reshape((image.shape[0], image.shape[1]))
 
-    # Call the function to read barcodes/QR codes from the image
+    # Call the function to read barcodes from the image
     results = read_barcodes(img)
 
     if len(results) == 0:
-        return "No barcode or QR code found."
+        return "No barcode found."
     else:
         output = ""
         for barcode_type, barcode_data in results:
@@ -38,5 +41,5 @@ inputs = gr.inputs.Image(label="Input Image")
 output = gr.outputs.Textbox(label="Extracted Text")
 
 # Create the Gradio interface
-gr.Interface(fn=process_image, inputs=inputs, outputs=output, title="Barcode/QR Code Reader",
-             description="Upload an image to extract text from barcodes or QR codes.").launch()
+gr.Interface(fn=process_image, inputs=inputs, outputs=output, title="Barcode Reader", 
+             description="Upload an image to extract text from DataMatrix barcodes.").launch()