import streamlit as st
import requests as req 
import pandas as pd
st.set_page_config(page_title="PREP exporter", page_icon=":shark:", layout="wide")


st.title("PREP exporter")

candidatos = req.get("https://prepgto2024.ieeg.mx/assets/json/infoStaging.json").json()

# st.write(candidatos.keys())

canditatos_ayuntamientos = candidatos['ayuntamientos']['candidaturas']['candidatos']

candidatos_diputaciones = candidatos['diputaciones']['candidaturas']['candidatos']

# st.write(canditatos_ayuntamientos)
st.write('### IEEG ayuntamientos')
ayuntamientos = req.get("https://2024elecciones.s3.amazonaws.com/ayuntamientos2024.json").json()
# st.write(ayuntamientos['entidad']['entidad'][0])

data = ayuntamientos['entidad']['entidad']

dataset =[]
i = 0
for entidad in data:
    candidatos_entidad = canditatos_ayuntamientos[i]
    e = {
        "entidad": entidad['nombreEntidad']
    }
    for candidato in candidatos_entidad:
        e[candidato['partido'] + ' candidato'] = candidato['candidato']
    for candidatura in entidad['votosCandidatura']['votos']:
        e[candidatura['imagenPartidos'] + ' votos' ] = candidatura['votos']
        e[candidatura['imagenPartidos'] + ' porcentaje' ] = candidatura['porcentaje']
    dataset.append(e)
    i += 1

df_ayuntamientos = pd.DataFrame(dataset)
ordered_columns = sorted(df_ayuntamientos.columns)
# send 'entidad' to the first column
ordered_columns.remove('entidad')
ordered_columns.insert(0, 'entidad')
df_ayuntamientos = df_ayuntamientos.reindex(ordered_columns, axis=1)

st.write(df_ayuntamientos)

st.write('### IEEG diputaciones')

diputaciones = req.get("https://2024elecciones.s3.amazonaws.com/diputaciones2024.json").json()
distritos = {
    'DISTRITO 1': 'Dolores Hidalgo Cuna de la Independencia Nacional',
    'DISTRITO 2': 'San Luis de la Paz',
    'DISTRITO 3': 'León',
    'DISTRITO 4': 'León',
    'DISTRITO 5': 'León',
    'DISTRITO 6': 'León',
    'DISTRITO 7': 'León',
    'DISTRITO 8': 'Guanajuato',
    'DISTRITO 9': 'San Miguel de Allende',
    'DISTRITO 10': 'San Francisco del Rincón',
    'DISTRITO 11': 'Irapuato',
    'DISTRITO 12': 'Irapuato',
    'DISTRITO 13': 'Cortazar',
    'DISTRITO 14': 'Salamanca',
    'DISTRITO 15': 'Celaya',
    'DISTRITO 16': 'Celaya',
    'DISTRITO 17': 'Apaseo el Grande',
    'DISTRITO 18': 'Pénjamo',
    'DISTRITO 19': 'Valle de Santiago',
    'DISTRITO 20': 'Yuriria',
    'DISTRITO 21': 'León',
    'DISTRITO 22': 'Acámbaro'
}

data = diputaciones['entidad']['entidad']

dataset =[]

i = 0
for entidad in data:
    candidatos_entidad = candidatos_diputaciones[i]
    e = {
        "distrito": distritos[entidad['nombreEntidad']]
    }
    for candidato in candidatos_entidad:
        e[candidato['partido'] + ' candidato'] = candidato['candidato']
    for candidatura in entidad['votosCandidatura']['votos']:
        e[candidatura['imagenPartidos'] + ' votos' ] = candidatura['votos']
        e[candidatura['imagenPartidos'] + ' porcentaje' ] = candidatura['porcentaje']
    dataset.append(e)
    i += 1

df_diputaciones = pd.DataFrame(dataset)
ordered_columns = sorted(df_diputaciones.columns)
# send 'entidad' to the first column

ordered_columns.remove('distrito')
ordered_columns.insert(0, 'distrito')
df_diputaciones = df_diputaciones.reindex(ordered_columns, axis=1)

st.write(df_diputaciones)