app.py

import gradio as gr
import pandas as pd
from pandasai import SmartDataframe

df_global = None
sdf = None

def cargar_csv(file):
    global df_global, sdf
    try:
        df_global = pd.read_csv(file.name)
        sdf = SmartDataframe(df_global, config={"enable_cache": False})
        return f"Archivo cargado con éxito. Filas: {len(df_global)}, Columnas: {len(df_global.columns)}"
    except Exception as e:
        return f"Error al cargar el archivo: {str(e)}"

def responder_pregunta(pregunta):
    global sdf
    if sdf is None:
        return "Primero debes cargar un archivo CSV."
    try:
        respuesta = sdf.chat(pregunta)
        if isinstance(respuesta, bool):
            return f"No se puede interpretar la pregunta '{pregunta}' sin un modelo LLM. Por favor, intenta una consulta más simple."
        return str(respuesta)
    except Exception as e:
        return f"Ocurrió un error al responder la pregunta: {str(e)}"

with gr.Blocks() as demo:
    gr.Markdown("# Analizador CSV sin LLM externo (versión robusta)")

    with gr.Row():
        archivo = gr.File(label="Sube tu archivo CSV", file_types=[".csv"])
        salida_carga = gr.Textbox(label="Resultado de carga")
        boton_carga = gr.Button("Cargar CSV")

    with gr.Row():
        pregunta = gr.Textbox(label="Escribe tu pregunta sobre los datos")
        salida_respuesta = gr.Textbox(label="Respuesta")
        boton_pregunta = gr.Button("Responder")

    boton_carga.click(cargar_csv, inputs=[archivo], outputs=[salida_carga])
    boton_pregunta.click(responder_pregunta, inputs=[pregunta], outputs=[salida_respuesta])

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