import streamlit as st

from pages.vrp.Origen import Grafo, Vehiculos
from pages.vrp.Algoritmo_genético import Algoritmo_Genetico

#######################
## FUNCIONES AUXILIARES
#######################

def grafar(reparto : list[list[int]]):   
    
    result = "digraph { \n"

    for ruta in reparto:
        result += "Almacén ->"
        for nodo in ruta:
            result += str(nodo) + "\n"
            result += str(nodo) + "->"
        result += "Almacén \n"

    result += "}" 

    return result

def presentar(reparto : list[list[int]]):

    result = ""

    for i in range(len(reparto)):
        result += "Camion " + str(i) + ": " + "Almacén -> "
        for parada in reparto[i]:
            result += str(parada) + " -> "
        result += "Almacén \n"

    return result

st.set_page_config(page_title="LupercAI", page_icon= "🚛")

st.title("LupercAI 🚛")

valores_vitales = ["distancias", "demandas_clientes", "capacidad_vehiculos"]
importado = True
for valor in valores_vitales: importado = importado and valor in st.session_state

if not importado: 
    st.warning("Faltan datos", icon="⚠️")
else:
    #if "num_nodos" not in st.session_state: st.session_state["num_nodos"] = len(st.session_state["demandas_clientes"])
    #if "num_vehiculos" not in st.session_state: st.session_state["num_vehiculos"] = len(st.session_state["capacidad_vehiculos"])

    grafo = Grafo(st.session_state["distancias"], st.session_state["demandas_clientes"])
    vehiculos = Vehiculos(st.session_state["capacidad_vehiculos"])

    if "activar_generaciones" in st.session_state:
        if st.session_state["activar_generaciones"]:
            if "alg_genetico" not in st.session_state: st.session_state["alg_genetico"] = Algoritmo_Genetico(tamano_poblacion = st.session_state["tamano_poblacion"], 
                generaciones = st.session_state["generaciones"], tamano_torneo = st.session_state["tamano_torneo"], 
                crossover_prob = st.session_state["crossover_prob"], mutation_prob = st.session_state["mutation_prob"])
        else:
            if "alg_genetico" not in st.session_state: st.session_state["alg_genetico"] = Algoritmo_Genetico()
    else:
        if "alg_genetico" not in st.session_state: st.session_state["alg_genetico"] = Algoritmo_Genetico()

    alg_genetico = st.session_state["alg_genetico"]

    from threading import Thread
    Thread(target=alg_genetico.generar, args = (grafo, vehiculos)).start()
    

    #Computación:

    empty_latest_iteration = st.empty()
    empty_mejor_coste = st.empty()

    st.subheader("Barra de progreso")
    barra_progreso = st.progress(0)  

    st.subheader("Grafo del mejor recorrido actual")
    
    empty_txt = st.empty()
    
    empty_grafo = st.empty()

    i = 0
    while alg_genetico.generacion_actual < alg_genetico.generaciones:
        #Generación por escrito
        empty_latest_iteration.write("Generación: " + str(alg_genetico.generacion_actual))

        #Barra de progreso
        barra_progreso.progress( alg_genetico.generacion_actual/alg_genetico.generaciones)

        #Mejor coste actual (texto)
        empty_mejor_coste.write("**Mejor coste**: " + str(alg_genetico.mejor_candidato[1]))

        #Descarga de grafo
        empty_txt.download_button("Mejor recorrido 🔽", presentar(alg_genetico.mejor_candidato[0]), file_name="Mejor reparto - Generacion " + str(alg_genetico.generacion_actual) + ".txt", key=str(i))
        i += 1

        #Mostrar el grafo
        empty_grafo.graphviz_chart(grafar(alg_genetico.mejor_candidato[0]))

        
    #Ultima iteracion
    empty_latest_iteration.write("Generación: " + str(alg_genetico.generacion_actual))
    barra_progreso.progress( alg_genetico.generacion_actual/alg_genetico.generaciones)
    empty_mejor_coste.write("**Mejor coste**: " + str(alg_genetico.mejor_candidato[1]))
    empty_txt.download_button("Mejor recorrido 🔽", presentar(alg_genetico.mejor_candidato[0]), file_name="Mejor reparto - Generacion " + str(alg_genetico.generacion_actual) + ".txt", key=str(i))
    empty_grafo.graphviz_chart(grafar(alg_genetico.mejor_candidato[0]))