Create app.py

app.py

@@ -0,0 +1,74 @@
+import os; os.system("pip install git+https://github.com/broadfield-dev/gradio-keylock.git")
+import gradio as gr
+from gradio_keylock.component import KeylockDecoderComponent, AppServerLogic
+import json
+
+server_logic = AppServerLogic()
+
+example_db = {"USER":"TestUser","PASS":"TestPass"}
+
+def login_fn(username,password):
+    if username==example_db.get("USER") and password==example_db.get("PASS"):
+        return "### ✅ Logged in"
+    else:
+        return "### 🛑 Incorrect Username or Password"
+
+def process_image_and_display_payload(image):
+    if image is None:
+        return "Upload a KeyLock image to auto-fill credentials.", {}
+
+    result_dict = server_logic.decode_payload(image)
+    
+    if result_dict.get("status") == "Success":
+        user = result_dict.get("payload", {}).get("USER", "")
+        user_pass = result_dict.get("payload", {}).get("PASS", "")
+        
+        GROQ_API_KEY = result_dict.get("payload", {}).get("GROQ_API_KEY", "")
+        HF_TOKEN = result_dict.get("payload", {}).get("HF_TOKEN", "")
+        OPENAI_API_KEY = result_dict.get("payload", {}).get("OPENAI_API_KEY", "")
+        OPENROUTER_API_KEY = result_dict.get("payload", {}).get("OPENROUTER_API_KEY", "")
+        
+        status_message = f"<p style='color:green; font-weight:bold;'>✅ Success! Decoded credentials for '{user}'.</p>"
+        payload = result_dict.get("payload", {})
+    else:
+        message = result_dict.get("message", "An unknown error occurred.")
+        status_message = f"<p style='color:red; font-weight:bold;'>❌ Error: {message}</p>"
+        payload = {}
+
+    return status_message, payload, user, user_pass, GROQ_API_KEY, HF_TOKEN, OPENAI_API_KEY, OPENROUTER_API_KEY
+
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# Gradio KeyLock Component Demo")
+    gr.Markdown(
+        "**Instructions:**\n"
+        "1. Use the 'Generate Encrypted Image' section to create a new secure image.\n"
+        "2. Download the generated image or drag it to the 'KeyLock Image' upload area.\n"
+        "3. The decoded data will **automatically** appear below."
+    )
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("## Login")
+            user_name=gr.Textbox(label='User Name')
+            user_pass=gr.Textbox(label='Password', type='password')
+            login_btn=gr.Button("Login")
+            login_msg=gr.Markdown("### Enter user name and password")
+            gr.Markdown("## API Keys (Demo)")
+            user_GROQ=gr.Textbox(label='GROQ_API_KEY')
+            user_HF=gr.Textbox(label='HF_TOKEN')
+            user_OPENAI=gr.Textbox(label='OPENAI_API_KEY')
+            user_OPENROUTER=gr.Textbox(label='OPENROUTER_API_KEY')
+            
+        with gr.Column():
+            keylock_builder = KeylockDecoderComponent(server_logic)
+            image_input, status_display = keylock_builder.build_ui()
+        
+            output_json = gr.JSON(label="Decoded Payload")
+    
+    login_btn.click(login_fn,[user_name,user_pass], login_msg)
+    image_input.upload(
+        fn=process_image_and_display_payload,
+        inputs=image_input,
+        outputs=[status_display, output_json, user_name, user_pass, user_GROQ, user_HF, user_OPENAI, user_OPENROUTER]
+    )
+
+demo.launch()