from fastapi import FastAPI, File, UploadFile, Form
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
import shutil
import tempfile
import os
import json

# Orquestador Médico
from ADNI_Fusion_Engine import PerdixAdniEngine

app = FastAPI(title="Perdix ADNI Medical Node", version="1.0.0")

# Permitir a NextJS conectarse a este motor en Hugging Face
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"], # En producción cambiar por la URL de vercel
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

print("Inicializando Motor ADNI Neuronal en Hugging Face...")
engine = PerdixAdniEngine()

@app.get("/")
def read_root():
    return {"status": "online", "message": "Perdix ADNI Engine is running. Send POST to /diagnose."}

@app.post("/diagnose")
async def diagnose(
    age: float = Form(65.0), 
    mmse_score: float = Form(25.0), 
    edu_years: float = Form(12.0),
    mri_scan: UploadFile = File(...)
):
    try:
        # Guardar la imagen subida en disco temporalmente
        ext = os.path.splitext(mri_scan.filename)[1]
        if not ext:
            ext = ".png"
            
        with tempfile.NamedTemporaryFile(delete=False, suffix=ext) as tmp:
            shutil.copyfileobj(mri_scan.file, tmp)
            tmp_path = tmp.name
        
        # Ejecutamos la Inferencia con PyTorch
        result = engine.diagnose_patient(
            image_path=tmp_path, 
            age=age, 
            mmse_score=mmse_score, 
            edu_years=edu_years
        )
        
        # Inyectar ID Clínico
        import uuid
        result["diagnosis_id"] = "ADNI-" + str(uuid.uuid4())[:8]
        
        # Limpieza de Disco 
        os.unlink(tmp_path)
        
        return result
        
    except Exception as e:
        return {"status": "error", "message": str(e)}

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=7860)