api.py

import os
from fastapi import FastAPI, HTTPException
from fastapi_mcp import FastApiMCP
from manageyourdata.data_manager import DataManager
from manageyourdata.utils import constants

dm = DataManager()
# Create MCP server from FastApi app.
app = FastAPI(title="ManageYourData")
mcp = FastApiMCP(app, name="ManageYourData")
mcp.mount()


@app.get("/")
def read_root():
    return {"message": "¡Bienvenido a ManageYourData!"}


@app.get("/files/")
def list_files():
    """Lista los archivos disponibles en el directorio /data."""
    try:
        files = os.listdir("./data")
        return {"files": files}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@app.post("/load-data/")
def load_data(file_name: str):
    """Carga un archivo de datos desde el directorio /data."""
    try:
        file_path = f"./data/{file_name}"
        dm.load_data(file_path)
        return {"message": f"Archivo {file_name} cargado correctamente."}
    except FileNotFoundError:
        raise HTTPException(status_code=404, detail="Archivo no encontrado.")
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@app.post("/generate-report/")
def generate_report():
    """Genera un informe PDF del archivo cargado."""
    try:
        if not dm.file_name:
            raise HTTPException(status_code=400, detail="No se ha cargado ningún archivo.")
        report_path = f"./reports/{dm.file_name}-report.pdf"
        dm.report_pdf(report_path)
        return {"message": f"Informe generado correctamente en {report_path}."}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@app.post("/export-data/")
def export_data(format: str):
    """Exporta los datos cargados en el formato especificado."""
    try:
        if not dm.file_name:
            raise HTTPException(status_code=400, detail="No se ha cargado ningún archivo.")
        if format not in constants.FORMAT:
            raise HTTPException(status_code=400, detail="Formato no soportado.")
        file_extension = constants.FORMAT[format]
        dm.export_data(file_extension)
        return {"message": f"Datos exportados correctamente."}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@app.post("/create-assistant/")
def create_assistant(provider: str, model: str, api_key: str = None):
    """Crea un asistente de IA para interactuar con los datos cargados."""
    try:
        if not dm.file_name:
            raise HTTPException(status_code=400, detail="No se ha cargado ningún archivo.")
        dm.create_assistant(provider, model, api_key)
        return {"message": "Asistente de IA creado correctamente."}
    except ValueError as e:
        raise HTTPException(status_code=500, detail=str(e))
    

@app.post("/chat_with_assistant/")
def chat_with_assistant(prompt: str):
    """Interactúa con el asistente de IA utilizando el prompt proporcionado."""
    try:
        if not dm.agent:
            raise HTTPException(status_code=400, detail="No se ha creado ningún asistente de IA.")
        response = dm.chat_with_assistant(prompt)
        return {"response": response}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))