import gradio as gr
import pandas as pd
import geopandas as gpd
import openpyxl

# ---------------------------------------
# Função para formatar visualização
# ---------------------------------------
def formatar_float_visualizacao(df, casas=4):
    df_visual = df.copy()
    float_cols = df_visual.select_dtypes(include=['float']).columns
    df_visual[float_cols] = df_visual[float_cols].round(casas)
    return df_visual

# ---------------------------------------
# Carregar shapefile dos eixos
# ---------------------------------------
gdf_eixos = gpd.read_file("EixosLogradouros.shp", engine="fiona")
gdf_eixos_proj = gdf_eixos.to_crs("EPSG:4326")

# ---------------------------------------
# Funções auxiliares
# ---------------------------------------
def carregar_abas(arquivo_excel):
    if arquivo_excel is None:
        return gr.update(choices=[]), None
    xls = pd.ExcelFile(arquivo_excel.name)
    abas = xls.sheet_names
    return gr.update(choices=abas, value=abas[0]), abas[0]


def listar_colunas(arquivo_excel, aba_selecionada):
    if arquivo_excel is None or aba_selecionada is None:
        return gr.update(choices=[]), gr.update(choices=[])

    df = pd.read_excel(arquivo_excel.name, sheet_name=aba_selecionada)
    colunas = df.columns.tolist()
    return gr.update(choices=colunas), gr.update(choices=colunas)


def exibir_tabela(arquivo_excel, aba_selecionada, nome_coluna_num):
    if arquivo_excel is None or aba_selecionada is None:
        return None, ""

    df = pd.read_excel(arquivo_excel.name, sheet_name=aba_selecionada)

    if nome_coluna_num in df.columns:
        df[nome_coluna_num] = pd.to_numeric(df[nome_coluna_num], errors='coerce').fillna(0).astype(int)

    return formatar_float_visualizacao(df), f"O DataFrame possui {df.shape[0]} linhas e {df.shape[1]} colunas."


def obter_cdlog_por_nmidelog(nome_logradouro):
    if not nome_logradouro:
        return None
    cdlog_unicos = gdf_eixos_proj[gdf_eixos_proj["NMIDEABR"] == nome_logradouro]["CDLOG"].unique()
    return int(cdlog_unicos[0]) if len(cdlog_unicos) > 0 else None


# ---------------------------------------
# Interpolação
# ---------------------------------------
def interpolar_enderecos(df, col_cdlog, col_num):

    df[col_num] = pd.to_numeric(df[col_num], errors='coerce').fillna(0).astype(int)
    resultados = []
    falhas = []

    for _, row in df.iterrows():

        cdlog = row[col_cdlog]
        numero = row[col_num]

        segmentos = gdf_eixos_proj[gdf_eixos_proj['CDLOG'] == cdlog]

        if segmentos.empty:
            resultados.append((None, None))
            falhas.append({"cdlog": cdlog, "numero": numero, "motivo": "CDLOG não encontrado"})
            continue

        if numero % 2 == 0:
            cond = (segmentos['NRPARINI'] <= numero) & (segmentos['NRPARFIN'] >= numero)
        else:
            cond = (segmentos['NRIMPINI'] <= numero) & (segmentos['NRIMPFIN'] >= numero)

        segmentos_validos = segmentos[cond]

        if segmentos_validos.empty:
            resultados.append((None, None))
            falhas.append({"cdlog": cdlog, "numero": numero, "motivo": "Numeração fora do intervalo"})
            continue

        linha = segmentos_validos.iloc[0]
        geom = linha.geometry

        ini = linha['NRPARINI'] if numero % 2 == 0 else linha['NRIMPINI']
        fim = linha['NRPARFIN'] if numero % 2 == 0 else linha['NRIMPFIN']

        frac = (numero - ini) / (fim - ini)
        frac = max(0, min(1, frac))

        ponto = geom.interpolate(geom.length * frac)

        resultados.append((ponto.x, ponto.y))

    df["lon"] = [x for x, y in resultados]
    df["lat"] = [y for x, y in resultados]

    output_path = "dados_interpolados.xlsx"
    df.to_excel(output_path, index=False)

    return formatar_float_visualizacao(df), output_path, pd.DataFrame(falhas)


# ---------------------------------------
# Interface
# ---------------------------------------
with gr.Blocks() as app:

    gr.Markdown("## DAI - Geolocalização (Sem Mapa)")

    arquivo = gr.File(label="Arquivo Excel", file_types=[".xlsx"])

    dropdown_abas = gr.Dropdown(label="Aba")
    linhas_colunas_output = gr.Textbox(label="Linhas e Colunas", interactive=False)

    dropdown_cdlog = gr.Dropdown(label="Coluna CDLOG")
    dropdown_num = gr.Dropdown(label="Coluna Número")

    tabela_output = gr.Dataframe(label="Prévia", interactive=True)

    btn_confirmar = gr.Button("Confirmar Dados")
    btn_interpolar = gr.Button("Obter Coordenadas")

    tabela_interpolada = gr.Dataframe(label="Com Coordenadas")
    arquivo_excel_output = gr.File(label="Arquivo Gerado")
    falhas_output = gr.Dataframe(label="Falhas")

    # Eventos
    arquivo.change(fn=carregar_abas, inputs=arquivo, outputs=[dropdown_abas, dropdown_abas])

    dropdown_abas.change(
        fn=exibir_tabela,
        inputs=[arquivo, dropdown_abas, dropdown_num],
        outputs=[tabela_output, linhas_colunas_output]
    ).then(
        fn=listar_colunas,
        inputs=[arquivo, dropdown_abas],
        outputs=[dropdown_cdlog, dropdown_num]
    )

    btn_interpolar.click(
        fn=interpolar_enderecos,
        inputs=[tabela_output, dropdown_cdlog, dropdown_num],
        outputs=[tabela_interpolada, arquivo_excel_output, falhas_output]
    )

app.launch(server_name="0.0.0.0", server_port=7860)