from fastapi import APIRouter, UploadFile, File, HTTPException
import logging
from app.services.spreadsheet_service import SampleSpreadsheetImporter, SpreadsheetImportError

router = APIRouter()
logger = logging.getLogger(__name__)


@router.post("/upload")
async def upload_file(file: UploadFile = File(...)):
    try:
        logger.info(f"Received file: {file.filename}")

        # File type check
        if not file.filename.endswith('.xlsx'):
            logger.error("Invalid file format")
            raise HTTPException(status_code=400, detail="Invalid file format. Please upload an .xlsx file.")

        # Reading file
        importer = SampleSpreadsheetImporter()
        validated_model = importer.import_spreadsheet(file)
        logger.info(f"Validated model: {validated_model}")

        dewars = {sample.dewarname for sample in validated_model if sample.dewarname}
        pucks = {sample.puckname for sample in validated_model if sample.puckname}
        samples = {sample.crystalname for sample in validated_model if sample.crystalname}

        # Logging the sets of names
        logger.info(f"Dewar Names: {dewars}")
        logger.info(f"Puck Names: {pucks}")
        logger.info(f"Sample Names: {samples}")

        # Forming structured response
        response = {
            "dewars_count": len(dewars),
            "dewars": list(dewars),
            "pucks_count": len(pucks),
            "pucks": list(pucks),
            "samples_count": len(samples),
            "samples": list(samples)
        }

        logger.info(f"Returning response: {response}")
        return response
    except SpreadsheetImportError as e:
        logger.error(f"Spreadsheet import error: {str(e)}")
        raise HTTPException(status_code=400, detail=str(e))
    except Exception as e:
        logger.error(f"Failed to process file: {str(e)}")
        raise HTTPException(status_code=500, detail=f"Failed to upload file. Please try again. {str(e)}")