ai_server.py 中的 funasr 初始化迁移到 ai_task.py

ai_server.py

@@ -1,4 +1,3 @@
-import asyncio
 import logging
 logging.basicConfig(
     level=logging.INFO,
@@ -9,7 +8,6 @@ from contextlib import asynccontextmanager
 from fastapi import FastAPI
 from route import gpu, file, audio, text, image
 import service.ai_task as ai_task
-import service.ai_asr as pyasr
 
 logger = logging.getLogger(__name__)
 # 获取 uvicorn 的 logger
@@ -17,11 +15,10 @@ logger = logging.getLogger(__name__)
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    logger.info("🚀 服务已启动...")
-    asyncio.create_task(pyasr.init_funasr())
+    logger.info("🚀 pyai 服务已启动...")
     await ai_task.start_worker()
     yield
-    logger.info("🛑 服务已停止")
+    logger.info("🛑 pyai 服务已停止")
 app = FastAPI(title="GPU Worker Server", lifespan=lifespan)
 
 # 挂载子路由

service/ai_task.py

@@ -6,7 +6,6 @@ from concurrent.futures import ThreadPoolExecutor
 
 import service.pyav as pyav
 import service.pygpu as pygpu
-import service.ai_text_ollama as ai_text
 import service.ai_asr as pyasr
 from setting import OUTPUT_DIR
 
@@ -15,7 +14,10 @@ executor = ThreadPoolExecutor(max_workers=1)
 task_queue = asyncio.Queue()
 
 async def start_worker():
-    asyncio.create_task(gpu_worker())  # 开启 Worker
+    # 开启 funasr 初始化任务
+    asyncio.create_task(pyasr.init_funasr())
+    # 开启 Worker
+    asyncio.create_task(gpu_worker())
 
 
 # --- 后台工作进程 (Consumer) ---
@@ -71,9 +73,6 @@ async def gpu_worker():
 async def put_task(task_id, save_path, srt_path, video_path):
     await task_queue.put((task_id, save_path, srt_path, video_path))
 
+
 def get_tasks():
     return task_queue.qsize()
-
-def translate_to_zh(text):
-    result = ai_text.translate2zh(text)
-    return result