from fastapi import FastAPI
from pydantic import BaseModel
from mutation_analyzer import find_mutations

app = FastAPI()

class DNAInput(BaseModel):
    reference: str
    sample: str

@app.post("/analyze")
def analyze_dna(data: DNAInput):
    reference = data.reference.upper()
    sample = data.sample.upper()

    mutations = find_mutations(reference, sample)

    message = "No mutation found. Sequence matches reference."
    if len(mutations) > 0:
        message = f"Mutation found! Total mutations detected: {len(mutations)}"

    return {
        "message": message,
        "total_mutations": len(mutations),
        "mutations": mutations
    }