ROI Optimization for AI Implementation

class AIROICalculator:
    def __init__(self):
        self.costs = {
            'development': 0,
            'infrastructure': 0,
            'data_acquisition': 0,
            'training': 0,
            'maintenance': 0,
            'opportunity_cost': 0
        }
        
        self.benefits = {
            'cost_savings': 0,
            'revenue_increase': 0,
            'productivity_gains': 0,
            'risk_reduction': 0,
            'strategic_value': 0
        }
        
        self.timeline_months = 12
    
    def add_cost(self, category, amount, is_recurring=False):
        """Add cost to the calculation"""
        if is_recurring:
            amount *= self.timeline_months
        self.costs[category] += amount
    
    def add_benefit(self, category, amount, is_recurring=False):
        """Add benefit to the calculation"""
        if is_recurring:
            amount *= self.timeline_months
        self.benefits[category] += amount
    
    def calculate_roi(self):
        """Calculate ROI percentage"""
        total_costs = sum(self.costs.values())
        total_benefits = sum(self.benefits.values())
        
        if total_costs == 0:
            return float('inf')
        
        roi = ((total_benefits - total_costs) / total_costs) * 100
        return roi
    
    def calculate_payback_period(self, monthly_net_benefit):
        """Calculate payback period in months"""
        total_costs = sum(self.costs.values())
        if monthly_net_benefit <= 0:
            return float('inf')
        return total_costs / monthly_net_benefit
    
    def generate_report(self):
        """Generate comprehensive ROI report"""
        total_costs = sum(self.costs.values())
        total_benefits = sum(self.benefits.values())
        roi = self.calculate_roi()
        
        return {
            'total_costs': total_costs,
            'total_benefits': total_benefits,
            'net_benefit': total_benefits - total_costs,
            'roi_percentage': roi,
            'cost_breakdown': self.costs,
            'benefit_breakdown': self.benefits,
            'timeline_months': self.timeline_months
        }

# Example usage
calculator = AIROICalculator()

# Add costs
calculator.add_cost('development', 150000)  # One-time
calculator.add_cost('infrastructure', 5000, is_recurring=True)  # Monthly
calculator.add_cost('data_acquisition', 25000)  # One-time
calculator.add_cost('training', 30000)  # One-time
calculator.add_cost('maintenance', 8000, is_recurring=True)  # Monthly

# Add benefits
calculator.add_benefit('cost_savings', 15000, is_recurring=True)  # Monthly
calculator.add_benefit('revenue_increase', 25000, is_recurring=True)  # Monthly
calculator.add_benefit('productivity_gains', 10000, is_recurring=True)  # Monthly

# Calculate ROI
report = calculator.generate_report()
print(f"AI Project ROI: {report['roi_percentage']:.2f}%")
print(f"Net Benefit: {report['net_benefit']:,.2f}")
print(f"Payback Period: {calculator.calculate_payback_period(50000):.1f} months")