from fastapi import APIRouter
from src.app.schemas.requests.eda import EdaRequestSchema
from src.app.pipelines.eda.pipeline import EdaLoop
import os

eda_router = APIRouter()

def delete_dir_contents(directory: str)->None:
    for filename in os.listdir(directory):
        file_path = os.path.join(directory, filename)
        if os.path.isfile(file_path):
            os.remove(file_path)

@eda_router.post('/')
async def main(response: EdaRequestSchema):

    eda = EdaLoop(payload=response, verbose=True)
    logs = eda.loop(verbose=True)

    delete_dir_contents(directory="src/core/cache/dataset_logs")
    delete_dir_contents(directory="src/core/cache/downloads")

    return {    
        "execution_logs": logs
    }