bizy-ai 1.0.1__py3-none-any.whl → 1.1.0__py3-none-any.whl

agent/cli.py

@@ -2,6 +2,8 @@
 """Business Agent CLI Tool"""
 
 import click
+import os
+import sys
 from agent.tasks import TaskManager
 from agent.planner import BusinessPlanner
 from agent.research import ResearchAgent
@@ -9,6 +11,7 @@ from rich.console import Console
 from rich.table import Table
 from rich.panel import Panel
 from rich.markdown import Markdown
+from rich.prompt import Prompt, Confirm
 from datetime import datetime, timedelta
 from dotenv import load_dotenv
 
@@ -32,42 +35,172 @@ def task():
 @click.option('--priority', '-p', type=int, default=3)
 @click.option('--category', '-c', help='Task category')
 @click.option('--hours', '-h', type=float, help='Estimated hours')
-def add(title, description, priority, category, hours):
+@click.option('--goal', '-g', type=int, help='Goal ID to assign task to')
+def add(title, description, priority, category, hours, goal):
     """Add a new task"""
     task_mgr = TaskManager()
+    planner = BusinessPlanner()
+
+    goal_id = goal
+
+    # If no goal specified, prompt user to select or create one
+    if goal_id is None:
+        goals = planner.get_active_goals()
+
+        if goals:
+            console.print("\n[bold]📋 Available Goals:[/bold]")
+            goals_table = Table(show_header=True, header_style="bold cyan")
+            goals_table.add_column("ID", style="dim", width=6)
+            goals_table.add_column("Goal")
+            goals_table.add_column("Progress", justify="right", width=15)
+
+            for g in goals[:10]:  # Show max 10 goals
+                progress_bar = "█" * int(g.progress_percentage / 10) + "░" * (10 - int(g.progress_percentage / 10))
+                goals_table.add_row(
+                    str(g.id),
+                    g.title[:50],
+                    f"{progress_bar} {g.progress_percentage:.0f}%"
+                )
+            console.print(goals_table)
+
+            # Prompt user
+            console.print("\n[bold]Options:[/bold]")
+            console.print("  • Enter [cyan]goal ID[/cyan] to assign task to that goal")
+            console.print("  • Press [dim]Enter[/dim] to skip (create task without goal)")
+            console.print("  • Type [yellow]'new'[/yellow] to create a new goal")
+
+            choice = Prompt.ask("\n[bold]Your choice[/bold]", default="")
+
+            if choice.lower() == 'new':
+                # Create new goal interactively
+                console.print("\n[bold cyan]Creating New Goal[/bold cyan]")
+                goal_title = Prompt.ask("Goal title")
+                goal_horizon = Prompt.ask(
+                    "Horizon",
+                    choices=["daily", "weekly", "monthly", "quarterly", "yearly"],
+                    default="monthly"
+                )
+
+                try:
+                    new_goal = planner.create_goal(
+                        title=goal_title,
+                        horizon=goal_horizon
+                    )
+                    goal_id = new_goal.id
+                    console.print(f"[green]✓[/green] Created goal: {new_goal.title} (ID: {new_goal.id})\n")
+                except Exception as e:
+                    console.print(f"[red]✗[/red] Error creating goal: {e}")
+                    console.print("[yellow]Creating task without goal assignment[/yellow]\n")
+                    goal_id = None
+            elif choice and choice.isdigit():
+                goal_id = int(choice)
+                # Validate goal exists
+                goal_obj = planner.get_goal(goal_id)
+                if not goal_obj:
+                    console.print(f"[yellow]⚠[/yellow]  Goal #{goal_id} not found. Creating task without goal.\n")
+                    goal_id = None
+        else:
+            console.print("\n[yellow]No active goals found.[/yellow]")
+            if Confirm.ask("Would you like to create a new goal?", default=False):
+                console.print("\n[bold cyan]Creating New Goal[/bold cyan]")
+                goal_title = Prompt.ask("Goal title")
+                goal_horizon = Prompt.ask(
+                    "Horizon",
+                    choices=["daily", "weekly", "monthly", "quarterly", "yearly"],
+                    default="monthly"
+                )
+
+                try:
+                    new_goal = planner.create_goal(
+                        title=goal_title,
+                        horizon=goal_horizon
+                    )
+                    goal_id = new_goal.id
+                    console.print(f"[green]✓[/green] Created goal: {new_goal.title} (ID: {new_goal.id})\n")
+                except Exception as e:
+                    console.print(f"[red]✗[/red] Error creating goal: {e}\n")
+                    goal_id = None
+
+    # Create task with goal assignment
     task = task_mgr.create_task(
         title=title,
         description=description,
         priority=priority,
         category=category,
-        estimated_hours=hours
+        estimated_hours=hours,
+        parent_goal_id=goal_id
     )
+
     console.print(f"[green]✓[/green] Task created: {task.title} (ID: {task.id})")
+    if goal_id:
+        console.print(f"[dim]  Assigned to goal #{goal_id}[/dim]")
+        # Recalculate goal progress
+        planner.calculate_goal_progress(goal_id)
+
     task_mgr.close()
+    planner.close()
 
 @task.command()
-def list():
-    """List all pending tasks"""
+@click.option('--filter', '-f', type=click.Choice(['all', 'pending', 'completed', 'today']), default='all', help='Filter tasks')
+@click.option('--goal', '-g', type=int, help='Filter by goal ID')
+def list(filter, goal):
+    """List tasks with optional filters"""
     task_mgr = TaskManager()
-    tasks = task_mgr.get_tasks_for_today()
+
+    # Get tasks based on filter
+    if filter == 'all':
+        from agent.models import Task
+        if goal:
+            tasks = task_mgr.session.query(Task).filter_by(parent_goal_id=goal).all()
+        else:
+            tasks = task_mgr.session.query(Task).all()
+    elif filter == 'completed':
+        from agent.models import Task
+        query = task_mgr.session.query(Task).filter_by(status='completed')
+        if goal:
+            query = query.filter_by(parent_goal_id=goal)
+        tasks = query.all()
+    elif filter == 'pending':
+        from agent.models import Task
+        query = task_mgr.session.query(Task).filter(Task.status.in_(['pending', 'in_progress']))
+        if goal:
+            query = query.filter_by(parent_goal_id=goal)
+        tasks = query.all()
+    else:  # today
+        tasks = task_mgr.get_tasks_for_today()
 
     if not tasks:
-        console.print("[yellow]No pending tasks[/yellow]")
+        console.print(f"[yellow]No {filter} tasks found[/yellow]")
+        task_mgr.close()
         return
 
-    table = Table(title="📋 Your Tasks", show_header=True, header_style="bold cyan")
+    # Build title
+    title_parts = ["📋 Your Tasks"]
+    if filter != 'all':
+        title_parts.append(f"({filter.title()})")
+    if goal:
+        title_parts.append(f"for Goal #{goal}")
+
+    table = Table(title=" ".join(title_parts), show_header=True, header_style="bold cyan")
     table.add_column("ID", style="dim")
+    table.add_column("Status", justify="center")
     table.add_column("Priority", justify="center")
     table.add_column("Title")
     table.add_column("Category", style="cyan")
+    table.add_column("Due Date", style="dim")
 
     for task in tasks:
-        priority_str = "🔴" if task.priority == 1 else "🟡" if task.priority == 3 else "🟢"
+        status_icon = "✓" if task.status == 'completed' else "○"
+        priority_str = "🔴" if task.priority == 1 else "🟡" if task.priority == 2 else "🟢"
+        due_date = task.due_date.strftime('%Y-%m-%d') if task.due_date else "-"
+
         table.add_row(
             str(task.id),
+            status_icon,
             priority_str,
             task.title[:50],
-            task.category or "-"
+            task.category or "-",
+            due_date
         )
 
     console.print(table)
@@ -81,6 +214,14 @@ def complete(task_id):
     task = task_mgr.complete_task(task_id)
     if task:
         console.print(f"[green]✓[/green] Completed: {task.title}")
+
+        # Update goal progress if task is associated with a goal
+        if task.parent_goal_id:
+            from agent.planner import BusinessPlanner
+            planner = BusinessPlanner()
+            progress = planner.calculate_goal_progress(task.parent_goal_id)
+            console.print(f"[dim]Goal progress updated: {progress:.1f}%[/dim]")
+            planner.close()
     else:
         console.print(f"[red]✗[/red] Task {task_id} not found")
     task_mgr.close()
@@ -120,8 +261,16 @@ def list():
 
     if not goals:
         console.print("[yellow]No active goals[/yellow]")
+        planner.close()
         return
 
+    # Recalculate progress for all goals
+    for goal in goals:
+        planner.calculate_goal_progress(goal.id)
+
+    # Refresh goals after recalculation
+    goals = planner.get_active_goals()
+
     table = Table(title="🎯 Your Goals", show_header=True, header_style="bold cyan")
     table.add_column("ID", style="dim")
     table.add_column("Title")
@@ -212,16 +361,28 @@ def competitors(domain, offering):
 def stats():
     """Show statistics"""
     task_mgr = TaskManager()
-    weekly_stats = task_mgr.get_weekly_stats()
+    weekly_stats = task_mgr.get_weekly_task_stats()
     velocity = task_mgr.get_task_velocity()
     today_tasks = task_mgr.get_tasks_for_today()
 
     console.print("\n[bold cyan]📊 Your Statistics[/bold cyan]\n")
     console.print("[bold]This Week:[/bold]")
-    console.print(f"  • Tasks Completed: {weekly_stats['total_tasks_completed']}")
-    console.print(f"  • Completion Rate: {weekly_stats['average_completion_rate']:.0%}")
+    console.print(f"  • Tasks Completed: {weekly_stats['tasks_completed_this_week']}")
+    console.print(f"  • Tasks Created: {weekly_stats['tasks_created_this_week']}")
+    console.print(f"  • Completion Rate: {weekly_stats['completion_rate']:.1f}%")
+    console.print(f"  • Total Hours (Estimated): {weekly_stats['total_estimated_hours']:.1f}h")
+    if weekly_stats['total_actual_hours'] > 0:
+        console.print(f"  • Total Hours (Actual): {weekly_stats['total_actual_hours']:.1f}h")
     console.print(f"\n[bold]Velocity:[/bold] {velocity:.1f} tasks/day")
     console.print(f"\n[bold]Today:[/bold] {len(today_tasks)} tasks scheduled\n")
+
+    # Show breakdown by category if available
+    if weekly_stats['categories']:
+        console.print("[bold]By Category:[/bold]")
+        for category, count in sorted(weekly_stats['categories'].items(), key=lambda x: x[1], reverse=True):
+            console.print(f"  • {category}: {count} task(s)")
+        console.print()
+
     task_mgr.close()
 
 # DAILY/WEEKLY REVIEWS
@@ -243,5 +404,40 @@ def weekly():
     from agent.weekly_review import run_weekly_review
     run_weekly_review()
 
+# BUSINESS PLAN MANAGEMENT
+@cli.group()
+def plan():
+    """Manage business plan"""
+    pass
+
+@plan.command()
+def show():
+    """Review current business plan and goals"""
+    from agent.plan_manager import review_business_plan
+    review_business_plan()
+
+@plan.command()
+def create():
+    """Create new business plan"""
+    from agent.plan_manager import create_business_plan
+    create_business_plan()
+
+@plan.command()
+def update():
+    """Update existing business plan"""
+    from agent.plan_manager import update_business_plan
+    update_business_plan()
+
+@plan.command()
+@click.argument('file_path', required=False)
+@click.option('--name', '-n', help='Business plan name')
+def load(file_path, name):
+    """Load business plan from YAML file"""
+    # Import and run the standalone script
+    repo_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+    sys.path.insert(0, repo_root)
+    from scripts.load_business_plan import load_business_plan
+    load_business_plan(file_path, name)
+
 if __name__ == "__main__":
     cli()

agent/models.py

@@ -158,8 +158,9 @@ class BusinessMetric(Base):
 
 class BusinessPlan(Base):
     __tablename__ = 'business_plans'
-    
+
     id = Column(Integer, primary_key=True)
+    name = Column(String(255))
     version = Column(String(50))
     vision = Column(Text)
     mission = Column(Text)
@@ -171,6 +172,7 @@ class BusinessPlan(Base):
     key_partnerships = Column(JSON)
     cost_structure = Column(JSON)
     created_at = Column(DateTime, default=datetime.utcnow)
+    updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
     is_active = Column(Boolean, default=True)
     
     def to_dict(self):
@@ -193,15 +195,28 @@ class BusinessPlan(Base):
 # Database setup
 def get_engine(db_path=None):
     if db_path is None:
-        # Use environment variable or default to home directory
-        db_path = os.getenv('BUSINESS_AGENT_DB', os.path.expanduser('~/.business-agent/tasks.db'))
+        # Determine database path based on environment
+        env = os.getenv('BIZY_ENV', 'production')
+
+        if env == 'test':
+            # Test environment uses in-memory database (overridden by conftest.py)
+            return create_engine('sqlite:///:memory:', echo=False)
+        elif env == 'development':
+            # Development environment uses separate database
+            db_path = os.getenv('BUSINESS_AGENT_DB', os.path.expanduser('~/.business-agent/dev_tasks.db'))
+        else:  # production
+            # Production environment uses main database
+            db_path = os.getenv('BUSINESS_AGENT_DB', os.path.expanduser('~/.business-agent/tasks.db'))
 
-    # Create directory if it doesn't exist
-    db_dir = os.path.dirname(db_path)
-    if db_dir:
-        os.makedirs(db_dir, exist_ok=True)
+    # Create directory if it doesn't exist (not needed for in-memory)
+    if db_path and db_path != ':memory:':
+        db_dir = os.path.dirname(db_path)
+        if db_dir:
+            os.makedirs(db_dir, exist_ok=True)
+        return create_engine(f'sqlite:///{db_path}', echo=False)
 
-    return create_engine(f'sqlite:///{db_path}', echo=False)
+    # In-memory database
+    return create_engine('sqlite:///:memory:', echo=False)
 
 def get_session(engine=None):
     if engine is None:

agent/plan_manager.py

@@ -0,0 +1,320 @@
+#!/usr/bin/env python3
+"""
+Business Plan Management Script
+Review and update business plans, goals, and tasks
+"""
+
+import sys
+import os
+import yaml
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+
+from agent.core import BusinessAgent
+from agent.planner import BusinessPlanner
+from agent.tasks import TaskManager
+from agent.models import get_session, BusinessPlan
+from rich.console import Console
+from rich.panel import Panel
+from rich.table import Table
+from rich.prompt import Prompt, Confirm
+from rich.markdown import Markdown
+from datetime import datetime, timedelta
+from dotenv import load_dotenv
+
+load_dotenv()
+console = Console()
+
+def review_business_plan():
+    """Display comprehensive business plan review"""
+    try:
+        console.print()
+        console.print(f"[bold blue]{'='*70}[/bold blue]")
+        console.print(f"[bold blue]  📋 BUSINESS PLAN REVIEW - {datetime.now().strftime('%B %d, %Y')}[/bold blue]")
+        console.print(f"[bold blue]{'='*70}[/bold blue]")
+        console.print()
+
+        planner = BusinessPlanner()
+        task_mgr = TaskManager()
+        agent = BusinessAgent()
+
+        # Get active business plan
+        plan = planner.get_active_business_plan()
+
+        if not plan:
+            console.print("[yellow]⚠️  No active business plan found.[/yellow]")
+            console.print("\nWould you like to create one? Use: [cyan]bizy plan create[/cyan]")
+            return
+
+        # Display business plan
+        console.print("[bold]📊 Current Business Plan[/bold]")
+        console.print(f"[dim]Version: {plan.version} | Created: {plan.created_at.strftime('%Y-%m-%d')}[/dim]\n")
+
+        plan_info = f"""
+**Vision:** {plan.vision}
+
+**Mission:** {plan.mission}
+
+**Value Proposition:** {plan.value_proposition}
+
+**Target Market:** {plan.target_market}
+
+**Revenue Model:** {plan.revenue_model}
+"""
+        console.print(Panel(Markdown(plan_info), title="Business Plan", border_style="blue"))
+
+        # Get goals
+        console.print("\n[bold]🎯 Goals Overview[/bold]\n")
+        all_goals = planner.get_active_goals()
+
+        # Recalculate progress for all goals
+        for goal in all_goals:
+            planner.calculate_goal_progress(goal.id)
+
+        # Refresh goals after recalculation
+        all_goals = planner.get_active_goals()
+
+        if all_goals:
+            goals_table = Table(show_header=True, header_style="bold cyan")
+            goals_table.add_column("Horizon")
+            goals_table.add_column("Goal")
+            goals_table.add_column("Progress", justify="right")
+            goals_table.add_column("Target Date")
+
+            for goal in all_goals[:10]:
+                progress_bar = "█" * int(goal.progress_percentage / 10) + "░" * (10 - int(goal.progress_percentage / 10))
+                target = goal.target_date.strftime('%Y-%m-%d') if goal.target_date else "Not set"
+                goals_table.add_row(
+                    goal.horizon.upper(),
+                    goal.title[:40],
+                    f"{progress_bar} {goal.progress_percentage:.0f}%",
+                    target
+                )
+
+            console.print(goals_table)
+        else:
+            console.print("[yellow]No active goals found.[/yellow]")
+
+        # Get task summary
+        console.print("\n[bold]📋 Task Summary[/bold]\n")
+        weekly_stats = task_mgr.get_weekly_task_stats()
+        today_tasks = task_mgr.get_tasks_for_today()
+        overdue = task_mgr.get_overdue_tasks()
+
+        console.print(f"  • This Week: {weekly_stats['tasks_completed_this_week']} completed ({weekly_stats['completion_rate']:.1f}% completion rate)")
+        console.print(f"  • Today: {len(today_tasks)} tasks scheduled")
+        console.print(f"  • Overdue: {len(overdue)} tasks")
+
+        # AI Analysis
+        console.print("\n[dim]Generating AI analysis...[/dim]")
+
+        goals_summary = "\n".join([
+            f"- [{g.horizon.upper()}] {g.title}: {g.progress_percentage:.0f}% complete"
+            for g in all_goals[:5]
+        ]) if all_goals else "No active goals"
+
+        analysis_prompt = f"""Analyze this business plan and provide strategic insights:
+
+BUSINESS PLAN:
+- Vision: {plan.vision}
+- Mission: {plan.mission}
+- Value Proposition: {plan.value_proposition}
+
+GOALS PROGRESS:
+{goals_summary}
+
+EXECUTION METRICS:
+- Tasks Completed This Week: {weekly_stats['tasks_completed_this_week']}
+- Tasks Created This Week: {weekly_stats['tasks_created_this_week']}
+- Completion Rate: {weekly_stats['completion_rate']:.1f}%
+- Tasks Overdue: {len(overdue)}
+
+Provide:
+1. **Plan-Goal Alignment** - Are the goals aligned with the business plan?
+2. **Execution Health** - Is the execution on track?
+3. **Recommended Adjustments** - What should be updated or changed?
+4. **Next Strategic Moves** - What are the top 3 priorities?
+
+Be concise and actionable."""
+
+        analysis = agent.client.messages.create(
+            model=agent.model,
+            max_tokens=2000,
+            messages=[{"role": "user", "content": analysis_prompt}]
+        ).content[0].text
+
+        console.print(Panel(
+            Markdown(analysis),
+            title="🤖 AI Strategic Analysis",
+            border_style="blue",
+            padding=(1, 2)
+        ))
+
+        console.print()
+        planner.close()
+        task_mgr.close()
+
+    except Exception as e:
+        console.print(f"[bold red]❌ Error reviewing plan:[/bold red] {e}")
+        import traceback
+        console.print(traceback.format_exc())
+
+def create_business_plan():
+    """Interactive business plan creation"""
+    try:
+        console.print()
+        console.print(f"[bold green]{'='*70}[/bold green]")
+        console.print(f"[bold green]  📝 CREATE BUSINESS PLAN[/bold green]")
+        console.print(f"[bold green]{'='*70}[/bold green]")
+        console.print()
+
+        console.print("[cyan]Let's create your business plan. This will guide all your goals and tasks.[/cyan]\n")
+
+        vision = Prompt.ask("🔭 [bold]What is your vision?[/bold] (Where are you going?)")
+        mission = Prompt.ask("🎯 [bold]What is your mission?[/bold] (How will you get there?)")
+        value_prop = Prompt.ask("💎 [bold]What is your value proposition?[/bold] (Why you over competitors?)")
+        target_market = Prompt.ask("👥 [bold]Who is your target market?[/bold]")
+        revenue_model = Prompt.ask("💰 [bold]What is your revenue model?[/bold] (How will you make money?)")
+
+        version = Prompt.ask("📌 [bold]Version number?[/bold]", default="1.0")
+
+        console.print("\n[dim]Creating business plan...[/dim]")
+
+        planner = BusinessPlanner()
+        plan = planner.create_business_plan(
+            vision=vision,
+            mission=mission,
+            value_proposition=value_prop,
+            target_market=target_market,
+            revenue_model=revenue_model,
+            version=version
+        )
+
+        console.print(Panel(
+            f"[green]✅ Business Plan v{plan.version} Created Successfully![/green]\n\n"
+            f"This plan will now guide your goal setting and task management.",
+            title="Success",
+            border_style="green"
+        ))
+
+        # Ask if they want to create goals
+        if Confirm.ask("\n💡 Would you like to create your first goal now?"):
+            console.print("\n[cyan]Tip: Start with a yearly goal, then break it down into smaller goals.[/cyan]\n")
+
+            goal_title = Prompt.ask("🎯 [bold]Goal title[/bold]")
+            goal_desc = Prompt.ask("📝 [bold]Goal description[/bold]")
+            horizon = Prompt.ask(
+                "⏰ [bold]Time horizon[/bold]",
+                choices=["weekly", "monthly", "quarterly", "yearly"],
+                default="yearly"
+            )
+
+            # Calculate target date
+            days_map = {"weekly": 7, "monthly": 30, "quarterly": 90, "yearly": 365}
+            target_date = datetime.now() + timedelta(days=days_map[horizon])
+
+            goal = planner.create_goal(
+                title=goal_title,
+                description=goal_desc,
+                horizon=horizon,
+                target_date=target_date
+            )
+
+            console.print(f"\n[green]✅ Goal created (ID: {goal.id})[/green]")
+
+            if Confirm.ask("\n🤖 Would you like AI to break this goal into tasks?"):
+                console.print("\n[dim]AI is breaking down your goal into actionable tasks...[/dim]")
+                tasks = planner.break_down_goal(goal.id)
+
+                if tasks:
+                    console.print(f"\n[green]✅ Created {len(tasks)} tasks:[/green]")
+                    for i, task in enumerate(tasks[:5], 1):
+                        console.print(f"  {i}. {task.title}")
+                    if len(tasks) > 5:
+                        console.print(f"  ... and {len(tasks) - 5} more")
+
+        planner.close()
+        console.print()
+
+    except Exception as e:
+        console.print(f"[bold red]❌ Error creating plan:[/bold red] {e}")
+        import traceback
+        console.print(traceback.format_exc())
+
+def update_business_plan():
+    """Update existing business plan"""
+    try:
+        console.print()
+        console.print(f"[bold yellow]{'='*70}[/bold yellow]")
+        console.print(f"[bold yellow]  ✏️  UPDATE BUSINESS PLAN[/bold yellow]")
+        console.print(f"[bold yellow]{'='*70}[/bold yellow]")
+        console.print()
+
+        planner = BusinessPlanner()
+        plan = planner.get_active_business_plan()
+
+        if not plan:
+            console.print("[yellow]⚠️  No active business plan found.[/yellow]")
+            console.print("\nCreate one with: [cyan]bizy plan create[/cyan]")
+            return
+
+        console.print(f"[bold]Current Plan (v{plan.version})[/bold]\n")
+        console.print(f"Vision: {plan.vision}")
+        console.print(f"Mission: {plan.mission}")
+        console.print(f"Value Proposition: {plan.value_proposition}")
+        console.print(f"Target Market: {plan.target_market}")
+        console.print(f"Revenue Model: {plan.revenue_model}")
+        console.print()
+
+        console.print("[cyan]What would you like to update? (Press Enter to skip)[/cyan]\n")
+
+        vision = Prompt.ask("🔭 [bold]Vision[/bold]", default=plan.vision)
+        mission = Prompt.ask("🎯 [bold]Mission[/bold]", default=plan.mission)
+        value_prop = Prompt.ask("💎 [bold]Value Proposition[/bold]", default=plan.value_proposition)
+        target_market = Prompt.ask("👥 [bold]Target Market[/bold]", default=plan.target_market)
+        revenue_model = Prompt.ask("💰 [bold]Revenue Model[/bold]", default=plan.revenue_model)
+
+        # Create new version
+        version_parts = plan.version.split('.')
+        new_minor = int(version_parts[1]) + 1 if len(version_parts) > 1 else 1
+        new_version = f"{version_parts[0]}.{new_minor}"
+
+        new_version = Prompt.ask("📌 [bold]New version number[/bold]", default=new_version)
+
+        if Confirm.ask("\n💾 Save this as a new version of your business plan?"):
+            new_plan = planner.create_business_plan(
+                vision=vision,
+                mission=mission,
+                value_proposition=value_prop,
+                target_market=target_market,
+                revenue_model=revenue_model,
+                version=new_version
+            )
+
+            console.print(Panel(
+                f"[green]✅ Business Plan Updated to v{new_plan.version}[/green]\n\n"
+                f"The previous version (v{plan.version}) has been archived.",
+                title="Success",
+                border_style="green"
+            ))
+
+        planner.close()
+        console.print()
+
+    except Exception as e:
+        console.print(f"[bold red]❌ Error updating plan:[/bold red] {e}")
+        import traceback
+        console.print(traceback.format_exc())
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1:
+        command = sys.argv[1]
+        if command == "review":
+            review_business_plan()
+        elif command == "create":
+            create_business_plan()
+        elif command == "update":
+            update_business_plan()
+        else:
+            console.print(f"[red]Unknown command: {command}[/red]")
+    else:
+        review_business_plan()