import importlib import json import time import os import pyaudio import re from flask import Flask, render_template, request, jsonify, Response, send_file from flask_cors import CORS import requests import fay_booter from tts import tts_voice from gevent import pywsgi from scheduler.thread_manager import MyThread from utils import config_util, util from core import wsa_server from core import fay_core from core import content_db from core.interact import Interact from core import member_db import fay_booter from flask_httpauth import HTTPBasicAuth __app = Flask(__name__) auth = HTTPBasicAuth() CORS(__app, supports_credentials=True) def load_users(): with open('verifier.json') as f: users = json.load(f) return users users = load_users() @auth.verify_password def verify_password(username, password): if not users or config_util.start_mode == 'common': return True if username in users and users[username] == password: return username def __get_template(): return render_template('index.html') def __get_device_list(): if config_util.start_mode == 'common': audio = pyaudio.PyAudio() device_list = [] for i in range(audio.get_device_count()): devInfo = audio.get_device_info_by_index(i) if devInfo['hostApi'] == 0: device_list.append(devInfo["name"]) return list(set(device_list)) else: return [] @__app.route('/api/submit', methods=['post']) def api_submit(): data = request.values.get('data') config_data = json.loads(data) if(config_data['config']['source']['record']['enabled']): config_data['config']['source']['record']['channels'] = 0 audio = pyaudio.PyAudio() for i in range(audio.get_device_count()): devInfo = audio.get_device_info_by_index(i) if devInfo['name'].find(config_data['config']['source']['record']['device']) >= 0 and devInfo['hostApi'] == 0: config_data['config']['source']['record']['channels'] = devInfo['maxInputChannels'] config_util.save_config(config_data['config']) return '{"result":"successful"}' @__app.route('/api/get-data', methods=['post']) def api_get_data(): config_util.load_config() voice_list = tts_voice.get_voice_list() send_voice_list = [] if config_util.tts_module == 'ali': wsa_server.get_web_instance().add_cmd({ "voiceList": [ {"id": "abin", "name": "阿斌"}, {"id": "zhixiaobai", "name": "知小白"}, {"id": "zhixiaoxia", "name": "知小夏"}, {"id": "zhixiaomei", "name": "知小妹"}, {"id": "zhigui", "name": "知柜"}, {"id": "zhishuo", "name": "知硕"}, {"id": "aixia", "name": "艾夏"}, {"id": "zhifeng_emo", "name": "知锋_多情感"}, {"id": "zhibing_emo", "name": "知冰_多情感"}, {"id": "zhimiao_emo", "name": "知妙_多情感"}, {"id": "zhimi_emo", "name": "知米_多情感"}, {"id": "zhiyan_emo", "name": "知燕_多情感"}, {"id": "zhibei_emo", "name": "知贝_多情感"}, {"id": "zhitian_emo", "name": "知甜_多情感"}, {"id": "xiaoyun", "name": "小云"}, {"id": "xiaogang", "name": "小刚"}, {"id": "ruoxi", "name": "若兮"}, {"id": "siqi", "name": "思琪"}, {"id": "sijia", "name": "思佳"}, {"id": "sicheng", "name": "思诚"}, {"id": "aiqi", "name": "艾琪"}, {"id": "aijia", "name": "艾佳"}, {"id": "aicheng", "name": "艾诚"}, {"id": "aida", "name": "艾达"}, {"id": "ninger", "name": "宁儿"}, {"id": "ruilin", "name": "瑞琳"}, {"id": "siyue", "name": "思悦"}, {"id": "aiya", "name": "艾雅"}, {"id": "aimei", "name": "艾美"}, {"id": "aiyu", "name": "艾雨"}, {"id": "aiyue", "name": "艾悦"}, {"id": "aijing", "name": "艾婧"}, {"id": "xiaomei", "name": "小美"}, {"id": "aina", "name": "艾娜"}, {"id": "yina", "name": "伊娜"}, {"id": "sijing", "name": "思婧"}, {"id": "sitong", "name": "思彤"}, {"id": "xiaobei", "name": "小北"}, {"id": "aitong", "name": "艾彤"}, {"id": "aiwei", "name": "艾薇"}, {"id": "aibao", "name": "艾宝"}, {"id": "shanshan", "name": "姗姗"}, {"id": "chuangirl", "name": "小玥"}, {"id": "lydia", "name": "Lydia"}, {"id": "aishuo", "name": "艾硕"}, {"id": "qingqing", "name": "青青"}, {"id": "cuijie", "name": "翠姐"}, {"id": "xiaoze", "name": "小泽"}, {"id": "zhimao", "name": "知猫"}, {"id": "zhiyuan", "name": "知媛"}, {"id": "zhiya", "name": "知雅"}, {"id": "zhiyue", "name": "知悦"}, {"id": "zhida", "name": "知达"}, {"id": "zhistella", "name": "知莎"}, {"id": "kelly", "name": "Kelly"}, {"id": "jiajia", "name": "佳佳"}, {"id": "taozi", "name": "桃子"}, {"id": "guijie", "name": "柜姐"}, {"id": "stella", "name": "Stella"}, {"id": "stanley", "name": "Stanley"}, {"id": "kenny", "name": "Kenny"}, {"id": "rosa", "name": "Rosa"}, {"id": "mashu", "name": "马树"}, {"id": "xiaoxian", "name": "小仙"}, {"id": "yuer", "name": "悦儿"}, {"id": "maoxiaomei", "name": "猫小美"}, {"id": "aifei", "name": "艾飞"}, {"id": "yaqun", "name": "亚群"}, {"id": "qiaowei", "name": "巧薇"}, {"id": "dahu", "name": "大虎"}, {"id": "ailun", "name": "艾伦"}, {"id": "jielidou", "name": "杰力豆"}, {"id": "laotie", "name": "老铁"}, {"id": "laomei", "name": "老妹"}, {"id": "aikan", "name": "艾侃"} ] }) elif config_util.tts_module == 'volcano': wsa_server.get_web_instance().add_cmd({ "voiceList": [ {"id": "BV001_streaming", "name": "通用女声"}, {"id": "BV002_streaming", "name": "通用男声"}, {"id": "zh_male_jingqiangkanye_moon_bigtts", "name": "京腔侃爷/Harmony"}, {"id": "zh_female_shuangkuaisisi_moon_bigtts", "name": "爽快思思/Skye"}, {"id": "zh_male_wennuanahu_moon_bigtts", "name": "温暖阿虎/Alvin"}, {"id": "zh_female_wanwanxiaohe_moon_bigtts", "name": "湾湾小何"}, ] }) else: voice_list = tts_voice.get_voice_list() send_voice_list = [] for voice in voice_list: voice_data = voice.value send_voice_list.append({"id": voice_data['name'], "name": voice_data['name']}) wsa_server.get_web_instance().add_cmd({ "voiceList": send_voice_list }) wsa_server.get_web_instance().add_cmd({"deviceList": __get_device_list()}) if fay_booter.is_running(): wsa_server.get_web_instance().add_cmd({"liveState": 1}) return json.dumps({'config': config_util.config, 'voice_list' : send_voice_list}) @__app.route('/api/start-live', methods=['post']) def api_start_live(): # time.sleep(5) fay_booter.start() time.sleep(1) wsa_server.get_web_instance().add_cmd({"liveState": 1}) return '{"result":"successful"}' @__app.route('/api/stop-live', methods=['post']) def api_stop_live(): # time.sleep(1) fay_booter.stop() time.sleep(1) wsa_server.get_web_instance().add_cmd({"liveState": 0}) return '{"result":"successful"}' @__app.route('/api/send', methods=['post']) def api_send(): data = request.values.get('data') info = json.loads(data) interact = Interact("text", 1, {'user': info['username'], 'msg': info['msg']}) util.printInfo(3, "文字发送按钮", '{}'.format(interact.data["msg"]), time.time()) fay_booter.feiFei.on_interact(interact) return '{"result":"successful"}' #获取指定用户的消息记录 @__app.route('/api/get-msg', methods=['post']) def api_get_Msg(): data = request.form.get('data') data = json.loads(data) uid = member_db.new_instance().find_user(data["username"]) contentdb = content_db.new_instance() if uid == 0: return json.dumps({'list': []}) else: list = contentdb.get_list('all','desc',1000, uid) relist = [] i = len(list)-1 while i >= 0: relist.append(dict(type=list[i][0], way=list[i][1], content=list[i][2], createtime=list[i][3], timetext=list[i][4], username=list[i][5])) i -= 1 if fay_booter.is_running(): wsa_server.get_web_instance().add_cmd({"liveState": 1}) return json.dumps({'list': relist}) @__app.route('/v1/chat/completions', methods=['post']) @__app.route('/api/send/v1/chat/completions', methods=['post']) def api_send_v1_chat_completions(): data = request.json last_content = "" if 'messages' in data and data['messages']: last_message = data['messages'][-1] username = last_message.get('role', 'User') if username == 'user': username = 'User' last_content = last_message.get('content', 'No content provided') else: last_content = 'No messages found' username = 'User' model = data.get('model', 'fay') interact = Interact("text", 1, {'user': username, 'msg': last_content}) util.printInfo(3, "文字沟通接口", '{}'.format(interact.data["msg"]), time.time()) text = fay_booter.feiFei.on_interact(interact) if model == 'fay-streaming': return stream_response(text) else: return non_streaming_response(last_content, text) @__app.route('/api/get-member-list', methods=['post']) def api_get_Member_list(): memberdb = member_db.new_instance() list = memberdb.get_all_users() return json.dumps({'list': list}) def stream_response(text): def generate(): for chunk in text_chunks(text): message = { "id": "chatcmpl-8jqorq6Fw1Vi5XoH7pddGGpQeuPe0", "object": "chat.completion.chunk", "created": int(time.time()), "model": "fay-streaming", "choices": [ { "delta": { "content": chunk }, "index": 0, "finish_reason": None } ] } yield f"data: {json.dumps(message)}\n\n" time.sleep(0.1) # 发送最终的结束信号 yield 'data: [DONE]\n\n' return Response(generate(), mimetype='text/event-stream') def non_streaming_response(last_content, text): return jsonify({ "id": "chatcmpl-8jqorq6Fw1Vi5XoH7pddGGpQeuPe0", "object": "chat.completion", "created": int(time.time()), "model": "fay", "choices": [ { "index": 0, "message": { "role": "assistant", "content": text }, "logprobs": "", "finish_reason": "stop" } ], "usage": { "prompt_tokens": len(last_content), "completion_tokens": len(text), "total_tokens": len(last_content) + len(text) }, "system_fingerprint": "fp_04de91a479" }) def text_chunks(text, chunk_size=20): pattern = r'([^.!?;:，。！？]+[.!?;:，。！？]?)' chunks = re.findall(pattern, text) for chunk in chunks: yield chunk @__app.route('/', methods=['get']) @auth.login_required def home_get(): return __get_template() @__app.route('/', methods=['post']) @auth.login_required def home_post(): wsa_server.get_web_instance.add_cmd({"is_connect": wsa_server.get_instance().isConnect}) #TODO 不应放这里，同步数字人连接状态 return __get_template() @__app.route('/setting', methods=['get']) def setting(): return render_template('setting.html') #输出的音频http @__app.route('/audio/') def serve_audio(filename): audio_file = os.path.join(os.getcwd(), "samples", filename) return send_file(audio_file) #输出的表情git @__app.route('/robot/') def serve_gif(filename): gif_file = os.path.join(os.getcwd(), "gui", "robot", filename) return send_file(gif_file) def run(): server = pywsgi.WSGIServer(('0.0.0.0',5000), __app) server.serve_forever() def start(): MyThread(target=run).start()