fix spinning

wasm-demo.js

@@ -51,53 +51,59 @@ show('tokenizerSpin', false);
                 console.log(`[Main] Client key size: ${res.clientKey.length} bytes`);
                 console.log(`[Main] Server key size: ${res.serverKey.length} bytes`);
                 clientKey = res.clientKey; serverKey = res.serverKey;
-                $('keygenStatus').textContent = 'keys generated ✓';
-                enable('btnEncrypt');
                 
-                // Initialize encryption worker
-                encryptWorker = new Worker(new URL('./encrypt-worker.js', import.meta.url), { type: 'module' });
-                encryptWorker.onmessage = function(e) {
-                    if (e.data.type === 'ready') {
-                        console.log('[Main] Encryption worker ready');
-                    } else if (e.data.type === 'success') {
-                        encTokens = e.data.result;
-                        console.log(`[Main] Encryption completed: ${encTokens.length} bytes`);
-                        show('encryptSpin', false);
-                        show('encIcon', true);
-                        enable('btnEncrypt', true);
-                        enable('btnSend');
-                        $('encStatus').textContent = 'Your text is encrypted 🔒';
-                    } else if (e.data.type === 'error') {
-                        console.error('[Main] Encryption error:', e.data.error);
-                        show('encryptSpin', false);
-                        enable('btnEncrypt', true);
-                        $('encStatus').textContent = `Encryption failed: ${e.data.error}`;
-                        alert(`Encryption failed: ${e.data.error}`);
+                try {
+                    // Initialize encryption worker
+                    encryptWorker = new Worker(new URL('./encrypt-worker.js', import.meta.url), { type: 'module' });
+                    encryptWorker.onmessage = function(e) {
+                        if (e.data.type === 'ready') {
+                            console.log('[Main] Encryption worker ready');
+                        } else if (e.data.type === 'success') {
+                            encTokens = e.data.result;
+                            console.log(`[Main] Encryption completed: ${encTokens.length} bytes`);
+                            show('encryptSpin', false);
+                            show('encIcon', true);
+                            enable('btnEncrypt', true);
+                            enable('btnSend');
+                            $('encStatus').textContent = 'Your text is encrypted 🔒';
+                        } else if (e.data.type === 'error') {
+                            console.error('[Main] Encryption error:', e.data.error);
+                            show('encryptSpin', false);
+                            enable('btnEncrypt', true);
+                            $('encStatus').textContent = `Encryption failed: ${e.data.error}`;
+                            alert(`Encryption failed: ${e.data.error}`);
+                        }
+                    };
+                    
+                    // Initialize the worker with the client key
+                    encryptWorker.postMessage({ type: 'init', clientKey });
+                    
+                    console.log('[Main] Encoding server key to base64...');
+                    const serverKeyB64 = await uint8ToBase64(serverKey);
+                    console.log(`[Main] Server key base64 size: ${serverKeyB64.length} bytes`);
+                    const handshakeResponse = await fetch(`${SERVER}/handshake`, {
+                        method:'POST', 
+                        headers:{'Content-Type':'application/json'},
+                        body: JSON.stringify({
+                            server_key_b64: serverKeyB64
+                        })
+                    });
+                    
+                    if (!handshakeResponse.ok) {
+                        const errorText = await handshakeResponse.text();
+                        throw new Error(`Server handshake failed: ${handshakeResponse.status} ${errorText}`);
                     }
-                };
-                
-                // Initialize the worker with the client key
-                encryptWorker.postMessage({ type: 'init', clientKey });
-                
-                console.log('[Main] Encoding server key to base64...');
-                const serverKeyB64 = await uint8ToBase64(serverKey);
-                console.log(`[Main] Server key base64 size: ${serverKeyB64.length} bytes`);
-                const handshakeResponse = await fetch(`${SERVER}/handshake`, {
-                    method:'POST', 
-                    headers:{'Content-Type':'application/json'},
-                    body: JSON.stringify({
-                        server_key_b64: serverKeyB64
-                    })
-                });
-                
-                if (!handshakeResponse.ok) {
-                    const errorText = await handshakeResponse.text();
-                    throw new Error(`Server handshake failed: ${handshakeResponse.status} ${errorText}`);
+                    
+                    const { uid } = await handshakeResponse.json();
+                    sessionUid = uid;
+                    console.log('[Main] Server handshake successful');
+                    $('keygenStatus').textContent = 'keys generated ✓';
+                    enable('btnEncrypt');
+                } catch (error) {
+                    console.error('[Main] Server handshake error:', error);
+                    $('keygenStatus').textContent = `Server handshake failed: ${error.message}`;
+                    enable('btnEncrypt', false);
                 }
-                
-                const { uid } = await handshakeResponse.json();
-                sessionUid = uid;
-                console.log('[Main] Server handshake successful');
             } else {
                 console.error('[Main] Key generation error:', e.data.error);
                 $('keygenStatus').textContent = `Error generating keys: ${e.data.error}`;