File size: 3,244 Bytes
36a3555 8a2e87d bfe9d96 36a3555 ad29d5b c6a8702 0cb08b4 c4624e3 0cb08b4 ad29d5b 8a2e87d bfe9d96 8a2e87d bfe9d96 04b5728 49a6d0b 04b5728 49a6d0b 8a2e87d bfe9d96 8a2e87d 94a49bf bfe9d96 e82925f c98bcc3 36a3555 bfe9d96 8a2e87d 36a3555 e70f9b1 04f14c4 ad29d5b 04f14c4 e70f9b1 04f14c4 ad29d5b 8a2e87d c233e1a bfe9d96 36a3555 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
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
gr_head = "<style>" + KeylockDecoderComponent(server_logic=server_logic).CSS + "</style>"
with gr.Blocks(theme=gr.themes.Soft(), head = gr_head) 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() |