Update app.py

app.py

@@ -1,36 +1,65 @@
 # app.py
-
 import cv2
 import zxingcpp
 import gradio as gr
 import numpy as np
+from PIL import Image
+import base64
+import io
 
-def scan_barcode(image: np.ndarray) -> dict:
-    """Scan barcode from image array"""
-    # Convert RGB to BGR for OpenCV
-    bgr_image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
-    
-    # Scan barcodes
-    results = zxingcpp.read_barcodes(bgr_image)
-    
-    if results:
-        # Return first barcode found
-        return {
-            "status": "success",
-            "text": results[0].text,
-            "format": str(results[0].format)
-        }
-    return {"status": "error", "message": "No barcode detected"}
+def scan_barcode(image):
+    """Scan barcode from image"""
+    try:
+        # Handle different input types
+        if isinstance(image, str):
+            # If it's a base64 string, decode it
+            if image.startswith('data:image'):
+                # Remove the data:image/jpeg;base64, prefix
+                image_data = image.split(',')[1]
+                image_bytes = base64.b64decode(image_data)
+                pil_image = Image.open(io.BytesIO(image_bytes))
+                image = np.array(pil_image)
+            else:
+                # Direct base64 without prefix
+                image_bytes = base64.b64decode(image)
+                pil_image = Image.open(io.BytesIO(image_bytes))
+                image = np.array(pil_image)
+        
+        # Ensure image is numpy array
+        if not isinstance(image, np.ndarray):
+            return {"status": "error", "message": "Invalid image format"}
+        
+        # Convert RGB to BGR for OpenCV if needed
+        if len(image.shape) == 3 and image.shape[2] == 3:
+            bgr_image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
+        else:
+            bgr_image = image
+        
+        # Scan barcodes
+        results = zxingcpp.read_barcodes(bgr_image)
+        
+        if results:
+            # Return first barcode found
+            return {
+                "status": "success",
+                "text": results[0].text,
+                "format": str(results[0].format)
+            }
+        
+        return {"status": "error", "message": "No barcode detected"}
+        
+    except Exception as e:
+        return {"status": "error", "message": f"Processing error: {str(e)}"}
 
 # Create Gradio interface
 iface = gr.Interface(
     fn=scan_barcode,
-    api_name="scan",  # <-- ADD THIS LINE
     inputs=gr.Image(label="Barcode Image", type="numpy"),
     outputs=gr.JSON(label="Scan Result"),
     title="📦 Barcode Scanner API",
     description="Upload an image to scan barcode",
-    allow_flagging="never"
+    flagging_mode="never",  # Updated parameter
+    api_name="scan"
 )
 
 # Launch the application
@@ -38,5 +67,6 @@ if __name__ == "__main__":
     iface.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        show_error=True
+        show_error=True,
+        share=False  # Set to True if you want public URL
     )