import pyttsx3
import speech_recognition as sr
import datetime
import wikipedia
import webbrowser
import os
import smtplib
import sys
import random
from pygame import mixer
mixer.init()

print("Initializing Jarvis....")
MASTER = "Tony Stark"

engine = pyttsx3.init('sapi5') #
voices = engine.getProperty('voices')
engine.setProperty('voice', voices[0].id) #for male voices
# engine.setProperty('voice', voices[1].id)  #for female voice

""" RATE"""
rate = engine.getProperty('rate')   # getting details of current speaking rate
print ("current voice rate is: ",rate) #printing current voice rate
engine.setProperty('rate',170)     # setting up new voice rate
engine.runAndWait()

"""VOLUME"""
volume = engine.getProperty('volume')   #getting to know current volume level (min=0 and max=1)
print ("volume level is at : ",volume)                          #printing current volume level
engine.setProperty('volume',1.0)    # setting up volume level  between 0 and 1

popular_websites = {'google': 'https://www.google.com',
                    'youtube': 'https://www.youtube.com',
                    'wikipedia': 'https://www.wikipedia.org',
                    'amazon': 'https://www.amazon.com'}
search_engines = {'google': 'https://www.google.com',
                  'youtube': 'https://www.youtube.com',
                  'bing': 'https://www.bing.com'}


def open_url(url):
    webbrowser.open(url)
    chrome_path = 'open -a /Applications/Google\ Chrome.app %s'
    webbrowser.get(chrome_path).open(url)

def search(search_query, search_engine):
    try:
        open_url(f'{search_engines[search_engine]}/search?q={search_query}')
    except IndexError:
        open_url(f'https://www.google.com/search?q={search_query}')


def speak(text):
    engine.say(text)
    engine.runAndWait()


def wishMe():
    hour = datetime.datetime.now().hour
    # print(hour)
    if hour >= 0 and hour < 12:
        speak("Good Morning" + MASTER)

    elif hour >= 12 and hour < 18:
        speak("Good Afternoon" + MASTER)

    else:
        speak("Good Evening" + MASTER)

    # speak("Hey I am Jarvis. How may I help you")


# This is where our programme begins....
def takeCommand():
    r = sr.Recognizer()
    with sr.Microphone() as source:
        print("Listening....")
        r.pause_threshold = 0.5
        audio = r.listen(source)

    query = " "
    try:
        print("Recognizing....")
        query = r.recognize_google(audio, language='en-in')
        print("user said: " + query)

    except sr.UnknownValueError:
        print("Sorry Could You please try again")

    except Exception as e:
        print(e)
        print("Say That Again Please")
        query = None

    return query


speak("Initializing Jarvis....")
wishMe()
while True:
    query = takeCommand()

    # logic for executing basic tasks
    if 'wikipedia' in query.lower():
        speak('Searching wikipedia....')
        query = query.replace("wikipedia", "")
        results = wikipedia.summary(query, sentences=2)
        print(results)
        speak(results)

    elif "what\'s up" in query or 'how are you' in query:
        stMsgs = ['Just doing my thing!', 'I am fine!',
                  'Nice!', 'I am nice and full of energy']
        speak(random.choice(stMsgs))

    elif 'open' in query.lower():
        website = query.replace('open', '').strip().lower()
        try:
            open_url(popular_websites[website])
        except IndexError: # If the website is unknown
            print(f'Unknown website: {website}')
            speak(f'Sorry, i don\'t know the website {website}')

    elif 'search' in query.lower():
        search_query = query.split('for')[-1]
        search_engine = query.split('for')[0].replace('search', '').strip().lower()
        search(search_query, search_engine)

    elif 'mail' in query:
        speak('Who is the recipient? ')
        recipient = takeCommand()

        if 'me' in recipient:
            try:
                speak('What should I say? ')
                content = takeCommand()

                server = smtplib.SMTP('smtp.gmail.com', 587)
                server.ehlo()
                server.starttls()
                server.login("Your_Username", 'Your_Password')
                server.sendmail('Your_Username', "Recipient_Username", content)
                server.close()
                speak('Email sent!')

            except:
                speak('Sorry Sir! I am unable to send your message at this moment!')

    elif 'nothing' in query or 'abort' in query or 'stop' in query:
        speak('okay')
        speak('Bye Sir, have a good day.')
        sys.exit()

    elif 'hello' in query:
        speak('Hello Sir')

    elif 'bye' in query:
        speak('Bye Sir, have a good day.')
        sys.exit()

    elif 'play music' in query:
        music_folder = "Your_music_folder_path(absolute_path)"
        music = ['music1', 'music2', 'music3', 'music4']
        random_music = music_folder + random.choice(music) + '.mp3'
        speak('Playing your request')
        mixer.music.load(random_music)
        mixer.music.play()

    elif 'pause music' in query:
        mixer.music.pause()

    elif 'stop music' in query:
        mixer.music.stop()

    elif 'unpause' in query:
        mixer.music.unpause()

    speak('Next Command! Sir!')