from fastapi import FastAPI, HTTPException, Header
from fastapi.responses import FileResponse
from pydantic import BaseModel
import torch, uuid, os
from TTS.api import TTS

app = FastAPI(title="XTTS v2 API")

print("Loading model...")
tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2",
          gpu=torch.cuda.is_available())
print("Model ready!")

API_KEY = os.environ.get("API_KEY", "changeme")

class TTSRequest(BaseModel):
    text: str
    language: str = "en"
    speaker: str = "Damien Black"

@app.get("/")
def root():
    return {"status": "running"}

@app.get("/debug-key")
def debug_key():
    return {
        "key_length": len(API_KEY),
        "key_preview": API_KEY[:4] + "...",
        "is_default": API_KEY == "changeme"
    }

@app.post("/tts")
def generate(req: TTSRequest, x_api_key: str = Header(None)):
    if x_api_key != API_KEY:
        raise HTTPException(status_code=401, detail="Invalid API key")

    output = f"/tmp/{uuid.uuid4()}.wav"
    tts.tts_to_file(
        text=req.text,
        speaker=req.speaker,
        language=req.language,
        file_path=output
    )
    return FileResponse(output, media_type="audio/wav", filename="output.wav")

@app.post("/debug-auth")
def debug_auth(x_api_key: str = Header(None)):
    return {
        "received": x_api_key,
        "expected_preview": API_KEY[:4] + "...",
        "length_received": len(x_api_key) if x_api_key else 0,
        "matches": x_api_key == API_KEY
    }