Update app.py

app.py

@@ -1,45 +1,37 @@
-import cv2
-import pylibdmtx.pylibdmtx as pylibdmtx
 import gradio as gr
+import cv2
+from pyzbar.pyzbar import decode
+from PIL import Image
+import numpy as np
+
+def qr_barcode_scanner(image):
+    # Convert PIL image to numpy array
+    image_data = np.array(image)
+
+    # Convert RGB to grayscale if needed
+    if len(image_data.shape) == 3 and image_data.shape[2] == 3:
+        image_data = cv2.cvtColor(image_data, cv2.COLOR_RGB2GRAY)
 
-def read_barcodes(image):
-    # Load the image
-    img = cv2.imdecode(image, cv2.IMREAD_GRAYSCALE)
+    # Decode QR codes and barcodes
+    decoded_objects = decode(image_data)
 
-    # Decode DataMatrix barcodes
-    barcodes = pylibdmtx.decode(img)
+    if len(decoded_objects) == 0:
+        return "No QR code or barcode found."
 
-    # 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 = "DataMatrix"
-
-        # Store the barcode type and data
-        results.append((barcode_type, barcode_data))
-
-    return results
-
-def process_image(image):
-    # Convert the image data to a numpy array
-    img = image.reshape((image.shape[0], image.shape[1]))
-
-    # Call the function to read barcodes from the image
-    results = read_barcodes(img)
-
-    if len(results) == 0:
-        return "No barcode found."
-    else:
-        output = ""
-        for barcode_type, barcode_data in results:
-            output += f"Type: {barcode_type}, Data: {barcode_data}\n"
-        return output.strip()
-
-# Define the input and output interfaces using Gradio
-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 Reader", 
-             description="Upload an image to extract text from DataMatrix barcodes.").launch()
+    for obj in decoded_objects:
+        data = obj.data.decode("utf-8")
+        results.append(f"Type: {obj.type}, Data: {data}")
+
+    return "\n".join(results)
+
+iface = gr.Interface(
+    fn=qr_barcode_scanner,
+    inputs=gr.inputs.Image(type="pil", label="Select an image containing QR code or barcode."),
+    outputs=gr.outputs.Textbox(),
+    live=True,
+    capture_session=True,
+    interpretation="default"
+)
+
+iface.launch(share=True)