Update app.py

app.py

@@ -154,8 +154,17 @@ def api_decode_and_auth(image_base64_string: str) -> dict:
     except Exception as e:
         return {"status": "Error", "message": f"Decryption/Processing Failed: {e}", "details": {}}
 
-def load_endpoints(): return json.load(open(ENDPOINTS_FILE)) if os.path.exists(ENDPOINTS_FILE) else []
-def save_endpoints(endpoints_list): json.dump(endpoints_list, open(ENDPOINTS_FILE, 'w'), indent=2)
+def load_endpoints():
+    try:
+        if os.path.exists(ENDPOINTS_FILE):
+            with open(ENDPOINTS_FILE, 'r') as f:
+                return json.load(f)
+    except (FileNotFoundError, json.JSONDecodeError):
+        pass
+    return []
+def save_endpoints(endpoints_list):
+    with open(ENDPOINTS_FILE, 'w') as f:
+        json.dump(endpoints_list, f, indent=2)
 
 theme = gr.themes.Soft(primary_hue="sky", secondary_hue="blue", neutral_hue="slate")
 with gr.Blocks(theme=theme, title="KeyLock Showcase") as demo:
@@ -250,7 +259,13 @@ with gr.Blocks(theme=theme, title="KeyLock Showcase") as demo:
             return updates
         except Exception as e:
             gr.Error(f"Connection Failed: {e}")
-            return {}
+            return {
+                active_server_state: gr.update(),
+                client_status_display: gr.update(),
+                client_payload_input: gr.update(),
+                all_servers_state: gr.update(),
+                saved_servers_dropdown: gr.update()
+            }
 
     def client_generate_image_wrapper(active_server, payload_str, overlay, base_img, progress=gr.Progress(track_tqdm=True)):
         if not active_server: raise gr.Error("Not connected to a server.")