feat: 语音交互网页初始版本 - 前端+后端
This commit is contained in:
155
server.py
Normal file
155
server.py
Normal file
@@ -0,0 +1,155 @@
|
||||
"""
|
||||
语音交互网页后端
|
||||
代理转发到 Qwen2-Audio 模型服务
|
||||
"""
|
||||
|
||||
import os
|
||||
import logging
|
||||
from typing import Optional
|
||||
from datetime import datetime
|
||||
|
||||
import aiohttp
|
||||
from fastapi import FastAPI, UploadFile, File, HTTPException, Form
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
from fastapi.staticfiles import StaticFiles
|
||||
from pydantic import BaseModel
|
||||
|
||||
# 配置
|
||||
MODEL_SERVICE_URL = os.getenv("MODEL_SERVICE_URL", "http://localhost:19018")
|
||||
PORT = int(os.getenv("PORT", "19019"))
|
||||
|
||||
# 日志
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
app = FastAPI(
|
||||
title="Voice Chat Web",
|
||||
description="语音交互网页后端",
|
||||
version="1.0.0"
|
||||
)
|
||||
|
||||
# CORS
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
allow_origins=["*"],
|
||||
allow_credentials=True,
|
||||
allow_methods=["*"],
|
||||
allow_headers=["*"],
|
||||
)
|
||||
|
||||
|
||||
class VoiceResponse(BaseModel):
|
||||
"""语音响应"""
|
||||
reply: str
|
||||
conversation_id: str
|
||||
timestamp: str
|
||||
|
||||
|
||||
class StatusResponse(BaseModel):
|
||||
"""状态响应"""
|
||||
status: str
|
||||
model_service: str
|
||||
model_online: bool
|
||||
|
||||
|
||||
@app.get("/")
|
||||
async def root():
|
||||
"""根路径返回状态"""
|
||||
return {"status": "ok", "service": "voice-chat-web"}
|
||||
|
||||
|
||||
@app.get("/api/status", response_model=StatusResponse)
|
||||
async def get_status():
|
||||
"""检查服务状态"""
|
||||
try:
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.get(f"{MODEL_SERVICE_URL}/", timeout=aiohttp.ClientTimeout(total=5)) as resp:
|
||||
if resp.status == 200:
|
||||
data = await resp.json()
|
||||
return StatusResponse(
|
||||
status="ok",
|
||||
model_service=MODEL_SERVICE_URL,
|
||||
model_online=True
|
||||
)
|
||||
except Exception as e:
|
||||
logger.warning(f"Model service check failed: {e}")
|
||||
|
||||
return StatusResponse(
|
||||
status="partial",
|
||||
model_service=MODEL_SERVICE_URL,
|
||||
model_online=False
|
||||
)
|
||||
|
||||
|
||||
@app.post("/api/voice/chat", response_model=VoiceResponse)
|
||||
async def voice_chat(
|
||||
audio: UploadFile = File(..., description="音频文件"),
|
||||
conversation_id: Optional[str] = Form(None, description="对话ID")
|
||||
):
|
||||
"""
|
||||
语音聊天接口
|
||||
转发到模型服务
|
||||
"""
|
||||
try:
|
||||
# 读取音频数据
|
||||
audio_bytes = await audio.read()
|
||||
|
||||
# 转发到模型服务
|
||||
async with aiohttp.ClientSession() as session:
|
||||
form = aiohttp.FormData()
|
||||
form.add_field(
|
||||
'audio',
|
||||
audio_bytes,
|
||||
filename=audio.filename or 'audio.wav',
|
||||
content_type=audio.content_type or 'audio/wav'
|
||||
)
|
||||
if conversation_id:
|
||||
form.add_field('conversation_id', conversation_id)
|
||||
|
||||
async with session.post(
|
||||
f"{MODEL_SERVICE_URL}/api/voice/inference",
|
||||
data=form,
|
||||
timeout=aiohttp.ClientTimeout(total=120) # 模型推理可能较慢
|
||||
) as resp:
|
||||
if resp.status != 200:
|
||||
error_text = await resp.text()
|
||||
logger.error(f"Model service error: {error_text}")
|
||||
raise HTTPException(status_code=resp.status, detail=error_text)
|
||||
|
||||
data = await resp.json()
|
||||
return VoiceResponse(
|
||||
reply=data["reply"],
|
||||
conversation_id=data["conversation_id"],
|
||||
timestamp=data.get("timestamp", datetime.now().isoformat())
|
||||
)
|
||||
|
||||
except aiohttp.ClientError as e:
|
||||
logger.error(f"Connection error: {e}")
|
||||
raise HTTPException(status_code=503, detail="模型服务连接失败")
|
||||
except Exception as e:
|
||||
logger.error(f"Voice chat error: {e}", exc_info=True)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@app.delete("/api/conversation/{conversation_id}")
|
||||
async def delete_conversation(conversation_id: str):
|
||||
"""删除对话"""
|
||||
try:
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.delete(
|
||||
f"{MODEL_SERVICE_URL}/api/voice/conversation/{conversation_id}",
|
||||
timeout=aiohttp.ClientTimeout(total=10)
|
||||
) as resp:
|
||||
return await resp.json()
|
||||
except Exception as e:
|
||||
logger.error(f"Delete conversation error: {e}")
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
# 静态文件(前端页面)
|
||||
app.mount("/static", StaticFiles(directory="static"), name="static")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
import uvicorn
|
||||
uvicorn.run(app, host="0.0.0.0", port=PORT)
|
||||
Reference in New Issue
Block a user