import pymysql import time import wave import requests from pyaudio import PyAudio, paInt16 from aip import AipSpeech import pyttsx3 from tkinter import * from tkinter import messagebox import speech_recognition as sr from tkinter import Text, Entry, Button framer = 16000 num_samples = 2000 channels = 1 sampwidth = 2 APP_ID = '78542969' API_KEY = 'B37LWVxLfoHixHx52OrNCPij' SECRET_KEY = 'bkNnbC37K3Ac6gI8kQ8phG1h5RIivMwj' client = AipSpeech(APP_ID, API_KEY, SECRET_KEY) audio_file = '../pythonProject6/myvoices.wav' # Class for handling audio recording and saving class Speak(): def save_wave_file(self, filepath, data): wf = wave.open(filepath, 'wb') wf.setnchannels(channels) wf.setsampwidth(sampwidth) wf.setframerate(framer) wf.writeframes(b''.join(data)) wf.close() def my_record(self): pa = PyAudio() stream = pa.open(format=paInt16, channels=channels, rate=framer, input=True, frames_per_buffer=num_samples) my_buf = [] t_end = time.time() + 5 print('识别中...') while time.time() < t_end: string_audio_data = stream.read(num_samples) my_buf.append(string_audio_data) print('识别完成') self.save_wave_file(audio_file, my_buf) stream.close() class RobotSay(): def __init__(self): self.engine = pyttsx3.init() def say(self, msg): self.engine.say(msg) self.engine.runAndWait() class ReadWav(): def __init__(self): self.APP_ID = '78542969' self.API_KEY = 'B37LWVxLfoHixHx52OrNCPij' self.SECRET_KEY = 'bkNnbC37K3Ac6gI8kQ8phG1h5RIivMwj' self.client = AipSpeech(self.APP_ID, self.API_KEY, self.SECRET_KEY) def get_file_content(self, filePath): with open(filePath, 'rb') as fp: return fp.read() def predict(self): return client.asr(self.get_file_content('../pythonProject6/myvoices.wav'), 'wav', 16000, { 'dev_pid': 1537, }) # 定义登录窗口 class LoginWindow: def __init__(self, master): self.master = master self.master.title("登录") self.master.geometry("300x200") # Center align the login window self.screen_width = self.master.winfo_screenwidth() self.screen_height = self.master.winfo_screenheight() self.x_pos = (self.screen_width - 300) // 2 self.y_pos = (self.screen_height - 200) // 2 self.master.geometry(f"300x200+{self.x_pos}+{self.y_pos}") self.label_username = Label(master, text="账号") self.label_username.pack() self.entry_username = Entry(master) self.entry_username.pack() self.label_password = Label(master, text="密码") self.label_password.pack() self.entry_password = Entry(master, show="*") self.entry_password.pack() self.button_login = Button(master, text="登录", command=self.login) self.button_login.pack() def login(self): username = self.entry_username.get() password = self.entry_password.get() if username == "123" and password == "123": messagebox.showinfo("登陆成功", "恭喜你登陆成功") self.open_chat_window() else: messagebox.showerror("登录失败", "无效的用户名或密码") def open_chat_window(self): self.master.withdraw() chat_root = Toplevel(self.master) chat_window = ChatWindow(chat_root) chat_root.protocol("WM_DELETE_WINDOW", self.master.quit) class ChatWindow: def __init__(self, master): self.master = master self.master.title("开始聊天") self.master.geometry("700x500") self.screen_width = self.master.winfo_screenwidth() self.screen_height = self.master.winfo_screenheight() self.x_pos = (self.screen_width - 300) // 2 self.y_pos = (self.screen_height - 200) // 2 self.master.geometry(f"700x500+{self.x_pos}+{self.y_pos}") self.chat_text = Text(self.master) self.chat_text.pack() self.input_field = Entry(self.master) self.input_field.pack() self.send_button = Button(self.master, text="发送", command=self.send_message) self.send_button.pack() self.view_button = Button(self.master, text="查看聊天记录", command=self.view_chat_history) self.view_button.pack() self.delete_button = Button(self.master, text="删除聊天记录", command=self.delete_chat_history) self.delete_button.pack() self.recognizer = sr.Recognizer() self.engine = pyttsx3.init() self.connection = pymysql.connect(host="localhost", user="root", password="myroot", database="ai_chatter", port=3306) self.cursor = self.connection.cursor() self.chat_history = [] self.voice_button = Button(self.master, text="语音对话", command=self.start_voice_conversation) self.voice_button.pack() self.APP_ID = APP_ID self.API_KEY = API_KEY self.SECRET_KEY = SECRET_KEY self.client = AipSpeech(APP_ID, API_KEY, SECRET_KEY) ''' nltk.download('punkt') nltk.download('stopwords') nltk.download('wordnet') ''' def send_message(self): user_input = self.input_field.get() response = self.get_qingyunke_response(user_input) self.update_chat("你: " + user_input + "\n") self.update_chat("菲菲: " + response + "\n") self.speak(response) self.save_to_database(user_input, response) def start_voice_conversation(self): # Implement voice conversation functionality self.speak("开始语音对话，请说话...") speak_instance = Speak() speak_instance.my_record() text = self.transcribe_audio('../pythonProject6/myvoices.wav') self.update_chat("你说: " + text + "\n") response = self.get_qingyunke_response(text) self.update_chat("菲菲: " + response + "\n") self.speak(response) self.save_to_database(text, response) def get_qingyunke_response(self, message): api_url = "http://api.qingyunke.com/api.php?key=free&appid=0&msg=" + message response = requests.get(api_url) if response.status_code == 200: data = response.json() if data['result'] == 0: return data['content'] else: return "对不起，我不能理解." else: return "我目前无法处理您的请求." def update_chat(self, text): self.chat_text.insert(END, text) def speak(self, text): self.engine.say(text) self.engine.runAndWait() def transcribe_audio(self, audio_file): with open(audio_file, 'rb') as f: audio_data = f.read() result = self.client.asr(audio_data, 'wav', 16000, { 'dev_pid': 1537, # Use the model for Mandarin }) if 'result' in result: return result['result'][0] else: return "Baidu speech recognition failed" #使用mysql数据库来保存 def save_to_database(self, user_message, bot_response): sql = "INSERT INTO chat_history (user_message, bot_response) VALUES (%s, %s)" text = (user_message, bot_response) self.connection.cursor().execute(sql, text) self.connection.commit() print("聊天记录保存与数据库中.") def view_chat_history(self): self.chat_text.delete('1.0', END) sql = "SELECT * FROM ai_chatter.chat_history" self.cursor.execute(sql) records = self.cursor.fetchall() for record in records: self.update_chat("你: " + record[1] + "\n") self.update_chat("菲菲: " + record[2] + "\n") if self.chat_history: self.update_chat("\n--- 聊天记录 ---\n") for message in self.chat_history: self.update_chat(message + "\n") else: self.update_chat("没有找到聊天记录\n") def delete_chat_history(self): sql = "DELETE FROM chat_history" self.cursor.execute(sql) self.connection.commit() self.chat_text.delete('1.0', END) self.update_chat("聊天记录已删除。\n") if self.chat_history: self.chat_history.clear() self.update_chat("聊天记录已删除\n") else: self.update_chat("没有找到聊天记录\n") def close_connection(self): self.cursor.close() self.connection.close() #自然语言处理 #知识图谱模块 def fetch_knowledge_graph_info(self, query): # Assuming you are using Google Knowledge Graph API api_key = 'YOUR_API_KEY' base_url = 'https://kgsearch.googleapis.com/v1/entities:search' params = { 'query': query, 'key': api_key, 'limit': 1 # You can adjust the limit based on response requirements } response = requests.get(base_url, params=params).json() if 'itemListElement' in response: item = response['itemListElement'][0] if 'result' in item: return item['result']['name'], item['result']['detailedDescription']['articleBody'] def main(): root = Tk() login_window = LoginWindow(root) root.mainloop() if __name__ == "__main__": main()