提交源码

Author: yuanlei

.idea/ChatRoom.iml

@@ -0,0 +1,26 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+from chatterbot import ChatBot
+from chatterbot.trainers import ListTrainer
+from chatterbot.trainers import ChatterBotCorpusTrainer
+
+
+my_bot = ChatBot("Training demo",
+                 database="./db.sqlite3")
+
+# 直接写语句训练
+my_bot.set_trainer(ListTrainer)
+
+my_bot.train(["你叫什么名字？", "我叫小白兔！", ])
+my_bot.train([
+    "Test1",
+    "Test2",
+    "Test3",
+    "Test4",
+])
+
+# 使用自定义语句训练它
+my_bot.set_trainer(ChatterBotCorpusTrainer)
+my_bot.train("chatterbot.corpus.mytrain")
+# while True:
+#     print(my_bot.get_response(input("user:")))

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,122 @@
+import pygame
+from chatterbot import ChatBot
+import requests
+import json
+from config import *
+import time
+import os
+import random
+import urllib.request
+import base64
+
+
+# 初始化百度返回的音频文件地址，后面会变为全局变量，随需改变
+mp3_url = 'E:\Python_Doc\\voice_du\\voice_ss.mp3'
+
+
+# 播放Mp3文件
+def play_mp3():
+    # 接受服务器的消息
+    pygame.mixer.init()
+    pygame.mixer.music.load(mp3_url)
+    pygame.mixer.music.play()
+    while pygame.mixer.music.get_busy():
+        time.sleep(1)
+    pygame.mixer.music.stop()
+    pygame.mixer.quit()
+
+
+# 删除声音文件
+def remove_voice():
+    path = r"E:\Python_Doc\voice_du"
+    for i in os.listdir(path):
+        path_file = os.path.join(path, i)
+        try:
+            os.remove(path_file)
+        except:
+            continue
+
+
+# 图灵自动回复
+def tuling(info):
+    url = tuling_url + "?key=%s&info=%s" % (tuling_app_key, info)
+    content = requests.get(url, headers=headers)
+    answer = json.loads(content.text)
+    return answer['text']
+
+
+# 聊天机器人回复
+def chatbot(info):
+    my_bot = ChatBot("", read_only=True,
+                     database="./db.sqlite3")
+    res = my_bot.get_response(info)
+    return str(res)
+
+
+# 百度讲文本转为声音文件保存在本地 tts地址，无需token实时认证
+def baidu_api(answer):
+    api_url = '{11}?idx={0}&tex={1}&cuid={2}&cod={3}&lan={4}&ctp={5}&pdt={6}&spd={7}&per={8}&vol={9}&pit={10}'\
+        .format(baidu_api_set["idx"], answer, baidu_api_set["cuid"], baidu_api_set["cod"], baidu_api_set["lan"],
+                baidu_api_set["ctp"], baidu_api_set["pdt"], baidu_api_set["spd"], baidu_api_set["per"],
+                baidu_api_set["vol"], baidu_api_set["pit"], baidu_api_url)
+    res = requests.get(api_url, headers=headers2)
+    # 本地Mp3语音文件保存位置
+    iname = random.randrange(1, 99999)
+    global mp3_url
+    mp3_url = 'E:\Python_Doc\\voices\\voice_tts' + str(iname) + '.mp3'
+    with open(mp3_url, 'wb') as f:
+        f.write(res.content)
+
+
+# 百度讲文本转为声音文件保存在本地 方法2 tsn地址
+def baidu_api2(answer):
+    # 获取access_token
+    token = getToken()
+    get_url = baidu_api_url2 % (urllib.parse.quote(answer), "test", token)
+    voice_data = urllib.request.urlopen(get_url).read()
+    # 本地Mp3语音文件保存位置
+    name = random.randrange(1, 99999)
+    global mp3_url
+    mp3_url = 'E:\Python_Doc\\voice_du\\voice_tsn' + str(name) + '.mp3'
+    voice_fp = open(mp3_url, 'wb+')
+    voice_fp.write(voice_data)
+    voice_fp.close()
+    return
+
+
+# 百度语音转文本
+def getText(filename):
+    # 获取access_token
+    token = getToken()
+    data = {}
+    data['format'] = 'wav'
+    data['rate'] = 16000
+    data['channel'] = 1
+    data['cuid'] = str(random.randrange(123456, 999999))
+    data['token'] = token
+    wav_fp = open(filename, 'rb')
+    voice_data = wav_fp.read()
+    data['len'] = len(voice_data)
+    data['speech'] = base64.b64encode(voice_data).decode('utf-8')
+    post_data = json.dumps(data)
+    # 语音识别的api url
+    upvoice_url = 'http://vop.baidu.com/server_api'
+    r_data = urllib.request.urlopen(upvoice_url, data=bytes(post_data, encoding="utf-8")).read()
+    print(json.loads(r_data))
+    err = json.loads(r_data)['err_no']
+    if err == 0:
+        return json.loads(r_data)['result'][0]
+    else:
+        return json.loads(r_data)['err_msg']
+
+
+# 获取百度API调用的认证，实时生成，因为有时间限制
+def getToken():
+    # token认证的url
+    api_url = "https://openapi.baidu.com/oauth/2.0/token?" \
+                     "grant_type=client_credentials&client_id=%s&client_secret=%s"
+    token_url = api_url % (BaiDu_API_Key_GetVoi, BaiDu_Secret_Key_GetVoi)
+    r_str = urllib.request.urlopen(token_url).read()
+    token_data = json.loads(r_str)
+    token_str = token_data['access_token']
+    return token_str