import gradio as gr
import pandas as pd
from io import StringIO
from datetime import datetime
import boto3
import os
ACCESS = os.environ.get("ACCESS_KEY")
SECRET = os.environ.get("SECRET")

def validar(data):
    try:
        buffer = StringIO(data)
        df = pd.read_csv(buffer, sep="\t", na_values=["null", "NULL", "NaN", ""])
        df_limpio = df.dropna(axis=1, how='all')

        if df_limpio.shape[1] < 3:
            return None, "⚠️ Error: Debes tener al menos 3 columnas (Activity Type, Spreadsheet Field y una fecha)."

        col0, col1, col2 = df_limpio.columns[:3]
        if col0.strip() != 'Activity Type':
            return None, "⚠️ Error: La primera columna debe llamarse exactamente 'Activity Type'."
        if col1.strip() != 'Spreadsheet Field':
            return None, "⚠️ Error: La segunda columna debe llamarse exactamente 'Spreadsheet Field'."
        if len(df)!=2:
            return None, "⚠️ Error: Debes tener exactamente 2 filas de 'Spreadsheet Field'."
        df_limpio.fillna(0, inplace=True)
        try:
            # Identificar las columnas que contienen fechas (todas excepto las dos primeras)
            columnas_fecha = df_limpio.columns.difference(['Activity Type', 'Spreadsheet Field'])

            # Aplicar conversión solo en columnas de fechas
            df_limpio[columnas_fecha] = df_limpio[columnas_fecha].applymap(
                lambda x: float(str(x).replace(',', '.')) if isinstance(x, str) and ',' in x else x
            )
            datetime.strptime(col2.strip(), "%d-%b-%y")
        except ValueError:
            return None, "⚠️ Error: La tercera columna debe ser una fecha en formato dd-MMM-yy."

        return df_limpio, "✅ ¡Datos válidos! Ya puedes enviarlos al servidor."

    except Exception as e:
        if str(e) == 'No columns to parse from file':
            e = "⚠️ No se encontraron columnas para analizar. Asegúrate copiar la información correcta."
        return None, f"❌ Error inesperado: {str(e)}"

def enviar_al_servidor(df, EMPRESA, PROYECTO):
    print(EMPRESA, PROYECTO)
    # Inicializar cliente DynamoDB
    dynamodb = boto3.resource('dynamodb', region_name='us-east-1',
                                              aws_access_key_id=ACCESS,
                                              aws_secret_access_key=SECRET)

    # Validar existencia del proyecto
    tabla_proyecto = dynamodb.Table('proyecto')
    respuesta = tabla_proyecto.get_item(Key={'empresa': EMPRESA, 'proyecto': PROYECTO})

    if 'Item' not in respuesta:
        print("❌ El proyecto no existe en la tabla 'proyecto'.")
        return "❌ Error: El proyecto no existe en la tabla 'proyecto'."

    
    FINAL = df.groupby('Activity Type')[df.columns[1:]].agg('sum').reset_index()
    
    
    # Initialize the DynamoDB client
    table_name = "cum_budgeted_labor_units"  # Replace with your actual table name
    table = dynamodb.Table(table_name)

    FINAL = FINAL.map(str)
    FINAL.rename(columns={'Activity Type':'tipo'}, inplace=True)
    for index, row in FINAL.iterrows():
        
        item = row.to_dict()  
        item['id_empresa_proyecto'] = EMPRESA+'_'+PROYECTO
        response = table.put_item(Item=item)  # Insert into DynamoDB
        print(f"Inserted row {index}: {response}")

    print("Datos enviados al servidor:")
    return "✅ Datos enviados al servidor con éxito."

def enviar_al_servidor_botton(df_limpio, EMPRESA, PROYECTO):
    EMPRESA = EMPRESA.strip()
    PROYECTO = PROYECTO.strip()
    if df_limpio is not None:
        return enviar_al_servidor(df_limpio, EMPRESA, PROYECTO)
    return "❌ No hay datos válidos para enviar."

# Usamos gr.Blocks para mayor control
with gr.Blocks() as demo:
    gr.File(value="Layout cum_units.plf", label="📁 Descargar LAYOUT de P6 para extracción de HH", interactive=False)

    textbox_input0 = gr.Textbox(label="Escribe el nombre de la empresa en mayúsculas",
        lines=1,
        placeholder= 'Ejemplo: TECNASIC'
        )

    textbox_input1 = gr.Textbox(label="Escribe el nombre del proyecto como se ve en plataforma",
        lines=1,
        placeholder= 'Ejemplo: OT2936 Lomas Bayas'
        )
    textbox_input2 = gr.Textbox(
        label="Pega aquí tu tabla de remanentes de la Línea Base",
        lines=4,
        placeholder="Activity Type\tSpreadsheet Field\t06-May-24\t13-May-24\t20-May-24\nTask Dependent\tCum Earned Value Labor Units\t\t551\t\t10\t\t700\nTask Dependent\tCum Remaining Labor Units\t\t551\t\t10\t\t700"
    )
    validar_btn = gr.Button("Validar datos")
    df_output = gr.Dataframe(label="Remanentes limpios")
    status_box = gr.Textbox(label="Estado de validación", interactive=False)
    enviar_btn = gr.Button("Enviar datos al servidor")

    validar_btn.click(fn=validar, inputs=textbox_input2, outputs=[df_output, status_box])
    enviar_btn.click(fn=enviar_al_servidor_botton, inputs=[df_output,textbox_input0,textbox_input1], outputs=status_box)

if __name__ == "__main__":
    demo.launch()