Create ui.py

server/ui.py

@@ -0,0 +1,56 @@
+import gradio as gr
+from .server import (
+    get_public_key,
+    get_server_info,
+    decode_data,
+    generate_rsa_keys,
+    PUBLIC_KEY_PEM_STRING,
+    KEYLOCK_STATUS_MESSAGE
+)
+
+theme = gr.themes.Soft()
+
+with gr.Blocks(title="Secure Decoder API", theme=theme) as demo:
+    gr.Markdown("# 🔐 Secure KeyLock Decoder & Auth API")
+    gr.Markdown("This application provides secure API endpoints to decrypt data from images and perform mock authentication.")
+
+    with gr.Tabs():
+        with gr.TabItem("🚀 API Documentation"):
+            gr.Markdown("## How to Use This API")
+            gr.Markdown(
+                "This server exposes three main API endpoints for programmatic use:\n"
+                "1.  **`/keylock-info`**: A `GET` request returns a JSON object with server metadata.\n"
+                "2.  **`/keylock-pub`**: A `GET` request returns the server's public RSA key required for encryption.\n"
+                "3.  **`/keylock-server`**: A `POST` request with a base64-encoded image string decrypts the data and uses it to authenticate."
+            )
+            gr.Markdown("### Required Payload for Authentication")
+            gr.Markdown("Your client must encrypt a JSON object containing `API_KEY` and `USER` keys.")
+            gr.Code(
+                language="json",
+                label="Example JSON Payload to Encrypt",
+                value='{\n  "API_KEY": "sk-12345-abcde",\n  "USER": "demo-user"\n}'
+            )
+            gr.Markdown("### Server's Public Key")
+            gr.Code(value=PUBLIC_KEY_PEM_STRING, language="pem", label="Server Public Key (from /keylock-pub)")
+
+        with gr.TabItem("⚙️ Server Status & Admin"):
+            gr.Markdown("## Server Status")
+            gr.Textbox(label="Private Key Status", value=KEYLOCK_STATUS_MESSAGE, interactive=False, lines=3)
+            gr.Markdown("---")
+            with gr.Accordion("🔑 Admin: Key Pair Generator", open=False):
+                gr.Markdown(
+                    "**For Administrators Only.** Use this tool to generate a new RSA key pair for the server. "
+                    "**This does NOT automatically apply the keys.** To use them, you must:\n"
+                    "1.  Copy the **Private Key** and update the `KEYLOCK_PRIV_KEY` secret in your deployment environment.\n"
+                    "2.  Restart the server for the changes to take effect. The public key will be derived automatically."
+                )
+                gen_keys_button = gr.Button("⚙️ Generate New 2048-bit Key Pair", variant="secondary")
+                with gr.Row():
+                    output_private_key = gr.Textbox(lines=10, label="Generated Private Key (KEEP THIS SECRET)", interactive=False, show_copy_button=True)
+                    output_public_key = gr.Textbox(lines=10, label="Generated Public Key (will be auto-derived)", interactive=False, show_copy_button=True)
+                gen_keys_button.click(fn=generate_rsa_keys, inputs=None, outputs=[output_private_key, output_public_key])
+
+    with gr.Row(visible=False):
+        gr.Interface(fn=get_public_key, inputs=None, outputs=gr.Textbox(), api_name="keylock-pub")
+        gr.Interface(fn=get_server_info, inputs=None, outputs=gr.JSON(), api_name="keylock-info")
+        gr.Interface(fn=decode_data, inputs=gr.Textbox(), outputs=gr.JSON(), api_name="keylock-server")