"""
语音交互网页 - 主入口
"""

import os
import uvicorn
import aiohttp
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.responses import FileResponse, Response
from fastapi.staticfiles import StaticFiles

# 导入后端服务
from server import app as api_app

# 主应用
app = FastAPI(title="Voice Chat Web")

# CORS
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


# ChatTTS 音频代理（解决 HTTPS 页面访问 HTTP 资源问题）
@app.get("/chattts/audio/{filename}")
async def proxy_chattts_audio(filename: str):
    """代理 ChatTTS 音频文件"""
    chattts_url = os.getenv("CHATTTS_URL", "http://192.168.2.5:12002")
    
    try:
        async with aiohttp.ClientSession() as session:
            async with session.get(
                f"{chattts_url}/audio/{filename}",
                timeout=aiohttp.ClientTimeout(total=30)
            ) as resp:
                if resp.status != 200:
                    return Response(content=b'{"detail":"Audio not found"}', status_code=404, media_type="application/json")
                
                audio_data = await resp.read()
                return Response(
                    content=audio_data,
                    media_type="audio/wav",
                    headers={"Cache-Control": "public, max-age=3600"}
                )
    except Exception as e:
        return Response(content=f'{"detail":"{str(e)}"}'.encode(), status_code=500, media_type="application/json")


# 挂载 API
app.mount("/api", api_app)

# 静态文件
app.mount("/static", StaticFiles(directory="static"), name="static")


@app.get("/")
async def index():
    """主页（原版）"""
    return FileResponse("static/index.html")


@app.get("/tts")
async def tts_page():
    """TTS版本页面"""
    return FileResponse("static/tts.html")


if __name__ == "__main__":
    PORT = int(os.getenv("PORT", "19019"))
    SSL_KEY = os.getenv("SSL_KEY", "key.pem")
    SSL_CERT = os.getenv("SSL_CERT", "cert.pem")
    
    # 检查是否有 SSL 证书
    if os.path.exists(SSL_KEY) and os.path.exists(SSL_CERT):
        uvicorn.run(app, host="0.0.0.0", port=PORT, ssl_keyfile=SSL_KEY, ssl_certfile=SSL_CERT)
    else:
        uvicorn.run(app, host="0.0.0.0", port=PORT)