debug.py

# -*- coding: utf-8-*-
import sys, os, time, random
import re
import json
import yaml
import argparse
import logging
import psutil
from multiprocessing import Process, Queue, Pipe

from lib.graphic.baiduGraphic import BaiduGraphic
from lib.voice.baiduVoice import BaiduVoice
from lib.voice.baseVoice import AbstractVoiceEngine
from plugin.bootstrap import Bootstrap
import lib.appPath
import lib.util
import plugin.volume.pulseAudio 
from plugin.fm.doubanFM import DoubanFM
from lib.mail import SMTPMail
from lib.aiChat import OpenaiChat
from plugin.monitor.people import PeopleMonitor
from plugin.feeds.jiqizhixin import JiqizhixinFeed
import plugin.feeds.jiqizhixin

def doubanFM(logger,args):
    speaker = BaiduVoice.get_instance()
    douban_fm = DoubanFM.get_instance()
    douban_fm.set_speaker(speaker)
    for i in range(0,2):
        song = douban_fm.playRandomLikeSong()
    

def pulseAudio(logger,args):
    baidu_voice = BaiduVoice.get_instance()

    out_to_fp, in_to_fp = Pipe(True)
    out_from_fp, in_from_fp = Pipe(True)
    
    son_p = Process(target=Bootstrap.son_process, 
                args=(baidu_voice, 
                    (out_to_fp, in_to_fp),
                    (out_from_fp, in_from_fp),
                plugin.volume.pulseAudio.son_process_handle,False))

    son_p.start()
    # 等to_pipe被fork 后，关闭主进程的输出端; 创建的Pipe一端连接着主进程的输入，一端连接着子进程的输出口
    out_to_fp.close()
    # 等from_pipe被fork 后，关闭主进程的输入端; 创建的Pipe一端连接着子进程的输入，一端连接着父进程的输出口
    in_from_fp.close()

    words = [
            u"打开声音",
            u"声音小一点", 
            u"声音小点", 
            u"声音再小一点", 
            u"声音大点",
            u"声音再大一点", 
            u"静音",
            u"打开声音",
            u"安静",
            u"打开声音",
            #u"声音放到最大",
        ]

    for text in words:
        is_valid = plugin.volume.pulseAudio.isValid(text)
        if is_valid is True:
            logger.debug("word %s is valid" % text)
            plugin.volume.pulseAudio.process_handle(text,in_to_fp,out_from_fp,son_p,baidu_voice)
            time.sleep(3)
        else:
            logger.debug("word %s is not valid" % text)

    in_to_fp.close()
    out_from_fp.close()
    son_p.join()

    logger.debug("debug pulseAudio is over")

def mail(logger,args):
    smtpMail = SMTPMail.get_instance()
    with open('./mind-idea.jpg', 'rb') as f:
        smtpMail.sendImageEmail(f.read())

    logger.debug("debug mail is over")

def peopleMonitor(logger,args):
    speaker = BaiduVoice.get_instance()
    people_monitor = PeopleMonitor.get_instance()
    people_monitor.set_speaker(speaker)

    def get_text_callback():
        index = random.choice([0,1])
        test_words = [
                u'打开人体监控',
                u'结束人体监控',
            ]
        logger.debug("index %d, text:%s",index,test_words[index])
        time.sleep(5)
        return test_words[index]
    people_monitor.start(get_text_callback)

    logger.debug("debug peopleMonitor is over")

def baiduGraphic(logger,args):
    def get_file_content(filePath):
        with open(filePath, 'rb') as fp:
            return fp.read()
    baidu_graphic = BaiduGraphic.get_instance()

    for detect_type in ["plant","dish","car","logo","animal","object","face"]:
        file = os.path.join(lib.appPath.APP_PATH, '.'.join([detect_type,'jpg']))
        img = get_file_content(file)
        res = baidu_graphic.detectImage(img,detect_type)
        logger.debug("%s: %s",detect_type,json.dumps(res,encoding="UTF-8",ensure_ascii=False))

    logger.debug("debug baiduGraphic is over")

def jiqizhixinFeed(logger,args):
    speaker = BaiduVoice.get_instance()

    out_to_fp, in_to_fp = Pipe(True)
    out_from_fp, in_from_fp = Pipe(True)
    
    son_p = Process(target=Bootstrap.son_process, 
                args=(speaker,
                    (out_to_fp, in_to_fp),
                    (out_from_fp, in_from_fp),
                plugin.feeds.jiqizhixin.son_process_handle,False))

    son_p.start()

    # 等to_pipe被fork 后，关闭主进程的输出端; 创建的Pipe一端连接着主进程的输入，一端连接着子进程的输出口
    out_to_fp.close()
    # 等from_pipe被fork 后，关闭主进程的输入端; 创建的Pipe一端连接着子进程的输入，一端连接着父进程的输出口
    in_from_fp.close()

    debug_words = [
            u"阅读机器之心新闻",
            u"阅读下一条", 
            u"下一条", 
            u"下一条", 
            u"结束阅读",
        ]
    for text in debug_words:
        is_valid = plugin.feeds.jiqizhixin.isValid(text)
        if is_valid is True:
            if any(word in text for word in [u'结束阅读',u'阅读机器之心']):
                time.sleep(60)

            plugin.feeds.jiqizhixin.process_handle(text,in_to_fp,out_from_fp,son_p,speaker)

            if any(word in text for word in [u'结束阅读']): break

            time.sleep(7)
        else:
            print("word %s is not valid" % text)

    in_to_fp.close()
    out_from_fp.close()
    son_p.join()

    '''
    instance = JiqizhixinFeed.get_instance()
    instance.set_speaker(speaker)
    instance.update_feeds()
    ct = instance.get_feeds_count()
    for i in range(0,ct):
        instance.get_next_feed()
    '''

    logger.debug("debug jiqizhixinFeed is over")

def debugOpenai(logger,args):
    print("debugOpenai")

    bootstrap_file = os.path.join(lib.appPath.CONFIG_PATH, 'bootstrap.yml')
    if os.path.exists(bootstrap_file) is False:
        logger.error("bootstrap file is not exists!")
        return False

    # Read config
    logger.debug("Trying to read config file: '%s'", bootstrap_file)
    try:
        with open(bootstrap_file, "r") as f:
            bootstrap_config = yaml.safe_load(f)
    except OSError:
        logger.error("Can't open config file: '%s'", bootstrap_file)
        raise

    ai_chat_engine = bootstrap_config['ai_chat_engine']
    robot = lib.util.get_engine_by_tag(ai_chat_engine,cate="aiChat").get_instance()
    #robot = OpenaiChat.get_instance()

    stream = robot.stream_chat("你好！")
    #stream = robot.stream_chat("LLM中transform是什么")

    speaker = BaiduVoice.get_instance()
    speaker.stream_say(stream)

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='debug')
    parser.add_argument('--debug', action='store_true',
                        help='Show debug messages')
    parser.add_argument('--pulseAudio', action='store_true',
                        help='Show debug pulse audio plugin messages')
    parser.add_argument('--doubanFM', action='store_true',
                        help='Show debug douban fm plugin messages')
    parser.add_argument('--mail', action='store_true',
                        help='Show debug mail lib messages')
    parser.add_argument('--peopleMonitor', action='store_true',
                        help='Show debug people monitor plugin messages')
    parser.add_argument('--baiduGraphic', action='store_true',
                        help='Show debug baidu graphic lib messages')
    parser.add_argument('--jiqizhixinFeed', action='store_true',
                        help='Show debug jiqizhixinFeed plugin messages')
    parser.add_argument('--debugOpenai', action='store_true',
                        help='Show debug openai')


    args = parser.parse_args()
    logging.basicConfig(stream=sys.stdout)
    logger = logging.getLogger("")
    if args.debug:
        logger.setLevel(logging.DEBUG)
    if args.pulseAudio:
        pulseAudio(logger,args)
        exit(0)

    if args.doubanFM:
        doubanFM(logger,args)
        exit(0)

    if args.mail:
        mail(logger,args)
        exit(0)

    if args.peopleMonitor:
        peopleMonitor(logger,args)
        exit(0)

    if args.baiduGraphic:
        baiduGraphic(logger,args)
        exit(0)

    if args.jiqizhixinFeed:
        jiqizhixinFeed(logger,args)
        exit(0)

    if args.debugOpenai:
        debugOpenai(logger,args)
        exit(0)