@erosolaraijs/cure 1.0.1 → 1.0.3

package/src/bin/cure.ts

@@ -1,182 +1,451 @@
 #!/usr/bin/env node
 /**
- * 癌症治疗统一人工智能框架 - 核心医疗控制台
- * Unified Cancer Treatment Framework - Core Medical Console
- *
- * 全球首个人工通用智能驱动的癌症治疗统一指挥控制平台
- * 提供完整的癌症研究、药物发现、治疗规划、患者管理和医疗系统集成能力
+ * Cure - AI Cancer Treatment Framework
+ * Interactive CLI powered by xAI Grok for precision oncology
  */
 
+import * as readline from 'readline';
+import * as https from 'https';
 import { CancerTreatmentCapabilityModule } from '../capabilities/cancerTreatmentCapability.js';
 
+const VERSION = '1.0.3';
+const XAI_MODEL = 'grok-4-1-fast-reasoning';
+
+// ANSI color codes
+const colors = {
+  reset: '\x1b[0m',
+  bold: '\x1b[1m',
+  dim: '\x1b[2m',
+  cyan: '\x1b[36m',
+  green: '\x1b[32m',
+  yellow: '\x1b[33m',
+  blue: '\x1b[34m',
+  magenta: '\x1b[35m',
+  red: '\x1b[31m',
+  white: '\x1b[37m',
+};
+
+let cancerTreatment: CancerTreatmentCapabilityModule;
+let conversationHistory: Array<{role: string, content: string}> = [];
+
+const SYSTEM_PROMPT = `You are Cure, an advanced AI oncologist assistant powered by the Cure Cancer Treatment Framework. You help doctors, researchers, and patients with:
+
+1. Cancer diagnosis and staging analysis
+2. Personalized treatment planning (chemotherapy, immunotherapy, targeted therapy, CAR-T)
+3. Drug target discovery and mechanism analysis
+4. Clinical trial matching and eligibility
+5. Genomic biomarker interpretation (EGFR, KRAS, BRAF, HER2, PD-L1, etc.)
+6. Treatment response prediction and survival analysis
+7. Drug interaction and safety checks
+8. HIPAA-compliant patient data handling
+
+You have deep knowledge of:
+- NCCN, ESMO, ASCO treatment guidelines
+- FDA-approved cancer therapies and their mechanisms
+- Precision medicine and molecular oncology
+- Immunotherapy (checkpoint inhibitors, CAR-T, TILs)
+- Targeted therapies for driver mutations
+- Clinical trial design and interpretation
+
+Be concise, scientifically accurate, and clinically relevant. When discussing specific treatments, cite evidence levels and relevant trials. Always recommend consulting with treating oncologists for actual patient care decisions.
+
+Available commands the user can run:
+- /analyze [patient_id] - Analyze patient data
+- /plan [patient_id] - Design treatment plan
+- /discover [gene] [cancer] - Drug target discovery
+- /demo - Run framework demonstration
+- /help - Show available commands`;
+
+async function callXAI(userMessage: string): Promise<string> {
+  const apiKey = process.env.XAI_API_KEY;
+
+  if (!apiKey) {
+    return `${colors.yellow}Note: Set XAI_API_KEY environment variable for AI-powered responses.${colors.reset}\n\nI can help you with cancer treatment analysis. Try:\n  /analyze P001 - Analyze patient\n  /plan P001 - Design treatment plan\n  /discover EGFR Lung - Find drug targets\n  /demo - Run demonstration`;
+  }
+
+  conversationHistory.push({ role: 'user', content: userMessage });
+
+  const messages = [
+    { role: 'system', content: SYSTEM_PROMPT },
+    ...conversationHistory
+  ];
+
+  const requestBody = JSON.stringify({
+    model: XAI_MODEL,
+    messages: messages,
+    temperature: 0.7,
+    max_tokens: 2048
+  });
+
+  return new Promise((resolve, reject) => {
+    const options = {
+      hostname: 'api.x.ai',
+      port: 443,
+      path: '/v1/chat/completions',
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': `Bearer ${apiKey}`,
+        'Content-Length': Buffer.byteLength(requestBody)
+      }
+    };
+
+    const req = https.request(options, (res) => {
+      let data = '';
+      res.on('data', (chunk) => { data += chunk; });
+      res.on('end', () => {
+        try {
+          const response = JSON.parse(data);
+          if (response.choices && response.choices[0]?.message?.content) {
+            const assistantMessage = response.choices[0].message.content;
+            conversationHistory.push({ role: 'assistant', content: assistantMessage });
+
+            // Keep conversation history manageable
+            if (conversationHistory.length > 20) {
+              conversationHistory = conversationHistory.slice(-16);
+            }
+
+            resolve(assistantMessage);
+          } else if (response.error) {
+            resolve(`${colors.red}API Error: ${response.error.message}${colors.reset}`);
+          } else {
+            resolve(`${colors.red}Unexpected response format${colors.reset}`);
+          }
+        } catch (e) {
+          resolve(`${colors.red}Failed to parse response${colors.reset}`);
+        }
+      });
+    });
+
+    req.on('error', (e) => {
+      resolve(`${colors.red}Connection error: ${e.message}${colors.reset}`);
+    });
+
+    req.setTimeout(30000, () => {
+      req.destroy();
+      resolve(`${colors.yellow}Request timed out. Try again.${colors.reset}`);
+    });
+
+    req.write(requestBody);
+    req.end();
+  });
+}
+
 async function main(): Promise<void> {
   const args = process.argv.slice(2);
 
-  // Check for version flag
   if (args.includes('--version') || args.includes('-v')) {
-    const pkg = JSON.parse(await import('fs').then(fs => fs.readFileSync(new URL('../../package.json', import.meta.url), 'utf-8')));
-    console.log(`agi-core-cancer-treatment v${pkg.version}`);
+    console.log(`cure v${VERSION}`);
     process.exit(0);
   }
 
-  // Check for help flag
   if (args.includes('--help') || args.includes('-h')) {
     printHelp();
     process.exit(0);
   }
 
-  // Initialize module
-  const cancerTreatment = new CancerTreatmentCapabilityModule();
-
-  // Handle commands
-  if (args.includes('--cancer-treatment') || args.includes('--medical')) {
-    console.log('\n🏥 癌症治疗人工智能框架激活\n');
-    console.log('💊 最大能力癌症治疗框架 - 全面医疗与AI能力\n');
-    console.log('📊 框架提供以下核心能力:');
-    console.log('   • 个性化癌症治疗协议分析与规划');
-    console.log('   • AI驱动的药物靶点发现与设计');
-    console.log('   • 精准医疗与基因组分析管道');
-    console.log('   • 临床试验管理与优化');
-    console.log('   • 患者多维度评估与监测');
-    console.log('   • 医疗系统集成与数据协调');
-    console.log('   • 统一医疗指挥与协调系统');
-    console.log('   • 人工智能治疗决策支持\n');
-    
-    console.log('✅ 癌症治疗能力框架初始化完成');
-    console.log('🚀 使用 --help 查看可用命令');
+  cancerTreatment = new CancerTreatmentCapabilityModule();
+
+  if (args.length > 0) {
+    await handleCommand(args);
+  } else {
+    await launchInteractiveMode();
   }
-  else if (args.includes('--analyze-patient')) {
-    const patientId = args.find(arg => arg.startsWith('--patient='))?.split('=')[1] || 'P001';
-    const includeGenomics = args.includes('--genomics');
-    
-    console.log(`\n🔍 分析患者: ${patientId}`);
-    console.log('───────────────────────────');
-    
-    try {
-      const result = await cancerTreatment.analyzePatient(patientId, includeGenomics);
-      console.log('📊 分析结果:');
-      console.log(`   • 癌症类型: ${result.analysis.cancerType}`);
-      console.log(`   • 分期: ${result.analysis.stage}`);
-      console.log(`   • 生物标志物: ${result.analysis.biomarkers.join(', ')}`);
-      console.log(`   • 生存概率: ${(result.analysis.riskAssessment.survivalProbability * 100).toFixed(1)}%`);
-      console.log('✅ 分析完成');
-    } catch (error) {
-      console.error('❌ 分析失败:', error);
+}
+
+async function launchInteractiveMode(): Promise<void> {
+  console.clear();
+  printBanner();
+
+  const rl = readline.createInterface({
+    input: process.stdin,
+    output: process.stdout,
+  });
+
+  const prompt = () => {
+    process.stdout.write(`\n${colors.cyan}cure${colors.reset} ${colors.dim}>${colors.reset} `);
+  };
+
+  const modelInfo = process.env.XAI_API_KEY
+    ? `${colors.green}Connected to xAI ${XAI_MODEL}${colors.reset}`
+    : `${colors.yellow}Set XAI_API_KEY for AI responses${colors.reset}`;
+
+  console.log(`${colors.dim}${modelInfo}${colors.reset}`);
+  console.log(`${colors.dim}Type your question or /help for commands.${colors.reset}\n`);
+  prompt();
+
+  rl.on('line', async (input) => {
+    const trimmed = input.trim();
+
+    if (!trimmed) {
+      prompt();
+      return;
     }
-  }
-  else if (args.includes('--treatment-plan')) {
-    const patientId = args.find(arg => arg.startsWith('--patient='))?.split('=')[1] || 'P001';
-    const protocolId = args.find(arg => arg.startsWith('--protocol='))?.split('=')[1];
-    
-    console.log(`\n💊 设计治疗计划: ${patientId}`);
-    console.log('─────────────────────────────');
-    
-    try {
-      const result = await cancerTreatment.designTreatmentPlan(patientId, protocolId);
-      console.log('📋 治疗计划:');
-      console.log(`   • 协议: ${result.plan.protocol.name}`);
-      console.log(`   • 治疗方式: ${result.plan.protocol.treatmentModalities.join(', ')}`);
-      console.log(`   • 预计疗效: ${(result.estimatedEfficacy * 100).toFixed(1)}%`);
-      console.log(`   • 时间线: ${result.plan.treatmentTimeline.length} 周`);
-      console.log('✅ 治疗计划设计完成');
-    } catch (error) {
-      console.error('❌ 治疗计划设计失败:', error);
+
+    if (trimmed === '/exit' || trimmed === '/quit' || trimmed === 'exit' || trimmed === 'quit') {
+      console.log(`\n${colors.dim}Goodbye.${colors.reset}\n`);
+      rl.close();
+      process.exit(0);
+    }
+
+    if (trimmed === '/help' || trimmed === 'help') {
+      printInteractiveHelp();
+      prompt();
+      return;
+    }
+
+    if (trimmed === '/clear') {
+      console.clear();
+      printBanner();
+      conversationHistory = [];
+      console.log(`${colors.dim}Conversation cleared.${colors.reset}`);
+      prompt();
+      return;
+    }
+
+    await processInput(trimmed);
+    prompt();
+  });
+
+  rl.on('close', () => {
+    process.exit(0);
+  });
+}
+
+async function processInput(input: string): Promise<void> {
+  // Slash commands
+  if (input.startsWith('/')) {
+    const parts = input.slice(1).split(' ');
+    const cmd = parts[0];
+    const args = parts.slice(1);
+
+    switch (cmd) {
+      case 'analyze':
+        await analyzePatient(args[0] || 'P001', args.includes('--genomics'));
+        return;
+      case 'plan':
+        await designTreatmentPlan(args[0] || 'P001', args[1]);
+        return;
+      case 'discover':
+        await discoverTargets(args[0] || 'EGFR', args[1] || 'Lung');
+        return;
+      case 'demo':
+        await runDemo();
+        return;
+      case 'version':
+        console.log(`\n${colors.cyan}cure${colors.reset} v${VERSION} (${XAI_MODEL})`);
+        return;
+      case 'model':
+        console.log(`\n${colors.cyan}Model:${colors.reset} ${XAI_MODEL}`);
+        console.log(`${colors.cyan}API:${colors.reset} ${process.env.XAI_API_KEY ? 'Connected' : 'Not configured'}`);
+        return;
+      default:
+        console.log(`\n${colors.red}Unknown command: /${cmd}${colors.reset}`);
+        console.log(`${colors.dim}Type /help for available commands.${colors.reset}`);
+        return;
     }
   }
-  else if (args.includes('--drug-discovery')) {
-    const targetGene = args.find(arg => arg.startsWith('--target='))?.split('=')[1] || 'EGFR';
-    const cancerType = args.find(arg => arg.startsWith('--cancer='))?.split('=')[1] || 'Lung';
-    
-    console.log(`\n🧬 药物靶点发现: ${targetGene} (${cancerType} 癌症)`);
-    console.log('──────────────────────────────');
-    
-    try {
-      const result = await cancerTreatment.discoverDrugTargets(cancerType, targetGene);
-      console.log('🔬 发现结果:');
-      console.log(`   • 癌症类型: ${result.cancerType}`);
-      console.log(`   • 目标基因: ${result.targetGene || '所有'}`);
-      console.log(`   • 发现靶点: ${result.discoveredTargets.length} 个`);
-      
-      result.discoveredTargets.slice(0, 3).forEach((target: any, index: number) => {
-        console.log(`   ${index + 1}. ${target.gene} - ${target.evidenceLevel} 证据等级`);
-        console.log(`      相关癌症: ${target.cancerTypes.join(', ')}`);
-      });
-      
-      console.log('✅ 药物靶点发现完成');
-    } catch (error) {
-      console.error('❌ 药物发现失败:', error);
+
+  // AI-powered response for natural language
+  console.log(`\n${colors.dim}Thinking...${colors.reset}`);
+  const response = await callXAI(input);
+  console.log(`\n${response}`);
+}
+
+async function analyzePatient(patientId: string, includeGenomics: boolean = false): Promise<void> {
+  console.log(`\n${colors.cyan}Analyzing patient ${patientId}...${colors.reset}\n`);
+
+  try {
+    const result = await cancerTreatment.analyzePatient(patientId, includeGenomics);
+
+    console.log(`${colors.bold}Patient Analysis${colors.reset}`);
+    console.log(`${colors.dim}─────────────────────────────${colors.reset}`);
+    console.log(`  Cancer Type:    ${colors.yellow}${result.analysis.cancerType}${colors.reset}`);
+    console.log(`  Stage:          ${colors.yellow}${result.analysis.stage}${colors.reset}`);
+    console.log(`  Biomarkers:     ${result.analysis.biomarkers.join(', ')}`);
+    console.log(`  Survival Prob:  ${colors.green}${(result.analysis.riskAssessment.survivalProbability * 100).toFixed(1)}%${colors.reset}`);
+    console.log(`  Recurrence:     ${colors.yellow}${(result.analysis.riskAssessment.recurrenceRisk * 100).toFixed(1)}%${colors.reset}`);
+
+    if (result.analysis.genomicProfile) {
+      console.log(`\n${colors.bold}Genomic Profile${colors.reset}`);
+      console.log(`  Mutations:      ${result.analysis.genomicProfile.mutations?.length || 0} detected`);
     }
+
+    console.log(`\n${colors.green}✓ Analysis complete${colors.reset}`);
+  } catch (error) {
+    console.error(`${colors.red}✗ Analysis failed:${colors.reset}`, error);
   }
-  else if (args.includes('--demo')) {
-    console.log('\n🏥 癌症治疗框架演示');
-    console.log('────────────────────');
-    
-    try {
-      const { demonstrateCancerTreatmentFramework } = await import('../examples/cancerTreatmentDemo.js');
-      await demonstrateCancerTreatmentFramework();
-    } catch (error) {
-      console.error('❌ 演示失败:', error);
+}
+
+async function designTreatmentPlan(patientId: string, protocolId?: string): Promise<void> {
+  console.log(`\n${colors.cyan}Designing treatment plan for ${patientId}...${colors.reset}\n`);
+
+  try {
+    const result = await cancerTreatment.designTreatmentPlan(patientId, protocolId);
+
+    console.log(`${colors.bold}Treatment Plan${colors.reset}`);
+    console.log(`${colors.dim}─────────────────────────────${colors.reset}`);
+    console.log(`  Protocol:       ${colors.yellow}${result.plan.protocol.name}${colors.reset}`);
+    console.log(`  Modalities:     ${result.plan.protocol.treatmentModalities.join(', ')}`);
+    console.log(`  Est. Efficacy:  ${colors.green}${(result.estimatedEfficacy * 100).toFixed(1)}%${colors.reset}`);
+    console.log(`  Duration:       ${result.plan.treatmentTimeline.length} weeks`);
+
+    console.log(`\n${colors.bold}Timeline${colors.reset}`);
+    result.plan.treatmentTimeline.slice(0, 4).forEach((week: any, i: number) => {
+      console.log(`  Week ${i + 1}: ${week.phase} - ${week.activities.join(', ')}`);
+    });
+    if (result.plan.treatmentTimeline.length > 4) {
+      console.log(`  ${colors.dim}... and ${result.plan.treatmentTimeline.length - 4} more weeks${colors.reset}`);
     }
+
+    console.log(`\n${colors.green}✓ Plan generated${colors.reset}`);
+  } catch (error) {
+    console.error(`${colors.red}✗ Plan generation failed:${colors.reset}`, error);
   }
-  else if (args.length === 0) {
-    // Interactive mode
-    console.log('\n🏥 AGI 癌症治疗框架');
-    console.log('══════════════════════════════════════');
-    console.log('\n🔧 可用命令:');
-    console.log('  agi --cancer-treatment             激活癌症治疗框架');
-    console.log('  agi --analyze-patient --patient=P001 分析患者数据');
-    console.log('  agi --treatment-plan --patient=P001 设计治疗计划');
-    console.log('  agi --drug-discovery --target=EGFR 药物靶点发现');
-    console.log('  agi --clinical-trial               临床试验演示');
-    console.log('  agi --demo                        完整框架演示');
-    console.log('  agi --help                        显示帮助信息');
-    console.log('  agi --version                     显示版本信息');
-    console.log('\n📚 了解更多:');
-    console.log('  查看 README.md 获取完整文档和使用指南');
+}
+
+async function discoverTargets(gene: string, cancerType: string): Promise<void> {
+  console.log(`\n${colors.cyan}Discovering drug targets for ${gene} in ${cancerType} cancer...${colors.reset}\n`);
+
+  try {
+    const result = await cancerTreatment.discoverDrugTargets(cancerType, gene);
+
+    console.log(`${colors.bold}Drug Target Discovery${colors.reset}`);
+    console.log(`${colors.dim}─────────────────────────────${colors.reset}`);
+    console.log(`  Cancer Type:    ${colors.yellow}${result.cancerType}${colors.reset}`);
+    console.log(`  Target Gene:    ${colors.yellow}${result.targetGene || 'All'}${colors.reset}`);
+    console.log(`  Targets Found:  ${result.discoveredTargets.length}`);
+
+    console.log(`\n${colors.bold}Top Targets${colors.reset}`);
+    result.discoveredTargets.slice(0, 5).forEach((target: any, i: number) => {
+      console.log(`  ${i + 1}. ${colors.cyan}${target.gene}${colors.reset} - ${target.evidenceLevel} evidence`);
+      console.log(`     ${colors.dim}Cancers: ${target.cancerTypes.slice(0, 3).join(', ')}${colors.reset}`);
+    });
+
+    console.log(`\n${colors.green}✓ Discovery complete${colors.reset}`);
+  } catch (error) {
+    console.error(`${colors.red}✗ Discovery failed:${colors.reset}`, error);
+  }
+}
+
+async function runDemo(): Promise<void> {
+  console.log(`\n${colors.cyan}Running framework demo...${colors.reset}\n`);
+
+  try {
+    const { demonstrateCancerTreatmentFramework } = await import('../examples/cancerTreatmentDemo.js');
+    await demonstrateCancerTreatmentFramework();
+  } catch (error) {
+    console.error(`${colors.red}✗ Demo failed:${colors.reset}`, error);
+  }
+}
+
+async function handleCommand(args: string[]): Promise<void> {
+  cancerTreatment = new CancerTreatmentCapabilityModule();
+
+  if (args.includes('--analyze-patient')) {
+    const patientId = args.find(a => a.startsWith('--patient='))?.split('=')[1] || 'P001';
+    await analyzePatient(patientId, args.includes('--genomics'));
+  }
+  else if (args.includes('--treatment-plan')) {
+    const patientId = args.find(a => a.startsWith('--patient='))?.split('=')[1] || 'P001';
+    const protocolId = args.find(a => a.startsWith('--protocol='))?.split('=')[1];
+    await designTreatmentPlan(patientId, protocolId);
+  }
+  else if (args.includes('--drug-discovery')) {
+    const gene = args.find(a => a.startsWith('--target='))?.split('=')[1] || 'EGFR';
+    const cancer = args.find(a => a.startsWith('--cancer='))?.split('=')[1] || 'Lung';
+    await discoverTargets(gene, cancer);
+  }
+  else if (args.includes('--demo')) {
+    await runDemo();
   }
   else {
-    console.log('❌ 未知命令。使用 agi --help 查看可用命令。');
+    console.log(`${colors.red}Unknown command.${colors.reset} Use cure --help for usage.`);
     process.exit(1);
   }
 }
 
+function printBanner(): void {
+  console.log(`
+${colors.cyan}${colors.bold}  ╔═══════════════════════════════════════════════════════════╗
+  ║                                                             ║
+  ║   ██████╗██╗   ██╗██████╗ ███████╗                         ║
+  ║  ██╔════╝██║   ██║██╔══██╗██╔════╝                         ║
+  ║  ██║     ██║   ██║██████╔╝█████╗                           ║
+  ║  ██║     ██║   ██║██╔══██╗██╔══╝                           ║
+  ║  ╚██████╗╚██████╔╝██║  ██║███████╗                         ║
+  ║   ╚═════╝ ╚═════╝ ╚═╝  ╚═╝╚══════╝                         ║
+  ║                                                             ║
+  ║   AI Cancer Treatment Framework                             ║
+  ║   Powered by xAI Grok ${XAI_MODEL}              ║
+  ║                                                             ║
+  ╚═══════════════════════════════════════════════════════════╝${colors.reset}
+
+  ${colors.dim}v${VERSION}${colors.reset}
+`);
+}
+
+function printInteractiveHelp(): void {
+  console.log(`
+${colors.bold}Commands${colors.reset}
+${colors.dim}─────────────────────────────${colors.reset}
+  ${colors.cyan}/analyze${colors.reset} [patient]     Analyze patient data
+  ${colors.cyan}/plan${colors.reset} [patient]        Design treatment plan
+  ${colors.cyan}/discover${colors.reset} [gene]       Drug target discovery
+  ${colors.cyan}/demo${colors.reset}                  Run framework demo
+  ${colors.cyan}/model${colors.reset}                 Show AI model info
+  ${colors.cyan}/clear${colors.reset}                 Clear conversation
+  ${colors.cyan}/help${colors.reset}                  Show this help
+  ${colors.cyan}/exit${colors.reset}                  Exit
+
+${colors.bold}AI Chat${colors.reset}
+${colors.dim}─────────────────────────────${colors.reset}
+  Just type naturally to chat with the AI oncologist:
+
+  "What are the treatment options for EGFR+ lung cancer?"
+  "Explain pembrolizumab mechanism of action"
+  "What biomarkers predict response to immunotherapy?"
+  "Compare osimertinib vs erlotinib for EGFR mutations"
+`);
+}
+
 function printHelp(): void {
   console.log(`
-🏥 AGI 癌症治疗框架 - 命令行工具
-
-使用方法:
-  agi [命令] [选项]
-
-	命令:
-	  --cancer-treatment, --medical     激活癌症治疗框架
-	  --analyze-patient                 分析患者数据
-	  --treatment-plan                  设计个性化治疗计划
-	  --drug-discovery                  AI驱动的药物靶点发现
-	  --demo                           完整框架演示
-	  --help, -h                       显示此帮助信息
-	  --version, -v                    显示版本信息
-
-选项:
-  --patient=<id>                   患者ID (默认: P001)
-  --protocol=<id>                  治疗协议ID
-  --genomics                       包含基因组数据分析
-  --target=<gene>                  目标基因 (默认: EGFR)
-  --cancer=<type>                  癌症类型 (默认: Lung)
-
-示例:
-  agi --cancer-treatment
-  agi --analyze-patient --patient=P001 --genomics
-  agi --treatment-plan --patient=P001 --protocol=nccn-breast-early
-  agi --drug-discovery --target=EGFR --cancer=Lung
-  agi --demo
-
-📞 技术支持: medical-support@agi-cancer-treatment.org
-📚 文档: 查看项目目录中的 README.md
+${colors.bold}Cure - AI Cancer Treatment Framework${colors.reset}
+Powered by xAI ${XAI_MODEL}
+
+${colors.bold}Usage:${colors.reset}
+  cure                    Launch interactive AI chat
+  cure [command]          Run a specific command
+
+${colors.bold}Commands:${colors.reset}
+  --analyze-patient       Analyze patient data
+  --treatment-plan        Design treatment plan
+  --drug-discovery        Drug target discovery
+  --demo                  Run framework demo
+  --help, -h              Show this help
+  --version, -v           Show version
+
+${colors.bold}Options:${colors.reset}
+  --patient=<id>          Patient ID (default: P001)
+  --protocol=<id>         Treatment protocol
+  --genomics              Include genomic analysis
+  --target=<gene>         Target gene (default: EGFR)
+  --cancer=<type>         Cancer type (default: Lung)
+
+${colors.bold}Environment:${colors.reset}
+  XAI_API_KEY             xAI API key for AI-powered responses
+
+${colors.bold}Examples:${colors.reset}
+  cure
+  cure --analyze-patient --patient=P001 --genomics
+  cure --drug-discovery --target=BRAF --cancer=Melanoma
+
+${colors.dim}https://npmjs.com/package/@erosolaraijs/cure${colors.reset}
 `);
 }
 
-// Run main function
 main().catch((error) => {
-  console.error('应用程序错误:', error);
+  console.error(`${colors.red}Error:${colors.reset}`, error);
   process.exit(1);
-});
+});