import gradio as gr
import requests
import base64
from PIL import Image
import io
import json
import os

# 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(Front)"
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=[
        ['d1.jpg'] # Example image path
    ]
)

# --- Launch the App ---
if __name__ == "__main__":
    demo.launch()