from form16x import TaxCalculationAPI, TaxRegime, EnhancedForm16Extractor
from decimal import Decimal

# Initialize API for tax calculations
api = TaxCalculationAPI()

# Calculate tax from Form16 PDF
result = api.calculate_tax_from_form16(
    form16_file="form16_sample.pdf",
    regime=TaxRegime.BOTH,
    bank_interest=Decimal("25000")
)

print(result['recommendation'])  # e.g., "NEW regime saves ₹25,000 annually"

from form16x.form16_parser.extractors.enhanced_form16_extractor import EnhancedForm16Extractor
from form16x.form16_parser.pdf.reader import RobustPDFProcessor

# Initialize extractor and PDF processor
extractor = EnhancedForm16Extractor()
pdf_processor = RobustPDFProcessor()

# Extract tables from PDF
extraction_result = pdf_processor.extract_tables("form16_sample.pdf")
tables = extraction_result.tables

# Extract Form16 data
form16_result = extractor.extract_all(tables)

# Access extracted data
print(form16_result.employee.name)
print(form16_result.salary.gross_salary)
print(form16_result.employer.name)

from form16x.form16_parser.api import TaxCalculationAPI, TaxRegime
from decimal import Decimal

# Initialize API
api = TaxCalculationAPI()

# Calculate tax from Form16 PDF
result = api.calculate_tax_from_form16(
    form16_file="form16_sample.pdf",
    regime=TaxRegime.BOTH,
    bank_interest=Decimal("25000")
)

# Calculate tax from manual input
result = api.calculate_tax_from_input(
    assessment_year="2024-25",
    gross_salary=Decimal("1200000"),
    regime=TaxRegime.BOTH,
    section_80c=Decimal("150000"),
    tds_paid=Decimal("180000")
)

print(result['recommendation'])  # e.g., "NEW regime saves ₹25,000 annually"

from form16x.form16_parser.api.tax_calculation_api import TaxCalculationAPI, TaxRegime, AgeCategoryEnum
from decimal import Decimal

# Initialize the tax calculation API
api = TaxCalculationAPI()

# Example: Software engineer with ₹15 lakhs gross salary
result = api.calculate_tax_from_input(
    assessment_year="2024-25",
    gross_salary=Decimal("1500000"),          # ₹15 lakhs gross salary
    basic_salary=Decimal("600000"),           # ₹6 lakhs basic salary
    hra_received=Decimal("180000"),           # ₹1.8 lakhs HRA received
    bank_interest=Decimal("45000"),           # ₹45,000 bank interest (triggers 80TTA)
    other_income=Decimal("75000"),            # ₹75,000 freelance/other income
    house_property_income=Decimal("120000"),  # ₹1.2 lakhs rental income
    section_80c=Decimal("150000"),            # ₹1.5 lakhs in PPF/ELSS/LIC (max limit)
    section_80ccd_1b=Decimal("50000"),        # ₹50,000 NPS contribution (additional)
    tds_paid=Decimal("185000"),               # ₹1.85 lakhs TDS already deducted
    city_type="metro",                        # Living in metro city (for HRA exemption)
    regime=TaxRegime.BOTH,                    # Calculate both old and new regime
    age_category=AgeCategoryEnum.BELOW_60,    # Below 60 years
    verbose=True
)

# Display results
if result['status'] == 'success':
    print(f"Assessment Year: {result['assessment_year']}")
    print(f"Regimes calculated: {', '.join(result['regimes_calculated'])}")
    print(f"\nRecommendation: {result['recommendation']}")
    
    # Show detailed breakdown for both regimes
    for regime_name, regime_data in result['results'].items():
        print(f"\n{'='*50}")
        print(f"{regime_name.upper()} REGIME CALCULATION")
        print(f"{'='*50}")
        print(f"Taxable Income: ₹{regime_data['taxable_income']:,.0f}")
        print(f"Tax Liability: ₹{regime_data['tax_liability']:,.0f}")
        print(f"TDS Paid: ₹{regime_data['tds_paid']:,.0f}")
        print(f"Balance: ₹{abs(regime_data['balance']):,.0f} ({'REFUND' if regime_data['balance'] > 0 else 'ADDITIONAL PAYABLE'})")
        print(f"Effective Tax Rate: {regime_data['effective_tax_rate']:.2f}%")
        
        # Show detailed tax calculation breakdown
        detailed = regime_data['detailed_calculation']
        print(f"\nTax Calculation Details:")
        print(f"  Tax Before Rebate: ₹{detailed['tax_before_rebate']:,.0f}")
        print(f"  Surcharge: ₹{detailed['surcharge']:,.0f}")
        print(f"  Health & Education Cess: ₹{detailed['cess']:,.0f}")
        print(f"  Rebate u/s 87A: ₹{detailed['rebate_87a']:,.0f}")
        
        # Show deductions breakdown for old regime
        if regime_name == 'old':
            deductions = detailed['deductions_used']
            print(f"\nDeductions Used:")
            print(f"  Section 80C: ₹{deductions['section_80c']:,.0f}")
            print(f"  Section 80CCD(1B): ₹{deductions['section_80ccd_1b']:,.0f}")
            print(f"  Total Deductions: ₹{deductions['total_deductions']:,.0f}")
    
    print(f"\n{'='*70}")
    print("INPUT SUMMARY")
    print(f"{'='*70}")
    input_data = result['input_data']
    print(f"Gross Salary: ₹{input_data['gross_salary']:,.0f}")
    print(f"Bank Interest: ₹{input_data['bank_interest']:,.0f}")
    print(f"Other Income: ₹{input_data['other_income']:,.0f}")
    print(f"House Property: ₹{input_data['house_property']:,.0f}")
    print(f"Section 80C: ₹{input_data['section_80c']:,.0f}")
    print(f"Section 80CCD(1B): ₹{input_data['section_80ccd_1b']:,.0f}")
    print(f"TDS Paid: ₹{input_data['tds_paid']:,.0f}")
    
else:
    print(f"Error: {result['error_message']}")

# Example output:
"""
Assessment Year: 2024-25
Regimes calculated: old, new

Recommendation: OLD regime recommended - saves ₹28,500 annually

==================================================
OLD REGIME CALCULATION
==================================================
Taxable Income: ₹1,540,000
Tax Liability: ₹156,000
TDS Paid: ₹185,000
Balance: ₹29,000 (REFUND)
Effective Tax Rate: 10.40%

Tax Calculation Details:
  Tax Before Rebate: ₹148,000
  Surcharge: ₹0
  Health & Education Cess: ₹5,920
  Rebate u/s 87A: ₹0

Deductions Used:
  Section 80C: ₹150,000
  Section 80CCD(1B): ₹50,000
  Total Deductions: ₹200,000

==================================================
NEW REGIME CALCULATION
==================================================
Taxable Income: ₹1,690,000
Tax Liability: ₹184,500
TDS Paid: ₹185,000
Balance: ₹500 (REFUND)  
Effective Tax Rate: 12.30%

Tax Calculation Details:
  Tax Before Rebate: ₹175,000
  Surcharge: ₹0
  Health & Education Cess: ₹7,000
  Rebate u/s 87A: ₹0

======================================================================
INPUT SUMMARY
======================================================================
Gross Salary: ₹1,500,000
Bank Interest: ₹45,000
Other Income: ₹75,000
House Property: ₹120,000
Section 80C: ₹150,000
Section 80CCD(1B): ₹50,000
TDS Paid: ₹185,000
"""