Update app.py

app.py

@@ -159,7 +159,7 @@ def save_endpoints(endpoints_list): json.dump(endpoints_list, open(ENDPOINTS_FIL
 
 theme = gr.themes.Soft(primary_hue="sky", secondary_hue="blue", neutral_hue="slate")
 with gr.Blocks(theme=theme, title="KeyLock Showcase") as demo:
-    all_servers_state = gr.State(load_endpoints)
+    all_servers_state = gr.State(value=load_endpoints())
     active_server_state = gr.State({})
 
     gr.Markdown("# 🔑 KeyLock Showcase")
@@ -196,7 +196,7 @@ with gr.Blocks(theme=theme, title="KeyLock Showcase") as demo:
             gr.Markdown("## Embedded Server Details")
             gr.Markdown("This Gradio app is also running its own KeyLock server. You can use its details to test the client.")
             gr.Textbox(label="Embedded Server Status", value=KEYLOCK_STATUS_MESSAGE, interactive=False, lines=3)
-            gr.Code(label="Embedded Server Public Key", value=PUBLIC_KEY_PEM_STRING, language="python")
+            gr.Code(label="Embedded Server Public Key", value=PUBLIC_KEY_PEM_STRING, language="pem")
             gr.JSON(label="Embedded Server Required Payload", value={k['key_name']: k['example'] for k in api_get_info()["required_payload_keys"]})
             
             with gr.Accordion("Generate Image with Embedded Server", open=False):
@@ -210,8 +210,8 @@ with gr.Blocks(theme=theme, title="KeyLock Showcase") as demo:
                     output_private_key = gr.Textbox(lines=8, label="Generated Private Key", interactive=False, show_copy_button=True)
                     output_public_key = gr.Textbox(lines=8, label="Generated Public Key", interactive=False, show_copy_button=True)
 
-    def initialize_ui(all_servers):
-        return gr.update(choices=[s['name'] for s in all_servers])
+    def initialize_ui(all_servers_list):
+        return gr.update(choices=[s['name'] for s in all_servers_list] if all_servers_list else [])
 
     def process_server_connection(server_data, all_servers):
         placeholder = "\n".join([f"{k['key_name']}: {k['example']}" for k in server_data['info'].get('required_payload_keys', [])])