Update app.py

app.py

@@ -1,43 +1,60 @@
-# app.py
-# =============
-# This is a complete app.py file for a QR code scanner using Gradio and Hugging Face Transformers.
-
 import gradio as gr
 import cv2
 from pyzbar.pyzbar import decode
 from PIL import Image
 import numpy as np
+import base64
+import io
 
 def scan_qr_code(image):
     """
     Scan QR code from the uploaded image.
 
     Args:
-    image (PIL.Image.Image or dict): The uploaded image or a dictionary containing the image file path.
+    image: The uploaded image (can be PIL Image, numpy array, or file path)
 
     Returns:
     str: The decoded QR code data with a descriptive message.
     """
-    # Convert file path to PIL image if necessary
-    if isinstance(image, dict):
-        image = Image.open(image['name'])
+    try:
+        # Handle different input types
+        if isinstance(image, np.ndarray):
+            open_cv_image = image
+        elif isinstance(image, Image.Image):
+            open_cv_image = np.array(image)
+            open_cv_image = open_cv_image[:, :, ::-1].copy()
+        elif isinstance(image, str):
+            # Handle base64 encoded images
+            if image.startswith('data:image'):
+                image = Image.open(io.BytesIO(base64.b64decode(image.split(',')[1])))
+                open_cv_image = np.array(image)
+                open_cv_image = open_cv_image[:, :, ::-1].copy()
+            else:
+                # Handle file paths
+                try:
+                    image = Image.open(image)
+                    open_cv_image = np.array(image)
+                    open_cv_image = open_cv_image[:, :, ::-1].copy()
+                except Exception as e:
+                    return f"Error opening image: {str(e)}"
+        else:
+            return "Unsupported image format"
 
-    # Convert PIL image to OpenCV format
-    open_cv_image = np.array(image)
-    open_cv_image = open_cv_image[:, :, ::-1].copy()
+        # Decode QR code
+        decoded_objects = decode(open_cv_image)
+        if decoded_objects:
+            for obj in decoded_objects:
+                decoded_data = obj.data.decode('utf-8')
+                return f"The code contains the following data: {decoded_data}"
+        return "No QR code found"
 
-    # Decode QR code
-    decoded_objects = decode(open_cv_image)
-    if decoded_objects:
-        for obj in decoded_objects:
-            decoded_data = obj.data.decode('utf-8')
-            return f"The code contains the following data: {decoded_data}"
-    return "No QR code found"
+    except Exception as e:
+        return f"Error processing image: {str(e)}"
 
 # Define the Gradio interface
 iface = gr.Interface(
     fn=scan_qr_code,
-    inputs=gr.Image(type="filepath"),
+    inputs=gr.Image(type="numpy"),  # Changed to numpy type
     outputs=gr.Textbox(),
     title="QR Code Scanner",
     description="Upload an image containing a QR code to decode it."