from docx import Document
import pandas as pd

# Crear documento de Word
doc = Document()
doc.add_heading('PAPEL DE TRABAJO', level=1)
doc.add_paragraph('Ejercicio fiscal: 2024')
doc.add_paragraph('Contribuyente: Rosa Aurora Ramón Díaz')
doc.add_paragraph('RFC: RADR630910V12')

# Sección 1: Ingresos acumulables
doc.add_heading('1. Ingresos acumulables por sueldos y salarios', level=2)
table1 = doc.add_table(rows=1, cols=3)
hdr_cells = table1.rows[0].cells
hdr_cells[0].text = 'Patrón'
hdr_cells[1].text = 'RFC'
hdr_cells[2].text = 'Ingresos'

rows1 = [
    ['Secretaría de Salud del Estado de Tabasco', 'OPD970314U91', '$68,363.99'],
    ['Servicios de Salud del IMSS para el Bienestar', 'SSI220901JS5', '$345,478.77'],
    ['Total de ingresos acumulables', '', '$413,842.76']
]

for row in rows1:
    cells = table1.add_row().cells
    for i, item in enumerate(row):
        cells[i].text = item

# Sección 2: ISR retenido
doc.add_heading('2. ISR retenido por los patrones', level=2)
table2 = doc.add_table(rows=1, cols=2)
hdr_cells = table2.rows[0].cells
hdr_cells[0].text = 'Patrón'
hdr_cells[1].text = 'ISR retenido'

rows2 = [
    ['Secretaría de Salud del Estado de Tabasco', '$10,763.20'],
    ['Servicios de Salud del IMSS para el Bienestar', '$57,219.74'],
    ['Total ISR retenido', '$67,982.94']
]

for row in rows2:
    cells = table2.add_row().cells
    for i, item in enumerate(row):
        cells[i].text = item

# Sección 3: Cálculo del ISR anual (aproximado)
doc.add_heading('3. Cálculo del ISR anual (aproximado)', level=2)
doc.add_paragraph("""
- Ingresos acumulables: $413,842.76
- Deducciones personales: $0.00
- Base gravable: $413,842.76

Usando tarifa ISR anual aproximada:
- ISR causado: $69,261.00
- Menos ISR retenido: $67,982.94
""")

# Sección 4: Resultado
doc.add_heading('4. Resultado', level=2)
doc.add_paragraph('Saldo a cargo aproximado: $1,278.06')

# Guardar el documento Word
doc.save("Papel_de_Trabajo_Rosa_Ramon.docx")

# Crear archivo Excel
data_ingresos = {
    "Patrón": ["Secretaría de Salud del Estado de Tabasco",
               "Servicios de Salud del IMSS para el Bienestar"],
    "RFC": ["OPD970314U91", "SSI220901JS5"],
    "Ingresos": [68363.99, 345478.77]
}

data_isr = {
    "Patrón": ["Secretaría de Salud del Estado de Tabasco",
               "Servicios de Salud del IMSS para el Bienestar"],
    "ISR retenido": [10763.20, 57219.74]
}

df_ingresos = pd.DataFrame(data_ingresos)
df_isr = pd.DataFrame(data_isr)

with pd.ExcelWriter("Papel_de_Trabajo_Rosa_Ramon.xlsx") as writer:
    df_ingresos.to_excel(writer, sheet_name="Ingresos", index=False)
    df_isr.to_excel(writer, sheet_name="ISR Retenido", index=False)