You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
import uvicorn
|
|
from multiprocessing import Process, Event, freeze_support, Manager
|
|
from loguru import logger
|
|
|
|
from helper import PromptQueue, LiveChatConfig
|
|
from liveMan import DouyinLiveWebFetcher, DouyinLiveWebReply
|
|
|
|
logger.add("log.log", encoding="utf-8", rotation="500MB")
|
|
|
|
def start_fastapi_server(shared_config):
|
|
from app import app
|
|
app.shared_config = shared_config
|
|
uvicorn.run(app, host="0.0.0.0", port=8000)
|
|
|
|
def fetch_user_chat_content(ws_open_event, queue):
|
|
fetcher = DouyinLiveWebFetcher(ws_open_event, queue)
|
|
fetcher.start()
|
|
|
|
def reply_user_chat_content(queue, config):
|
|
reply = DouyinLiveWebReply(queue, config)
|
|
reply()
|
|
|
|
if __name__ == "__main__":
|
|
freeze_support()
|
|
|
|
LiveChatConfig().update_chat_enable_status("启动中")
|
|
with Manager() as manager:
|
|
prediction_queue = manager.list() # 共享队列，用来存放预测时间（字符串）
|
|
|
|
shared_config = manager.dict({
|
|
"control_mode": 1
|
|
})
|
|
|
|
queue = PromptQueue(10)
|
|
ws_open_event = Event()
|
|
|
|
api_process = Process(
|
|
target=start_fastapi_server,
|
|
args=(shared_config,),
|
|
name="FastAPI Server"
|
|
)
|
|
fetch_process = Process(
|
|
target=fetch_user_chat_content,
|
|
args=(ws_open_event, queue),
|
|
name="Fetcher"
|
|
)
|
|
reply_process = Process(
|
|
target=reply_user_chat_content,
|
|
args=(queue, shared_config),
|
|
name="Replier"
|
|
)
|
|
|
|
api_process.start()
|
|
fetch_process.start()
|
|
ws_open_event.wait()
|
|
reply_process.start()
|
|
|
|
fetch_process.join()
|
|
reply_process.join()
|
|
api_process.join()
|