┌─────────────────────────────────────────────────────────────────────────────┐
│                    TOKEN DISTRIBUTION STAGES                                │
├─────────────────────────────────────────────────────────────────────────────┤
│                                                                             │
│  SEED ROUND                                                                 │
│  ┌──────────────────────────────────────────────────────────────────────┐   │
│  │ • Early-stage investors                                             │   │
│  │ • Large discount                                                     │   │
│  │ • Long vesting periods (12-24 months)                               │   │
│  │ • Small allocation (~1-5%)                                           │   │
│  └──────────────────────────────────────────────────────────────────────┘   │
│                                    │                                        │
│                                    v                                        │
│  PRIVATE SALE                                                               │
│  ┌──────────────────────────────────────────────────────────────────────┐   │
│  │ • VC firms, strategic partners                                      │   │
│  │ • Discounted price                                                   │   │
│  │ • Medium vesting (6-12 months)                                      │   │
│  │ • Medium allocation (~5-15%)                                         │   │
│  └──────────────────────────────────────────────────────────────────────┘   │
│                                    │                                        │
│                                    v                                        │
│  PUBLIC SALE (ICO/IDO/IEO)                                                 │
│  ┌──────────────────────────────────────────────────────────────────────┐   │
│  │ • General public                                                   │   │
│  │ • Market price                                                       │   │
│  │ • Limited or no vesting                                             │   │
│  │ • Variable allocation                                                │   │
│  └──────────────────────────────────────────────────────────────────────┘   │
│                                    │                                        │
│                                    v                                        │
│  COMMUNITY & ECOSYSTEM                                                     │
│  ┌──────────────────────────────────────────────────────────────────────┐   │
│  │ • Airdrops                                                          │   │
│  │ • Staking rewards                                                   │   │
│  │ • Liquidity mining                                                  │   │
│  │ • Governance participation                                          │   │
│  └──────────────────────────────────────────────────────────────────────┘   │
│                                    │                                        │
│                                    v                                        │
│  TEAM & ADVISORS                                                          │
│  ┌──────────────────────────────────────────────────────────────────────┐   │
│  │ • Core team                                                         │   │
│  │ • Advisors                                                          │   │
│  │ • Long vesting (24-48 months)                                       │   │
│  │ • Significant allocation (~15-25%)                                  │   │
│  └──────────────────────────────────────────────────────────────────────┘   │
│                                                                             │
└─────────────────────────────────────────────────────────────────────────────┘

┌─────────────────────────────────────────────────────────────────────────────┐
│                    VESTING MODELS                                           │
├─────────────────────────────────────────────────────────────────────────────┤
│                                                                             │
│  1. LINEAR VESTING                                                          │
│  ┌──────────────────────────────────────────────────────────────────────┐   │
│  │ • Tokens released evenly over time                                   │   │
│  │ • Predictable, simple                                               │   │
│  │ • Example: 1/24 per month for 24 months                             │   │
│  └──────────────────────────────────────────────────────────────────────┘   │
│                                                                             │
│  2. CLIFF VESTING                                                           │
│  ┌──────────────────────────────────────────────────────────────────────┐   │
│  │ • No tokens released until cliff date                                │   │
│  │ • Large release at cliff date                                       │   │
│  │ • Example: 25% after 6 months, then linear                          │   │
│  └──────────────────────────────────────────────────────────────────────┘   │
│                                                                             │
│  3. GRADUAL VESTING                                                         │
│  ┌──────────────────────────────────────────────────────────────────────┐   │
│  │ • Increasing release rate over time                                 │   │
│  │ • Rewards long-term commitment                                      │   │
│  │ • Example: 10% first year, 20% second year                          │   │
│  └──────────────────────────────────────────────────────────────────────┘   │
│                                                                             │
│  4. PERFORMANCE-BASED VESTING                                               │
│  ┌──────────────────────────────────────────────────────────────────────┐   │
│  │ • Release tied to milestones                                         │   │
│  │ • Aligns with project goals                                         │   │
│  │ • Example: 10% per major protocol upgrade                          │   │
│  └──────────────────────────────────────────────────────────────────────┘   │
│                                                                             │
└─────────────────────────────────────────────────────────────────────────────┘

# ===================================================================
# MODULE 6, LESSON 2: TOKEN DISTRIBUTION AND ALLOCATION
# ===================================================================

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
from typing import Dict, List
import warnings
warnings.filterwarnings('ignore')

print("="*70)
print("TOKEN DISTRIBUTION AND ALLOCATION")
print("="*70)

# ----------------------------------------------------------------
# PART A: VESTING SCHEDULE SIMULATION
# -----------------------------------------------------------------

print("\n" + "-"*60)
print("PART A: Vesting Schedule Simulation")
print("-"*60)

class VestingSchedule:
    """
    Simulated token vesting schedule.
    """
    def __init__(self, total_tokens: int, cliff_months: int, vesting_months: int,
                 tge_percentage: float = 0.0):
        self.total_tokens = total_tokens
        self.cliff_months = cliff_months
        self.vesting_months = vesting_months
        self.tge_percentage = tge_percentage
        self.released = 0
        self.vested = []
        self.schedule = []
    
    def calculate_vesting(self, months: int) -> Dict:
        """Calculate vesting at a given month."""
        if months < self.cliff_months:
            return {'released': 0, 'vested': 0, 'lock': self.total_tokens}
        
        # TGE release
        tge_tokens = self.total_tokens * self.tge_percentage
        
        # Linear vesting after cliff
        months_after_cliff = months - self.cliff_months
        if months_after_cliff >= self.vesting_months:
            tge_tokens = self.total_tokens
        else:
            remaining_tokens = self.total_tokens - tge_tokens
            monthly_vest = remaining_tokens / self.vesting_months
            vested_after_cliff = monthly_vest * months_after_cliff
            vested = tge_tokens + vested_after_cliff
        
        return {
            'released': min(vested, self.total_tokens),
            'vested': min(vested, self.total_tokens),
            'lock': max(0, self.total_tokens - min(vested, self.total_tokens))
        }
    
    def generate_schedule(self, total_months: int) -> pd.DataFrame:
        """Generate a full vesting schedule."""
        data = []
        for month in range(total_months + 1):
            result = self.calculate_vesting(month)
            data.append({
                'month': month,
                'released': result['released'],
                'vested': result['vested'],
                'lock': result['lock'],
                'percentage_released': result['released'] / self.total_tokens * 100
            })
        self.schedule = data
        return pd.DataFrame(data)

# Simulate different vesting schedules
print("Vesting Schedule Simulations:")

# Schedule 1: Standard team vesting
team_vesting = VestingSchedule(
    total_tokens=1000000,
    cliff_months=12,
    vesting_months=24,
    tge_percentage=0.0
)

# Schedule 2: Investor vesting
investor_vesting = VestingSchedule(
    total_tokens=1000000,
    cliff_months=6,
    vesting_months=12,
    tge_percentage=0.25
)

# Schedule 3: Community vesting
community_vesting = VestingSchedule(
    total_tokens=1000000,
    cliff_months=0,
    vesting_months=6,
    tge_percentage=0.50
)

# Generate schedules
schedules = {
    'Team': team_vesting,
    'Investor': investor_vesting,
    'Community': community_vesting
}

for name, schedule in schedules.items():
    df = schedule.generate_schedule(36)
    print(f"\n{name} Vesting Schedule (36 months):")
    print(f"  Total Tokens: {schedule.total_tokens:,}")
    print(f"  Cliff: {schedule.cliff_months} months")
    print(f"  Vesting: {schedule.vesting_months} months")
    print(f"  TGE Release: {schedule.tge_percentage:.0%}")
    print(f"  Released at 36 months: {df.iloc[-1]['percentage_released']:.0f}%")

# Visualise vesting schedules
fig, ax = plt.subplots(figsize=(12, 6))

for name, schedule in schedules.items():
    df = schedule.generate_schedule(36)
    ax.plot(df['month'], df['percentage_released'], label=name, linewidth=2)

ax.set_xlabel('Months Since TGE')
ax.set_ylabel('Percentage of Tokens Released')
ax.set_title('Token Vesting Comparison')
ax.legend()
ax.grid(True, alpha=0.3)

plt.tight_layout()
plt.savefig('vesting_schedules.png', dpi=300, bbox_inches='tight')
plt.show()
print("Vesting schedule chart saved as 'vesting_schedules.png'")

# ----------------------------------------------------------------
# PART B: TOKEN ALLOCATION MODELS
# -----------------------------------------------------------------

print("\n" + "-"*60)
print("PART B: Token Allocation Models")
print("-"*60)

allocation_models = {
    "Balanced": {
        'Team': 20, 'Advisors': 5, 'Seed': 8, 'Private': 15,
        'Public': 12, 'Community': 25, 'Treasury': 15
    },
    "Community-Focused": {
        'Team': 12, 'Advisors': 3, 'Seed': 5, 'Private': 10,
        'Public': 15, 'Community': 40, 'Treasury': 15
    },
    "Investor-Focused": {
        'Team': 25, 'Advisors': 5, 'Seed': 12, 'Private': 22,
        'Public': 10, 'Community': 16, 'Treasury': 10
    },
    "Decentralised": {
        'Team': 10, 'Advisors': 2, 'Seed': 3, 'Private': 5,
        'Public': 20, 'Community': 45, 'Treasury': 15
    }
}

# Create comparison dataframe
comparison_data = []
for model_name, allocations in allocation_models.items():
    row = {'Model': model_name}
    row.update(allocations)
    comparison_data.append(row)

comparison_df = pd.DataFrame(comparison_data)
print("Token Allocation Models:")
print(comparison_df.to_string(index=False))

# Visualise allocation models
fig, axes = plt.subplots(2, 2, figsize=(14, 10))

for idx, (model_name, allocations) in enumerate(allocation_models.items()):
    ax = axes[idx // 2, idx % 2]
    labels = list(allocations.keys())
    values = list(allocations.values())
    colors = ['#ff6b6b', '#ffd93d', '#6bcb77', '#4d96ff', '#9b59b6', '#1abc9c', '#f39c12']
    ax.pie(values, labels=labels, autopct='%1.1f%%', startangle=90, colors=colors[:len(labels)])
    ax.set_title(f'{model_name} Allocation Model')

plt.tight_layout()
plt.savefig('token_allocation_models.png', dpi=300, bbox_inches='tight')
plt.show()
print("Allocation model chart saved as 'token_allocation_models.png'")

# ----------------------------------------------------------------
# PART C: DISTRIBUTION METHOD COMPARISON
# -----------------------------------------------------------------

print("\n" + "-"*60)
print("PART C: Distribution Method Comparison")
print("-"*60)

method_data = {
    'Method': ['ICO', 'IDO', 'IEO', 'Airdrop', 'Mining', 'Staking', 'Liquidity Mining'],
    'Centralisation': ['High', 'Medium', 'High', 'Low', 'Very Low', 'Low', 'Low'],
    'Cost': ['High', 'Medium', 'Medium', 'Low', 'High', 'Medium', 'Medium'],
    'Access': ['Limited', 'Open', 'Limited', 'Open', 'Open', 'Open', 'Open'],
    'Regulatory Risk': ['High', 'Medium', 'Medium', 'Low', 'Low', 'Medium', 'Medium'],
    'Community Building': ['Low', 'Medium', 'Medium', 'High', 'High', 'High', 'High']
}

method_df = pd.DataFrame(method_data)
print(method_df.to_string(index=False))

# ----------------------------------------------------------------
# PART D: STAKEHOLDER INCENTIVE ALIGNMENT
# -----------------------------------------------------------------

print("\n" + "-"*60)
print("PART D: Stakeholder Incentive Alignment")
print("-"*60)

incentive_data = {
    'Stakeholder': ['Team', 'Advisors', 'Seed Investors', 'Private Investors', 'Public', 'Community'],
    'Primary Goal': ['Project Success', 'Guidance', 'Return on Investment', 'Return on Investment', 'Value', 'Participation'],
    'Time Horizon': ['Long', 'Medium', 'Medium-Long', 'Medium', 'Short-Medium', 'Long'],
    'Risk Tolerance': ['High', 'Medium', 'High', 'Medium', 'Low-Medium', 'Medium'],
    'Key Incentive': ['Tokens + Equity', 'Tokens', 'Tokens', 'Tokens', 'Tokens', 'Governance + Rewards']
}

incentive_df = pd.DataFrame(incentive_data)
print(incentive_df.to_string(index=False))

# ----------------------------------------------------------------
# PART E: SUMMARY AND RECOMMENDATIONS
# -----------------------------------------------------------------

print("\n" + "="*70)
print("PART E: Summary and Recommendations")
print("="*70)

print("""
Token Distribution and Allocation – Key Takeaways:

1. Token distribution stages: seed, private, public, community, team.
2. Vesting schedules: cliff period, vesting period, release schedule.
3. Vesting models: linear, cliff, gradual, performance-based.
4. Distribution methods: ICO, IDO, IEO, airdrops, mining, staking, liquidity mining.
5. Fair launch: no pre-allocation, decentralised distribution.
6. Pre-mined launch: pre-allocated tokens, faster growth.
7. Stakeholder allocation: team (15-25%), investors (20-30%), community (15-30%), treasury (10-20%).

Recommendations:
  - Design fair and transparent token distribution.
  - Implement appropriate vesting schedules for each stakeholder.
  - Align incentives across all stakeholders.
  - Consider decentralisation in allocation decisions.
  - Be transparent about allocation and vesting.
  - Monitor token concentration and distribution.
  - Engage with the community on distribution decisions.
""")