Create app.py

app.py

@@ -0,0 +1,84 @@
+import gradio as gr
+import requests
+import base64
+from PIL import Image
+import io
+import json
+
+# The target URL for the API endpoint
+API_URL = os.environ.get('PXiVision')
+
+def process_image(input_image: Image.Image):
+    """
+    Takes a PIL Image, converts it to base64, sends it to an API,
+    and returns the JSON response.
+
+    Args:
+        input_image: A PIL Image object from the Gradio input.
+
+    Returns:
+        A dictionary (JSON) from the API response or an error message.
+    """
+    if input_image is None:
+        return {"error": "Please upload an image first."}
+
+    try:
+        # Convert PIL Image to bytes
+        with io.BytesIO() as byte_arr:
+            input_image.save(byte_arr, format='JPEG') # You can change format if needed (e.g., PNG)
+            image_bytes = byte_arr.getvalue()
+
+        # Encode the bytes to a base64 string
+        base64_image = base64.b64encode(image_bytes).decode("utf-8")
+
+        # Prepare the request payload
+        payload = {
+            "image": base64_image
+        }
+
+        # Send the POST request to the API
+        response = requests.post(API_URL, json=payload, timeout=60) # Added a 60-second timeout
+
+        # Check if the request was successful
+        response.raise_for_status()  # This will raise an exception for HTTP errors (4xx, 5xx)
+
+        # Return the JSON response from the API
+        return response.json()
+
+    except requests.exceptions.RequestException as e:
+        # Handle network-related errors
+        return {"error": f"API request failed: {e}"}
+    except Exception as e:
+        # Handle other potential errors (e.g., image processing)
+        return {"error": f"An unexpected error occurred: {e}"}
+
+# --- Create the Gradio Interface ---
+
+# Define the input and output components for the interface
+image_input = gr.Image(type="pil", label="Upload National ID Image")
+json_output = gr.JSON(label="Extracted Information")
+
+# Create a descriptive title and description for the app
+title = "Egyptian National ID Extractor"
+description = """
+Upload a clear image of an Egyptian National ID card.
+The application will send the image to a processing service and display the extracted information,
+such as name, address, and National ID number.
+"""
+
+# Assemble the interface
+demo = gr.Interface(
+    fn=process_image,
+    inputs=image_input,
+    outputs=json_output,
+    title=title,
+    description=description,
+    allow_flagging="never",
+    examples=[
+        ['RyEJpZSdJPm4YU2QLH4khP5bzpbrZHC0Vj0oigis.jpg'] # Example image path
+    ]
+)
+
+# --- Launch the App ---
+if __name__ == "__main__":
+    demo.launch()