hedgequantx 2.9.20 → 2.9.22

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "hedgequantx",
-  "version": "2.9.20",
+  "version": "2.9.22",
   "description": "HedgeQuantX - Prop Futures Trading CLI",
   "main": "src/app.js",
   "bin": {

package/src/app.js

@@ -7,7 +7,7 @@ const chalk = require('chalk');
 const ora = require('ora');
 
 const { connections } = require('./services');
-const { getLogoWidth, centerText, prepareStdin } = require('./ui');
+const { getLogoWidth, centerText, prepareStdin, clearScreen } = require('./ui');
 const { logger, prompts } = require('./utils');
 const { setCachedStats, clearCachedStats } = require('./services/stats-cache');
 
@@ -17,6 +17,7 @@ const log = logger.scope('App');
 const { showStats } = require('./pages/stats');
 const { showAccounts } = require('./pages/accounts');
 const { algoTradingMenu } = require('./pages/algo');
+const { aiAgentsMenu, getActiveAgentCount } = require('./pages/ai-agents');
 
 // Menus
 const { rithmicMenu, dashboardMenu, handleUpdate } = require('./menus');
@@ -29,7 +30,7 @@ let currentService = null;
 
 const restoreTerminal = () => {
   try {
-    process.stdout.write('\x1B[?1049l');
+    // Show cursor
     process.stdout.write('\x1B[?25h');
     if (process.stdin.isTTY && process.stdin.isRaw) {
       process.stdin.setRawMode(false);
@@ -41,7 +42,15 @@ const restoreTerminal = () => {
 };
 
 process.on('exit', restoreTerminal);
-process.on('SIGINT', () => { restoreTerminal(); process.exit(0); });
+process.on('SIGINT', () => { 
+  restoreTerminal();
+  // Draw bottom border before exit
+  const termWidth = process.stdout.columns || 100;
+  const boxWidth = termWidth < 60 ? Math.max(termWidth - 2, 40) : Math.max(getLogoWidth(), 98);
+  console.log(chalk.cyan('\n╚' + '═'.repeat(boxWidth - 2) + '╝'));
+  console.log(chalk.gray('GOODBYE!'));
+  process.exit(0);
+});
 process.on('SIGTERM', () => { restoreTerminal(); process.exit(0); });
 
 process.on('uncaughtException', (err) => {
@@ -92,6 +101,7 @@ const refreshStats = async () => {
       balance: hasBalanceData ? totalBalance : null,
       pnl: hasPnlData ? totalPnl : null,
       pnlPercent: null,
+      agents: getActiveAgentCount(),
     });
   } catch (err) {
     log.warn('Failed to refresh stats', { error: err.message });
@@ -101,7 +111,7 @@ const refreshStats = async () => {
 // ==================== BANNER ====================
 
 const banner = async () => {
-  console.clear();
+  clearScreen();
   
   const termWidth = process.stdout.columns || 100;
   const isMobile = termWidth < 60;
@@ -123,8 +133,11 @@ const banner = async () => {
 
   console.log(chalk.cyan('╠' + '═'.repeat(innerWidth) + '╣'));
   
-  const tagline = isMobile ? `HQX v${version}` : `Prop Futures Algo Trading  v${version}`;
-  console.log(chalk.cyan('║') + chalk.white(centerText(tagline, innerWidth)) + chalk.cyan('║'));
+  const tagline = isMobile ? `HQX V${version}` : `PROP FUTURES ALGO TRADING  V${version}`;
+  console.log(chalk.cyan('║') + chalk.yellow(centerText(tagline, innerWidth)) + chalk.cyan('║'));
+  
+  // ALWAYS close the banner
+  console.log(chalk.cyan('╚' + '═'.repeat(innerWidth) + '╝'));
 };
 
 const getFullLogo = () => [
@@ -145,39 +158,37 @@ const getMobileLogo = () => [
   ['╚═╝  ╚═╝ ╚══▀▀═╝ ', '╚═╝  ╚═╝'],
 ];
 
-const bannerClosed = async () => {
-  await banner();
-  const termWidth = process.stdout.columns || 100;
-  const boxWidth = termWidth < 60 ? Math.max(termWidth - 2, 40) : Math.max(getLogoWidth(), 98);
-  console.log(chalk.cyan('╚' + '═'.repeat(boxWidth - 2) + '╝'));
-};
-
 // ==================== MAIN LOOP ====================
 
 const run = async () => {
   try {
     log.info('Starting HQX CLI');
-    await bannerClosed();
-
-    // Restore session
-    const spinner = ora({ text: 'Restoring session...', color: 'yellow' }).start();
+    
+    // First launch - show banner then try restore session
+    await banner();
+    
+    const spinner = ora({ text: 'LOADING DASHBOARD...', color: 'yellow' }).start();
+    
     const restored = await connections.restoreFromStorage();
 
     if (restored) {
-      spinner.succeed('Session restored');
       currentService = connections.getAll()[0].service;
       await refreshStats();
+      // Store spinner globally - dashboard will stop it when ready to display
+      global.__hqxSpinner = spinner;
     } else {
-      spinner.info('No active session');
+      spinner.stop(); // Stop spinner - no session to restore
+      global.__hqxSpinner = null;
     }
 
     // Main loop
     while (true) {
       try {
         prepareStdin();
-        await banner();
 
         if (!connections.isConnected()) {
+          // Not connected - show banner + propfirm selection
+          await banner();
           // Not connected - show propfirm selection directly
           const boxWidth = getLogoWidth();
           const innerWidth = boxWidth - 2;
@@ -188,13 +199,14 @@ const run = async () => {
           
           // Find max name length for alignment
           const maxNameLen = Math.max(...numbered.map(n => n.name.length));
-          const colWidth = 4 + 1 + maxNameLen + 2; // [##] + space + name + gap
-          const totalContentWidth = numCols * colWidth;
-          const leftMargin = Math.max(2, Math.floor((innerWidth - totalContentWidth) / 2));
+          const itemWidth = 4 + 1 + maxNameLen; // [##] + space + name
+          const gap = 3; // gap between columns
+          const totalContentWidth = (itemWidth * numCols) + (gap * (numCols - 1));
           
-          console.log(chalk.cyan('╠' + '═'.repeat(innerWidth) + '╣'));
+          // New rectangle (banner is always closed)
+          console.log(chalk.cyan('╔' + '═'.repeat(innerWidth) + '╗'));
           console.log(chalk.cyan('║') + chalk.white.bold(centerText('SELECT PROPFIRM', innerWidth)) + chalk.cyan('║'));
-          console.log(chalk.cyan('║') + ' '.repeat(innerWidth) + chalk.cyan('║'));
+          console.log(chalk.cyan('╠' + '═'.repeat(innerWidth) + '╣'));
           
           const rows = Math.ceil(numbered.length / numCols);
           for (let row = 0; row < rows; row++) {
@@ -211,29 +223,32 @@ const run = async () => {
               }
             }
             
-            let line = ' '.repeat(leftMargin);
+            // Build line content
+            let content = '';
             for (let i = 0; i < lineParts.length; i++) {
               if (lineParts[i]) {
-                line += chalk.cyan(lineParts[i].num) + ' ' + chalk.white(lineParts[i].name);
+                content += chalk.cyan(lineParts[i].num) + ' ' + chalk.white(lineParts[i].name);
               } else {
-                line += ' '.repeat(4 + 1 + maxNameLen);
+                content += ' '.repeat(itemWidth);
               }
-              if (i < lineParts.length - 1) line += '  ';
+              if (i < lineParts.length - 1) content += ' '.repeat(gap);
             }
             
-            const lineLen = line.replace(/\x1b\[[0-9;]*m/g, '').length;
-            const rightPad = Math.max(0, innerWidth - lineLen);
-            console.log(chalk.cyan('║') + line + ' '.repeat(rightPad) + chalk.cyan('║'));
+            // Center the content
+            const contentLen = content.replace(/\x1b\[[0-9;]*m/g, '').length;
+            const leftPad = Math.floor((innerWidth - contentLen) / 2);
+            const rightPad = innerWidth - contentLen - leftPad;
+            console.log(chalk.cyan('║') + ' '.repeat(leftPad) + content + ' '.repeat(rightPad) + chalk.cyan('║'));
           }
           
           console.log(chalk.cyan('╠' + '─'.repeat(innerWidth) + '╣'));
-          console.log(chalk.cyan('║') + chalk.red(centerText('[X] Exit', innerWidth)) + chalk.cyan('║'));
+          console.log(chalk.cyan('║') + chalk.red(centerText('[X] EXIT', innerWidth)) + chalk.cyan('║'));
           console.log(chalk.cyan('╚' + '═'.repeat(innerWidth) + '╝'));
           
-          const input = await prompts.textInput(chalk.cyan('Select number (or X):'));
+          const input = await prompts.textInput(chalk.cyan('SELECT (1-' + numbered.length + '/X): '));
           
           if (!input || input.toLowerCase() === 'x') {
-            console.log(chalk.gray('Goodbye!'));
+            console.log(chalk.gray('GOODBYE!'));
             process.exit(0);
           }
           
@@ -244,26 +259,26 @@ const run = async () => {
             const credentials = await loginPrompt(selectedPropfirm.name);
             
             if (credentials) {
-              const spinner = ora({ text: 'Connecting to Rithmic...', color: 'yellow' }).start();
+              const spinner = ora({ text: 'CONNECTING TO RITHMIC...', color: 'yellow' }).start();
               try {
                 const { RithmicService } = require('./services/rithmic');
                 const service = new RithmicService(selectedPropfirm.key);
                 const result = await service.login(credentials.username, credentials.password);
                 
                 if (result.success) {
-                  spinner.text = 'Fetching accounts...';
+                  spinner.text = 'FETCHING ACCOUNTS...';
                   const accResult = await service.getTradingAccounts();
                   connections.add('rithmic', service, service.propfirm.name);
-                  spinner.succeed(`Connected to ${service.propfirm.name} (${accResult.accounts?.length || 0} accounts)`);
+                  spinner.succeed(`CONNECTED TO ${service.propfirm.name.toUpperCase()} (${accResult.accounts?.length || 0} ACCOUNTS)`);
                   currentService = service;
                   await refreshStats();
                   await new Promise(r => setTimeout(r, 1500));
                 } else {
-                  spinner.fail(result.error || 'Authentication failed');
+                  spinner.fail((result.error || 'AUTHENTICATION FAILED').toUpperCase());
                   await new Promise(r => setTimeout(r, 2000));
                 }
               } catch (error) {
-                spinner.fail(`Connection error: ${error.message}`);
+                spinner.fail(`CONNECTION ERROR: ${error.message.toUpperCase()}`);
                 await new Promise(r => setTimeout(r, 2000));
               }
             }
@@ -296,13 +311,20 @@ const run = async () => {
               try {
                 await algoTradingMenu(currentService);
               } catch (err) {
-                console.log(chalk.red(`  Algo error: ${err.message}`));
+                console.log(chalk.red(`  ALGO ERROR: ${err.message.toUpperCase()}`));
                 prepareStdin();
               }
               break;
 
+            case 'aiagents':
+              await aiAgentsMenu();
+              break;
+
             case 'update':
-              await handleUpdate();
+              const updateResult = await handleUpdate();
+              if (updateResult === 'exit') {
+                running = false;
+              }
               break;
 
             case 'disconnect':

package/src/menus/connect.js

@@ -8,7 +8,7 @@ const ora = require('ora');
 const { connections } = require('../services');
 const { RithmicService } = require('../services/rithmic');
 const { PROPFIRM_CHOICES } = require('../config');
-const { getLogoWidth, centerText, prepareStdin } = require('../ui');
+const { getLogoWidth, centerText, prepareStdin, displayBanner , clearScreen } = require('../ui');
 const { validateUsername, validatePassword } = require('../security');
 const { prompts } = require('../utils');
 
@@ -18,16 +18,16 @@ const { prompts } = require('../utils');
 const loginPrompt = async (propfirmName) => {
   prepareStdin();
   console.log();
-  console.log(chalk.cyan(`Connecting to ${propfirmName}...`));
+  console.log(chalk.cyan(`CONNECTING TO ${propfirmName.toUpperCase()}...`));
   console.log();
 
-  const username = await prompts.textInput('Username:', '', (input) => {
-    try { validateUsername(input); return undefined; } catch (e) { return e.message; }
+  const username = await prompts.textInput('USERNAME:', '', (input) => {
+    try { validateUsername(input); return undefined; } catch (e) { return e.message.toUpperCase(); }
   });
   if (!username) return null;
   
-  const pwd = await prompts.passwordInput('Password:', (input) => {
-    try { validatePassword(input); return undefined; } catch (e) { return e.message; }
+  const pwd = await prompts.passwordInput('PASSWORD:', (input) => {
+    try { validatePassword(input); return undefined; } catch (e) { return e.message.toUpperCase(); }
   });
   if (!pwd) return null;
 
@@ -38,6 +38,10 @@ const loginPrompt = async (propfirmName) => {
  * Rithmic menu - Main connection menu
  */
 const rithmicMenu = async () => {
+  // Clear screen and show banner
+  clearScreen();
+  displayBanner();
+  
   const propfirms = PROPFIRM_CHOICES;
   const boxWidth = getLogoWidth();
   const innerWidth = boxWidth - 2;
@@ -46,7 +50,6 @@ const rithmicMenu = async () => {
   
   const numbered = propfirms.map((pf, i) => ({ num: i + 1, key: pf.value, name: pf.name }));
   
-  console.log();
   console.log(chalk.cyan('╔' + '═'.repeat(innerWidth) + '╗'));
   console.log(chalk.cyan('║') + chalk.white.bold(centerText('SELECT PROPFIRM', innerWidth)) + chalk.cyan('║'));
   console.log(chalk.cyan('║') + ' '.repeat(innerWidth) + chalk.cyan('║'));
@@ -71,10 +74,10 @@ const rithmicMenu = async () => {
   }
   
   console.log(chalk.cyan('╠' + '─'.repeat(innerWidth) + '╣'));
-  console.log(chalk.cyan('║') + chalk.red(centerText('[X] Exit', innerWidth)) + chalk.cyan('║'));
+  console.log(chalk.cyan('║') + chalk.red(centerText('[X] EXIT', innerWidth)) + chalk.cyan('║'));
   console.log(chalk.cyan('╚' + '═'.repeat(innerWidth) + '╝'));
 
-  const input = await prompts.textInput(chalk.cyan('Select number (or X):'));
+  const input = await prompts.textInput(chalk.cyan('SELECT NUMBER (OR X):'));
   if (!input || input.toLowerCase() === 'x') return null;
   
   const action = parseInt(input);
@@ -84,26 +87,26 @@ const rithmicMenu = async () => {
   const credentials = await loginPrompt(selectedPropfirm.name);
   if (!credentials) return null;
 
-  const spinner = ora({ text: 'Connecting to Rithmic...', color: 'yellow' }).start();
+  const spinner = ora({ text: 'CONNECTING TO RITHMIC...', color: 'yellow' }).start();
 
   try {
     const service = new RithmicService(selectedPropfirm.key);
     const result = await service.login(credentials.username, credentials.password);
 
     if (result.success) {
-      spinner.text = 'Fetching accounts...';
+      spinner.text = 'FETCHING ACCOUNTS...';
       const accResult = await service.getTradingAccounts();
       connections.add('rithmic', service, service.propfirm.name);
-      spinner.succeed(`Connected to ${service.propfirm.name} (${accResult.accounts?.length || 0} accounts)`);
+      spinner.succeed(`CONNECTED TO ${service.propfirm.name.toUpperCase()} (${accResult.accounts?.length || 0} ACCOUNTS)`);
       await new Promise(r => setTimeout(r, 1500));
       return service;
     } else {
-      spinner.fail(result.error || 'Authentication failed');
+      spinner.fail((result.error || 'AUTHENTICATION FAILED').toUpperCase());
       await new Promise(r => setTimeout(r, 2000));
       return null;
     }
   } catch (error) {
-    spinner.fail(`Connection error: ${error.message}`);
+    spinner.fail(`CONNECTION ERROR: ${error.message.toUpperCase()}`);
     await new Promise(r => setTimeout(r, 2000));
     return null;
   }

package/src/menus/dashboard.js

@@ -7,9 +7,10 @@ const ora = require('ora');
 const { execSync, spawn } = require('child_process');
 
 const { connections } = require('../services');
-const { getLogoWidth, centerText, prepareStdin } = require('../ui');
+const { getLogoWidth, centerText, prepareStdin, displayBanner, clearScreen } = require('../ui');
 const { getCachedStats } = require('../services/stats-cache');
 const { prompts } = require('../utils');
+const { getActiveAgentCount } = require('../pages/ai-agents');
 
 /**
  * Dashboard menu after login
@@ -17,6 +18,16 @@ const { prompts } = require('../utils');
 const dashboardMenu = async (service) => {
   prepareStdin();
   
+  // Stop any global spinner before clearing
+  if (global.__hqxSpinner) {
+    global.__hqxSpinner.stop();
+    global.__hqxSpinner = null;
+  }
+  
+  // Clear screen and show banner (always closed)
+  clearScreen();
+  displayBanner();
+  
   const boxWidth = getLogoWidth();
   const W = boxWidth - 2;
   
@@ -30,9 +41,9 @@ const dashboardMenu = async (service) => {
     return chalk.cyan('║') + content + ' '.repeat(Math.max(0, padding)) + chalk.cyan('║');
   };
   
-  // Continue from banner (use ╠ not ╔)
-  console.log(chalk.cyan('╠' + '═'.repeat(W) + '╣'));
-  console.log(makeLine(chalk.yellow.bold('Welcome, HQX Trader!'), 'center'));
+  // New rectangle (banner is always closed)
+  console.log(chalk.cyan('╔' + '═'.repeat(W) + '╗'));
+  console.log(makeLine(chalk.yellow.bold('WELCOME, HQX TRADER!'), 'center'));
   console.log(chalk.cyan('╠' + '═'.repeat(W) + '╣'));
   
   // Show connected propfirms
@@ -51,8 +62,8 @@ const dashboardMenu = async (service) => {
     const balStr = statsInfo.balance !== null ? `$${statsInfo.balance.toLocaleString()}` : '--';
     const balColor = statsInfo.balance !== null ? chalk.green : chalk.gray;
     
-    // AI Agents status
-    const agentCount = statsInfo.agents || 0;
+    // AI Agents status - get fresh count, not from cache
+    const agentCount = getActiveAgentCount();
     const agentDisplay = agentCount > 0 ? 'ON' : 'OFF';
     const agentColor = agentCount > 0 ? chalk.green : chalk.red;
     
@@ -87,9 +98,9 @@ const dashboardMenu = async (service) => {
   
   // Find max width for alignment
   const menuItems = [
-    { left: '[1] View Accounts', right: '[2] View Stats' },
-    { left: '[+] Add Prop-Account', right: '[A] Algo-Trading' },
-    { left: '[U] Update HQX', right: '[X] Disconnect' },
+    { left: '[1] VIEW ACCOUNTS', right: '[2] VIEW STATS' },
+    { left: '[+] ADD PROP-ACCOUNT', right: '[A] ALGO-TRADING' },
+    { left: '[I] AI AGENTS', right: '[U] UPDATE HQX' },
   ];
   
   const maxLeftLen = Math.max(...menuItems.map(m => m.left.length));
@@ -119,20 +130,25 @@ const dashboardMenu = async (service) => {
     );
   };
   
-  menuRow('[1] View Accounts', '[2] View Stats', chalk.cyan, chalk.cyan);
-  menuRow('[+] Add Prop-Account', '[A] Algo-Trading', chalk.cyan, chalk.magenta);
-  menuRow('[U] Update HQX', '[X] Disconnect', chalk.yellow, chalk.red);
+  menuRow('[1] VIEW ACCOUNTS', '[2] VIEW STATS', chalk.cyan, chalk.cyan);
+  menuRow('[+] ADD PROP-ACCOUNT', '[A] ALGO-TRADING', chalk.cyan, chalk.magenta);
+  menuRow('[I] AI AGENTS', '[U] UPDATE HQX', chalk.green, chalk.yellow);
+  
+  // Separator and centered Disconnect button
+  console.log(chalk.cyan('╠' + '─'.repeat(W) + '╣'));
+  console.log(makeLine(chalk.red('[X] DISCONNECT'), 'center'));
   
   console.log(chalk.cyan('╚' + '═'.repeat(W) + '╝'));
   
   // Simple input - no duplicate menu
-  const input = await prompts.textInput(chalk.cyan('Select (1/2/+/A/U/X)'));
+  const input = await prompts.textInput(chalk.cyan('SELECT (1/2/+/A/I/U/X):'));
   
   const actionMap = {
     '1': 'accounts',
     '2': 'stats',
     '+': 'add_prop_account',
     'a': 'algotrading',
+    'i': 'aiagents',
     'u': 'update',
     'x': 'disconnect'
   };
@@ -144,93 +160,95 @@ const dashboardMenu = async (service) => {
  * Handle update process
  */
 const handleUpdate = async () => {
+  clearScreen();
+  displayBanner();
   prepareStdin();
   
+  const boxWidth = getLogoWidth();
+  const W = boxWidth - 2;
+  
+  console.log(chalk.cyan('╔' + '═'.repeat(W) + '╗'));
+  console.log(chalk.cyan('║') + chalk.yellow.bold(centerText('UPDATE HQX', W)) + chalk.cyan('║'));
+  console.log(chalk.cyan('╚' + '═'.repeat(W) + '╝'));
+  
   let spinner = null;
   let currentVersion = 'unknown';
   
   try {
+    // Get current version
     try {
       currentVersion = require('../../package.json').version || 'unknown';
     } catch (e) {}
     
-    console.log(chalk.cyan(`\n  Current version: v${currentVersion}`));
-    spinner = ora({ text: 'Checking for updates...', color: 'yellow' }).start();
+    console.log(chalk.cyan(`\n  CURRENT VERSION: V${currentVersion.toUpperCase()}`));
+    spinner = ora({ text: 'CHECKING FOR UPDATES...', color: 'yellow' }).start();
     
+    // Check latest version from npm
     let latestVersion;
     try {
-      latestVersion = execSync('npm view hedgequantx version', { 
+      latestVersion = execSync('npm view hedgequantx version 2>/dev/null', { 
         stdio: ['pipe', 'pipe', 'pipe'], 
         timeout: 30000, 
         encoding: 'utf8'
       }).trim();
       
       if (!latestVersion || !/^\d+\.\d+\.\d+/.test(latestVersion)) {
-        throw new Error('Invalid version format');
+        throw new Error('INVALID VERSION FORMAT');
       }
     } catch (e) {
-      spinner.fail('Cannot reach npm registry');
-      console.log(chalk.gray(`  Error: ${e.message}`));
-      console.log(chalk.yellow('  Try manually: npm install -g hedgequantx@latest'));
+      spinner.fail('CANNOT REACH NPM REGISTRY');
+      console.log(chalk.yellow('\n  TRY MANUALLY: npm update -g hedgequantx'));
       await prompts.waitForEnter();
       return;
     }
     
-    spinner.succeed(`Latest version: v${latestVersion}`);
+    spinner.succeed(`LATEST VERSION: V${latestVersion.toUpperCase()}`);
     
+    // Already up to date
     if (currentVersion === latestVersion) {
-      console.log(chalk.green('  Already up to date!'));
+      console.log(chalk.green('\n  ✓ ALREADY UP TO DATE!'));
       await prompts.waitForEnter();
       return;
     }
     
-    console.log(chalk.yellow(`  Update available: v${currentVersion} → v${latestVersion}`));
-    spinner = ora({ text: 'Installing update...', color: 'yellow' }).start();
+    // Update available
+    console.log(chalk.yellow(`\n  UPDATE AVAILABLE: V${currentVersion} → V${latestVersion}`));
+    spinner = ora({ text: 'INSTALLING UPDATE...', color: 'yellow' }).start();
     
+    // Try to install update
     try {
-      // Try with sudo first on Unix systems
-      const isWindows = process.platform === 'win32';
-      const cmd = isWindows 
-        ? 'npm install -g hedgequantx@latest'
-        : 'npm install -g hedgequantx@latest';
-      
-      execSync(cmd, { 
+      execSync('npm update -g hedgequantx 2>/dev/null', { 
         stdio: ['pipe', 'pipe', 'pipe'], 
         timeout: 180000, 
         encoding: 'utf8'
       });
     } catch (e) {
-      spinner.fail('Update failed - permission denied?');
-      console.log(chalk.gray(`  Error: ${e.message}`));
-      console.log(chalk.yellow('  Try manually with sudo:'));
-      console.log(chalk.white('  sudo npm install -g hedgequantx@latest'));
-      await prompts.waitForEnter();
-      return;
+      // Try without redirecting stderr
+      try {
+        execSync('npm update -g hedgequantx', { 
+          stdio: ['pipe', 'pipe', 'pipe'], 
+          timeout: 180000, 
+          encoding: 'utf8'
+        });
+      } catch (e2) {
+        spinner.fail('UPDATE FAILED');
+        console.log(chalk.yellow('\n  TRY MANUALLY:'));
+        console.log(chalk.white('  npm update -g hedgequantx'));
+        console.log(chalk.gray('  OR WITH SUDO:'));
+        console.log(chalk.white('  sudo npm update -g hedgequantx'));
+        await prompts.waitForEnter();
+        return;
+      }
     }
     
-    spinner.succeed(`Updated to v${latestVersion}!`);
-    console.log(chalk.cyan('  Restarting HQX...'));
-    
-    await new Promise(r => setTimeout(r, 1500));
-    
-    try {
-      const child = spawn('hqx', [], { 
-        stdio: 'inherit', 
-        detached: true, 
-        shell: true 
-      });
-      child.unref();
-      process.exit(0);
-    } catch (e) {
-      console.log(chalk.yellow('\n  Please restart HQX manually:'));
-      console.log(chalk.white('  hqx'));
-      await prompts.waitForEnter();
-    }
+    spinner.succeed(`UPDATED TO V${latestVersion}!`);
+    console.log(chalk.green('\n  ✓ UPDATE SUCCESSFUL!'));
+    console.log(chalk.yellow('\n  Run ') + chalk.cyan('hqx') + chalk.yellow(' to start the new version.\n'));
+    process.exit(0);
     
   } catch (error) {
-    if (spinner) spinner.fail('Update error');
-    console.log(chalk.gray(`  Error: ${error.message}`));
-    console.log(chalk.yellow('  Try manually: npm install -g hedgequantx@latest'));
+    if (spinner) spinner.fail('UPDATE ERROR');
+    console.log(chalk.yellow('\n  TRY MANUALLY: npm update -g hedgequantx'));
     await prompts.waitForEnter();
   }
 };