Create app.py

app.py

@@ -0,0 +1,111 @@
+import gradio as gr
+import requests
+import base64
+import json
+from PIL import Image
+import io
+
+# ==============================================================================
+#  CONFIGURATION
+# ==============================================================================
+
+# This is the full URL to the MCP (base64) endpoint of your SERVER Space.
+# Replace this with the URL of your deployed Python API server.
+DECODER_API_ENDPOINT = "https://https://broadfield-dev-keylock-auth.hf.space/run/mcp_decode"
+# Example: "https://broadfield-dev-keylock-api-python.hf.space/run/mcp_decode"
+
+# Link to the server space for user reference
+SERVER_SPACE_URL = "https://huggingface.co/spaces/broadfield-dev/keylock-auth"
+# Example: "https://huggingface.co/spaces/broadfield-dev/KeyLock-API-Python"
+
+
+# ==============================================================================
+#  API CALL LOGIC
+# ==============================================================================
+
+def call_decoder_api(image: Image.Image) -> dict:
+    """
+    Takes a PIL image, converts it to base64, sends it to the decoder API,
+    and returns the decrypted JSON data.
+    """
+    if image is None:
+        raise gr.Error("Please upload an encrypted image.")
+
+    try:
+        # 1. Convert the PIL Image to a base64 string
+        buffer = io.BytesIO()
+        image.save(buffer, format="PNG")
+        image_bytes = buffer.getvalue()
+        image_base64 = base64.b64encode(image_bytes).decode('utf-8')
+
+        # 2. Prepare the JSON payload for the Gradio API
+        # Gradio's /run/predict endpoint expects this specific structure
+        payload = {
+            "data": [
+                image_base64
+            ]
+        }
+
+        # 3. Make the POST request to the server API
+        headers = {"Content-Type": "application/json"}
+        response = requests.post(DECODER_API_ENDPOINT, headers=headers, json=payload)
+        
+        # Raise an error for bad status codes (4xx or 5xx)
+        response.raise_for_status()
+
+        # 4. Process the response
+        response_json = response.json()
+        
+        # Gradio API wraps results in a "data" list. The actual result is the first element.
+        if "data" in response_json and len(response_json["data"]) > 0:
+            return response_json["data"][0]
+        elif "error" in response_json:
+            # Handle application-level errors returned by the server
+            raise gr.Error(f"API Error: {response_json['error']}")
+        else:
+            raise gr.Error(f"API returned an unexpected response format: {response_json}")
+
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Network error calling API: {e}")
+        raise gr.Error(f"Could not connect to the decoder API. Please check the URL and server status. Error: {e}")
+    except Exception as e:
+        logger.error(f"An unexpected error occurred: {e}")
+        raise gr.Error(f"An unexpected error occurred: {e}")
+
+# ==============================================================================
+#  GRADIO INTERFACE
+# ==============================================================================
+
+with gr.Blocks(theme=gr.themes.Soft(), title="KeyLock API Client") as demo:
+    gr.Markdown(f"""
+    # 🔑 KeyLock API Client
+    This is a user interface for the [Secure Decoder API]({SERVER_SPACE_URL}). 
+    
+    Upload an image that was encrypted with the service's public key, and this client will send it to the API for decryption.
+    """)
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            image_input = gr.Image(
+                type="pil",
+                label="Upload Encrypted PNG Image",
+                sources=["upload", "clipboard"]
+            )
+            submit_button = gr.Button("Decrypt via API", variant="primary")
+            
+            gr.Markdown("---")
+            gr.Markdown(f"This client sends requests to the following API endpoint: `{DECODER_API_ENDPOINT}`")
+
+        with gr.Column(scale=2):
+            gr.Markdown("## Decrypted Data")
+            json_output = gr.JSON(label="Result from Server")
+
+    # Connect the UI components to the API call function
+    submit_button.click(
+        fn=call_decoder_api,
+        inputs=[image_input],
+        outputs=[json_output]
+    )
+
+if __name__ == "__main__":
+    demo.launch()