Create app.py

app.py

@@ -0,0 +1,144 @@
+import gradio as gr
+import json
+import logging
+import tempfile
+import core  # Import our new core module
+
+# --- Configure Logging ---
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+# ==============================================================================
+#  UI HELPER FUNCTIONS
+# ==============================================================================
+
+def update_service_dropdown(json_config_string: str):
+    """Parses the JSON config and updates the dropdown choices."""
+    if not json_config_string.strip():
+        return gr.Dropdown(choices=[], value=None, label="Decoder Service (Paste JSON config)"), "Status: Waiting for JSON configuration."
+
+    try:
+        endpoints = json.loads(json_config_string)
+        if not isinstance(endpoints, list):
+            raise TypeError("JSON root must be a list of objects.")
+
+        service_names = []
+        for service in endpoints:
+            # Validate that each service has the required keys
+            if not all(k in service for k in ("name", "link", "public_key")):
+                raise ValueError("Each service object must contain 'name', 'link', and 'public_key' keys.")
+            service_names.append(service["name"])
+        
+        # If valid, update the dropdown with the new choices
+        if not service_names:
+            return gr.Dropdown(choices=[], value=None, label="Decoder Service (No services found)"), "Status: JSON parsed, but no services were found."
+            
+        logger.info(f"Updated dropdown with services: {service_names}")
+        return gr.Dropdown(choices=service_names, value=service_names[0], label="Select Decoder Service"), f"Status: Loaded {len(service_names)} services. '{service_names[0]}' selected."
+    except (json.JSONDecodeError, TypeError, ValueError) as e:
+        logger.error(f"JSON parsing error: {e}")
+        return gr.Dropdown(choices=[], value=None, label="Decoder Service (Invalid JSON)"), f"Status: Error parsing JSON - {e}"
+
+def generate_image(json_config_string: str, selected_service_name: str, secret_data_str: str):
+    """
+    The main function called by the 'Create' button.
+    It finds the selected service's public key and calls the core encryption logic.
+    """
+    if not all([json_config_string, selected_service_name, secret_data_str]):
+        raise gr.Error("Configuration, a selected service, and secret data are all required.")
+
+    try:
+        endpoints = json.loads(json_config_string)
+        
+        # Find the public key for the selected service
+        public_key = None
+        for service in endpoints:
+            if service.get("name") == selected_service_name:
+                public_key = service.get("public_key")
+                break
+        
+        if not public_key:
+            raise gr.Error(f"Could not find the service '{selected_service_name}' in the provided JSON.")
+
+        # Call the core function to do the heavy lifting
+        encrypted_image = core.create_encrypted_image(secret_data_str, public_key)
+        
+        # Save to a temporary file for download
+        with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp_file:
+            encrypted_image.save(tmp_file.name)
+            logger.info(f"Generated image saved to temporary file: {tmp_file.name}")
+            return encrypted_image, tmp_file.name, f"✅ Success! Image created for '{selected_service_name}'. You can now download it."
+
+    except Exception as e:
+        logger.error(f"Image creation failed: {e}", exc_info=True)
+        # Return empty outputs on failure
+        return None, None, f"❌ Error: {e}"
+
+# ==============================================================================
+#  GRADIO INTERFACE
+# ==============================================================================
+
+# Example JSON for the user to copy
+EXAMPLE_JSON = """
+[
+  {
+    "name": "My Python Decoder API",
+    "link": "https://huggingface.co/spaces/YOUR_USERNAME/KeyLock-API-Python",
+    "public_key": "-----BEGIN PUBLIC KEY-----\\nMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAy2sWjB1iQ3vK03U7e/9E\\nO6J1K/s0tBq4Pz8F3r9/i8s7t9R1p8Z4Y6h4f4O7w9p9Z0c8t7m4J1e9g7K9m6f3\\nR1k3y7v1w0l7z6s5v2l8l4t9v8z7y6t5k2x1c9v7z3k1y9w8r5t3s1v9a8d7g6f5\\ne4d3c2b1a9f8e7d6c5b4a3f2e1d0c9b8a7f6e5d4c3b2a1f0e9d8c7b6a5f4e3d2\\nc1b0a9f8e7d6c5b4a3f2e1d0c9b8a7f6e5d4c3b2a1f0e9d8c7b6a5f4e3d2c1b0\\na9f8e7d6c5b4a3f2e1d0c9b8a7f6e5d4c3b2a1f0e9d8c7b6a5f4e3d2c1b0a9f8\\ne7d6\\n-----END PUBLIC KEY-----"
+  }
+]
+"""
+
+with gr.Blocks(theme=gr.themes.Soft(), title="KeyLock Image Creator") as demo:
+    gr.Markdown("# 🏭 KeyLock Image Creator")
+    gr.Markdown("Create secure, encrypted images by providing a JSON configuration of decoder endpoints and your secret data.")
+
+    with gr.Row():
+        with gr.Column(scale=2):
+            gr.Markdown("### 1. Configure Decoder Services")
+            json_config = gr.Textbox(
+                lines=10,
+                label="JSON Configuration",
+                placeholder="Paste your JSON list of services here...",
+                info="Each service must have a 'name', 'link', and 'public_key'."
+            )
+            gr.Markdown("### 2. Enter Your Secret Data")
+            secret_data = gr.Textbox(
+                lines=5,
+                label="Secret Data (Key-Value Pairs)",
+                placeholder="USERNAME: user@example.com\nAPI_KEY: sk-12345..."
+            )
+            
+        with gr.Column(scale=1):
+            gr.Markdown("### 3. Select Service and Create")
+            service_dropdown = gr.Dropdown(
+                label="Decoder Service (Paste JSON config)",
+                choices=[],
+                interactive=True
+            )
+            create_button = gr.Button("Create Encrypted Image", variant="primary")
+            
+            status_output = gr.Textbox(label="Status", interactive=False)
+            image_output = gr.Image(label="Generated Image", type="pil")
+            download_output = gr.File(label="Download Image")
+
+    # --- Interactivity ---
+    # When the JSON config textbox changes, update the dropdown menu
+    json_config.change(
+        fn=update_service_dropdown,
+        inputs=[json_config],
+        outputs=[service_dropdown, status_output]
+    )
+
+    # When the create button is clicked, run the main generation function
+    create_button.click(
+        fn=generate_image,
+        inputs=[json_config, service_dropdown, secret_data],
+        outputs=[image_output, download_output, status_output]
+    )
+    
+    with gr.Accordion("Example JSON Configuration", open=False):
+        gr.Code(value=EXAMPLE_JSON, language="json")
+
+if __name__ == "__main__":
+    demo.launch()