ae1d2ae292
+ 修复多个bug：消息框换行及空格问题、语音识别优化； + 彩蛋转正，Fay沟通与ChatGPT并行； + 加入yolov8姿态识别； + 加入VisualGLM-6B多模态单机离线大语言模型。
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
import os
|
||
import sys
|
||
from io import BytesIO
|
||
|
||
from PyQt5 import QtGui
|
||
from PyQt5.QtWidgets import QApplication
|
||
|
||
from ai_module import ali_nls
|
||
from core import wsa_server
|
||
from gui import flask_server
|
||
from gui.window import MainWindow
|
||
from utils import config_util
|
||
from scheduler.thread_manager import MyThread
|
||
from core.content_db import Content_Db
|
||
|
||
def __clear_samples():
|
||
if not os.path.exists("./samples"):
|
||
os.mkdir("./samples")
|
||
for file_name in os.listdir('./samples'):
|
||
if file_name.startswith('sample-') and file_name.endswith('.mp3'):
|
||
os.remove('./samples/' + file_name)
|
||
|
||
|
||
def __clear_songs():
|
||
if not os.path.exists("./songs"):
|
||
os.mkdir("./songs")
|
||
for file_name in os.listdir('./songs'):
|
||
if file_name.endswith('.mp3'):
|
||
os.remove('./songs/' + file_name)
|
||
|
||
|
||
|
||
|
||
if __name__ == '__main__':
|
||
__clear_samples()
|
||
__clear_songs()
|
||
config_util.load_config()
|
||
dbstatus = os.path.exists("fay.db")
|
||
if(dbstatus == False):
|
||
contentdb = Content_Db()
|
||
contentdb.init_db()
|
||
ws_server = wsa_server.new_instance(port=10002)
|
||
ws_server.start_server()
|
||
web_ws_server = wsa_server.new_web_instance(port=10003)
|
||
web_ws_server.start_server()
|
||
#Edit by xszyou in 20230516:增加本地asr后，aliyun调成可选配置
|
||
if config_util.ASR_mode == "ali" and config_util.config['source']['record']['enabled']:
|
||
ali_nls.start()
|
||
flask_server.start()
|
||
app = QApplication(sys.argv)
|
||
app.setWindowIcon(QtGui.QIcon('icon.png'))
|
||
win = MainWindow()
|
||
win.show()
|
||
app.exit(app.exec_())
|
||
|
||
|