# Библиотеки распознавания и синтеза речи
import speech_recognition as sr
from gtts import gTTS
# Воспроизведение речи
import pygame
from pygame import mixer
mixer.init()
import os
import sys
import time
import datetime
import logging
import webbrowser
import subprocess
class Speech_AI:
    def __init__(self):
        self._recognizer = sr.Recognizer()
        self._microphone = sr.Microphone()
        
        now_time = datetime.datetime.now()
        self._mp3_name = now_time.strftime("recordingAss_%d_%m_%Y_%I_%M_%S")+".mp3"
        self._mp3_nameold='111'
    def work(self):
        print("[log] Обработка шума...")
        with self._microphone as source:
            self._recognizer.adjust_for_ambient_noise(source)
            print("[log] Обработка шума завершина!")
        try:
            while True:
                print("[log] Жду указаний")
                with self._microphone as source:
                    audio = self._recognizer.listen(source)
                print("[log] Выполняю сэр")
                try:
                    statement = self._recognizer.recognize_google(audio, language="ru_RU")
                    statement=statement.lower()
                    # Команды для открытия различных внешних приложений
                    if((statement.find("калькулятор")!=-1) or (statement.find("calculator")!=-1) or (statement.find("открой калькулятор")!=-1)):
                        self.osrun('calc')
                               
                    if((statement.find("блокнот")!=-1) or (statement.find("notepad")!=-1) or (statement.find("открой блокнот")!=-1)):
                        self.osrun('notepad')
                    if((statement.find("browser")!=-1) or (statement.find("браузер")!=-1) or (statement.find("открой браузер")!=-1)):
                        self.openurl('http://google.com', 'Открываю браузер')
 
                    # Команды для открытия URL в браузере
                    
                    if((statement.find("youtube")!=-1) or (statement.find("youtub")!=-1) or (statement.find("открыть ютуб")!=-1) or (statement.find("смотреть ютуб")!=-1)):                        
                        self.openurl('http://youtube.com', 'Открываю ютуб')
                    if((statement.find("gmail")!=-1) or (statement.find("открой почту")!=-1)):
                        self.openurl('https://mail.google.com/mail/u/0/#inbox', 'Открываю почту')
                        
                    if((statement.find("вконтакте")!=-1) or (statement.find("открой вк")!=-1)):
                        self.openurl('http://vk.com', 'Открываю Вконтакте')
                    # Команды для поиска в сети интернет
                  
                    if((statement.find("найти")!=-1) or (statement.find("поиск")!=-1) or (statement.find("найди")!=-1)):
                        self.openurl('https://www.google.com/search?q=' + statement, "Я нашла следующие результаты")
                        
                    if((statement.find("смотреть")!=-1) and ((statement.find("фильм")!=-1) or (statement.find("film")!=-1))):
                        self.openurl('https://www.google.com/search?q=Смотреть+онлайн+фильм+' + statement, "Выберите сайт где смотреть фильм")
                    if(((statement.find("youtube")!=-1) or (statement.find("ютуб")!=-1) or (statement.find("смотреть")!=-1)) and (statement.find("на")!=-1) and (statement.find("ютуб")!=-1)):
                        self.openurl('http://www.youtube.com/results?search_query=' + statement, 'Ищу в ютуб')
                    if((statement.find("включи музыку")!=-1)):
                        os.system("C:\\Users\\Dmitriy\\Documents\\PY\\Project\\Speech_AI-master\\res\\1.mp3")
                    # Поддержание диалога
                    
                    if((statement.find("до свидания")!=-1) or (statement.find("досвидания")!=-1)):
                        answer = "Пока!"
                        self.say(str(answer))
                        while pygame.mixer.music.get_busy():
                            time.sleep(0.1)
                        sys.exit()
                    
                    print("Вы сказали: {}".format(statement))
                    
                except sr.UnknownValueError:
                    print("Упс! Кажется, я тебя не поняла, повтори еще раз")
                except sr.RequestError as e:
                    print("Не могу получить данные от сервиса Google Speech Recognition; {0}".format(e))
        except KeyboardInterrupt:
            self._clean_up()
            print("Пока!")
        
    def osrun(self, cmd):
        PIPE = subprocess.PIPE
        p = subprocess.Popen(cmd, shell=True, stdin=PIPE, stdout=PIPE, stderr=subprocess.STDOUT)
    def openurl(self, url, ans):
        webbrowser.open(url)
        self.say(str(ans))
        while pygame.mixer.music.get_busy():
            time.sleep(0.1)
    def say(self, phrase):
        tts = gTTS(text=phrase, lang="ru")
        tts.save(self._mp3_name)
        # Play answer
        mixer.music.load(self._mp3_name)
        mixer.music.play()
        if(os.path.exists(self._mp3_nameold)):
            os.remove(self._mp3_nameold)
       
        now_time = datetime.datetime.now()
        self._mp3_nameold=self._mp3_name
        self._mp3_name = now_time.strftime("recordingAss_%d_%m_%Y_%I_%M_%S")+".mp3"
    
    def _clean_up(self):
        def clean_up():
            os.remove(self._mp3_name)
def main():
    ai = Speech_AI()
    ai.work()
main()