rihanna.py

import random as r
import re
import time
import traceback

import pyttsx3
import requests
import wikipedia
from chatterbot import ChatBot
from chatterbot.trainers import ListTrainer
from selenium import webdriver

import config
from rihanna_bot import rihanna_football, rihanna_speak, rihanna_tweet, rihanna_news, rihanna_skype, rihanna_one_char, \
    rihanna_time, rihanna_maths as calc, rihanna_email, rihanna_tfl, rihanna_spell, rihanna_facebook, rihanna_amazon, \
    rihanna_dict, rihanna_iot, rihanna_wc, rihanna_sound_cloud, ri_news, rihanna_science, rihanna_maps, \
    rihanna_man, rihanna_job, ri_youtube, ri_image, rihanna_windows, ri_docker, rihanna_nhs, hot100, \
    rihanna_movies, rihanna_lyrics, rihanna_spotify, rihanna_itunes, rihanna_docx, rihanna_games, rihanna_covid, \
    rihanna_movieRecommender, ri_sensor, ri_windowsDocker, rihanna_Trivia, rihanna_mma

bot = ChatBot('Bot', storage_adapter='chatterbot.storage.SQLStorageAdapter',
              logic_adapters=[
                  {'import_path': 'chatterbot.logic.BestMatch'},
                  {'import_path': 'chatterbot.logic.LowConfidenceAdapter',
                   'threshold': 0.50,
                   'default_response': 'I am sorry. I am not allowed to answer that question'
                   }
              ],
              trainer='chatterbot.trainers.ListTrainer')
bot.set_trainer(ListTrainer)

break_words = ["yes", "no", "okay", "yeah", "ok", "nah", "alright", "i see", "really"]
_date = ("what is the date", "what is todays date", "todays date", "current date", "date")
_time = ("what is the time", "time", "what is the current time", "current time")
email = {'msg': '', 'address': '', 'subject': '', 'run': 0}
run_email = {1: 'which email address do you want to send to?', 2: 'what is the subject?',
             3: 'what do you wish to send to '}
laugh = ["haha", "lol", "hahaha", "ha", "lool"]
IBN = 0
heart_effect = 0


def email_thread(message):
    global email

    if email['run'] == 4:
        msg = message
        address = email['address']
        subject = email['subject']
        email = {'msg': '', 'address': '', 'subject': '', 'run': 0}
        return rihanna_email.send_email(subject=subject, msg=msg, _send_email=address)

    elif email['run'] == 3:
        email['subject'] = message
        reply = run_email[3] + email['address']
        return {'display': reply, 'say': reply}

    elif email['run'] == 1:
        reply = run_email[1]
        return {'display': reply, 'say': reply}

    elif email['run'] == 2:
        if rihanna_email.check(message) == 'valid':
            email['address'] = message
            reply = run_email[2]
            return {'display': reply, 'say': reply}
        elif message in rihanna_email.contact:
            email['address'] = rihanna_email.contact[message]
            reply = run_email[2]
            return {'display': reply, 'say': reply}
        else:
            email['run'] -= 1
            reply = f"{message} is an invalid email, please give a valid mail"
            return {'display': reply, 'say': reply}


def rihanna_voice(word_speech):
    engine = pyttsx3.init()
    engine.say(word_speech)
    engine.runAndWait()


def google_search(query):
    try:
        driver = webdriver.Chrome(executable_path=r"chrome_driver\chromedriver.exe")
        google = "https://www.google.com/search?q="
        search = google + query
        driver.get(search)
        return driver
    except Exception as e:
        return "Web Driver Failed"


def play_song(song):
    try:
        driver = webdriver.Chrome(executable_path=r"chrome_driver/chromedriver.exe")
        query = "https://www.youtube.com/results?search_query="
        search = query + song
        driver.get(search)
        driver.find_element_by_xpath(
            xpath='//*[@id="yDmH0d"]/c-wiz/div/div/div/div[2]/div[1]/div[4]/form/div[1]/div/button').click()
        driver.find_element_by_xpath(xpath='//*[@id="dismissible"]').click()
    except:
        print('Rihanna Detected a Bug')
        traceback.print_exc()


def add_heart_effect():
    global heart_effect
    say = 'I am not programmed to love. However, your mouse cursor is'

    if heart_effect == 0:
        display = f'<script>putMouse()</script> {say}'

        heart_effect = 1
    else:
        display = "<img src='img/heart.png' width='70px'>"
        say = 'I know, you have already mentioned it'
    return {'display': display, 'say': say}


def remove_heart_effect():
    global heart_effect
    say = 'Oh well. It was good while it lasted. Mouse, Deactivate!'

    if heart_effect == 1:
        display = f'<script>removeMouse()</script> {say}'

        heart_effect = 0
    else:
        display = say = f"That doesnt hurt my feelings. Because, I dont have one"
    return {'display': display, 'say': say}


def format_string(string):
    d = "!?\|:;@'][<>"

    for c in d:
        if c in string:
            string = string.replace(c, '')
    return string


def weather_format(msg):
    image = {'rain': 'rain.webp', 'clear sky': 'sun.png', 'snow': 'snow.gif', 'thunder': 'thunder_storm.png',
             'sun': 'sun_cloud.png', 'storm': 'thunder_storm.png', 'clouds': 'cloud.png'}
    for key in image:
        if key in msg:
            img = image[key]
            display = f"<div style='background-color:#69BFF7; width:600px; height:250px;'>" \
                      f"<img src='img/weather/{img}' width='200px' alt='{key}' style='display:block; margin-left:auto; " \
                      f"margin-right:auto;'>" \
                      f"</div><div style='background-color:black; width:600px; text-align:center; " \
                      f"color:white; font-size:18px;'>{msg}</div>"
            return {'display': display, 'say': msg}
    display = f"<div style='background-color:black; width:600px; text-align:center; color:white; font-size:18px;'>" \
              f"{msg}</div>"
    return {'display': display, 'say': msg}


def weather_f(json_data):
    from datetime import timedelta
    # image = {'rain': 'rain.gif', 'clear sky': 'clear_sky.jpeg', 'snow': 'snow.gif', 'thunder': 'thunderstorm.jpg',
    #          'sun': 'sun_clouds.jpg', 'storm': 'thunderstorm.jpg', 'clouds': 'clouds.jpg', 'other': 'other.jpg'}
    image = {'rain': 'rain.gif', 'clear sky': 'clear_sky.gif', 'snow': 'snow.gif', 'thunder': 'thunder.gif',
             'sun': 'sun.gif', 'storm': 'thunder.gif', 'broken clouds': 'clouds.gif',
             'scattered clouds': 'scattered_clouds.gif', 'other': 'other.jpg'}
    night = ['night1.gif', 'night.gif']
    img = ''
    sr = time.localtime(json_data['sys']['sunrise'])
    ss = time.localtime(json_data['sys']['sunset'])
    tt = time.localtime(json_data['dt'] + json_data['timezone'] - 3600)
    d_sr = timedelta(minutes=sr.tm_hour * 60 + sr.tm_min)
    d_ss = timedelta(minutes=ss.tm_hour * 60 + ss.tm_min)
    d_tt = timedelta(minutes=tt.tm_hour * 60 + tt.tm_min)
    # print(f'd_sr = {d_sr} \nd_ss = {d_ss} \nd_tt = {d_tt}')
    if d_sr > d_ss:
        # print('swapped')
        sr, ss = ss, sr
        d_sr, d_ss = d_ss, d_sr
    if (d_tt < d_sr) or (d_tt > d_ss):
        img = r.choice(night)
    else:
        for key in image:
            if key in json_data['weather'][0]['description']:
                img = image[key]
                break
    if img == '':
        img = image['other']
    desc = json_data['weather'][0]['description']
    wind = json_data['wind']['speed']
    temp_c = round(json_data['main']['temp'] - 273)
    temp_min = round(json_data['main']['temp_min'] - 273)
    temp_max = round(json_data['main']['temp_max'] - 273)
    hum = json_data['main']['humidity']
    city = json_data['name']
    # time_now = time.strftime("%H:%M", time.localtime(json_data['dt']))
    # time_n = datetime.now(timezone(tzwhere.tzwhere().tzNameAt(json_data['coord']['lat'], json_data['coord']['lon'])))
    time_t = time.strftime("%H:%M", tt)
    sunrise = time.strftime("%H:%M", sr)
    sunset = time.strftime("%H:%M", ss)

    forecast = f"{desc.title()} in {city} at {time_t} O'clock. The temperature is {temp_c}° " \
               f"celcius with wind speed of {wind}"
    display = f"<table style='width:650px; text-align:center; background-color:black;'>" \
              f"<tr>" \
              f"<td><img src='img/weather/icons/temp.png' height='30px'></td> " \
              f"<td style='color:white;'>{temp_c}°C</td>" \
              f"<td><img src='img/weather/icons/min_temp.png' height='30px'></td> " \
              f"<td style='color:white;'>{temp_min}°C</td>" \
              f"<td><img src='img/weather/icons/max_temp.png' height='30px'></td> " \
              f"<td style='color:white;'>{temp_max}°C</td>" \
              f"<td><img src='img/weather/icons/hum.webp' height='30px'></td> " \
              f"<td style='color:white;'>{hum}%</td>" \
              f"<td><img src='img/weather/icons/wind.webp' height='30px'></td> " \
              f"<td style='color:white;'>{wind}</td>" \
              f"<td><img src='img/weather/icons/sunrise2.png' height='30px'></td> " \
              f"<td style='color:white;'>{sunrise}</td>" \
              f"<td><img src='img/weather/icons/sunset.svg' height='30px'></td> " \
              f"<td style='color:white;'>{sunset}</td>	" \
              f"</tr>" \
              f"</table>" \
              f"<div style='width: 650px; '>" \
              f"<img src='img/weather/new/{img}' width='650px'>" \
              f"</div>" \
              f"<table style='width:650px;'>" \
              f"<tr>" \
              f"<td style='text-align:center; background-color:black; color:white;'>{forecast}</td>" \
              f"</tr>" \
              f"</table>"
    return {'display': display, 'say': forecast}


def weather(place):
    try:
        api_address = f'http://api.openweathermap.org/data/2.5/weather?appid={config.weather_id}='
        word = re.findall('[a-z]+', place)
        if len(word) == 1:
            city = word[0]
        elif len(word) == 3:
            city = f'{word[0]} {word[1]},{word[2]}'
        else:
            city = place

        url = api_address + city

        json_data = requests.get(url).json()
        # print(json_data)
        # desc = json_data['weather'][0]['description']
        # temp_f = json_data['main']['temp']
        # wind = json_data['wind']['speed']
        #
        # temp_c = round(temp_f - 273)
        #
        # forecast = f"{desc} in {city}. The temperature is {temp_c}° celcius with wind speed of {wind}"
        return weather_f(json_data)

    except:
        reply = 'Sorry could not find location {}'.format(place)
        return {'display': reply, 'say': reply}


def stop_words():
    response = ["okay", "ok", "alright", "great", "Thought as much", "Good"]

    reply = response[r.randrange(len(response))]
    return {'display': reply, 'say': reply}


def funny():
    response = ["its not that funny", "lol", "i am surprised you have a sense of humor",
                "funny eh?", "I know, I am hilarious", "haha"]

    reply = response[r.randrange(len(response))]
    return {'display': reply, 'say': reply}


def sing_song():
    display = """
    <p>i dont sing, i rap</p>
    <audio autoplay>
      <source src="audio/rihanna_song.mp3" type="audio/mpeg">
      Your browser does not support the audio element.
    </audio>
    """
    say = 'i dont sing, i rap'
    return {'display': display, 'say': say}


def rihanna(message):
    global IBN
    if message.lower()[:len('man')] == 'man':
        return rihanna_man.selector(format_string(message).lower().strip())
    # if (message[:len("dictionary translate")] != "dictionary translate") and (
    #         rihanna_dict.detect_lang(message) != 'en') and ('spotify' not in message) and ('covid' not in message) and \
    #         ('rihanna' not in message) and ('solve' not in message):
    #     config.lang_code = rihanna_dict.detect_lang(message)
    #     print('lang=>', config.lang_code)
    #     message = rihanna_dict.translate_sentence_code(query=message, lang='en')['display']

    # print(f'trans: {message} \n l_code: {lang_code}')
    if message.lower() == 'ibn create a web portfolio':
        IBN = 1
        reply = 'I will need your github username'
        return {'display': reply, 'say': reply}
    elif IBN == 1:
        IBN = 0
        return ri_windowsDocker.selector(message, 1)

    if email['run'] == 0:
        if message.lower()[:3] == 'tfl':
            message = format_string(message).lower().strip()

        elif message[:len('word file')] == 'word file':
            return rihanna_docx.selector(message)
        elif message.lower()[:len('mma')] == 'mma':
            msg = message.lower()[len('mma '):]
            return rihanna_mma.selector(msg)

        elif message.lower()[:len('trivia')] == 'trivia':
            return rihanna_Trivia.selector(message.lower().strip())
        elif message.lower()[:5] == 'skype':
            return rihanna_skype.selector(format_string(message[6:]).lower().strip())
        elif message.lower()[:len('solve')] == 'solve':
            return rihanna_science.selector(message)
        elif message.lower()[:len('rihanna ')] == 'rihanna ':
            msg = message.lower()[len('rihanna '):]
            return rihanna_science.selector(msg)
        elif message[:len('map')] == 'map':
            return rihanna_maps.selector(format_string(message))
        elif message[:len('covid')] == 'covid':
            return rihanna_covid.selector(format_string(message).lower())
        elif message[:len('lyrics')] == 'lyrics':
            return rihanna_lyrics.selector(format_string(message))
        elif message[:len('billboard')] == 'billboard':
            return hot100.selector(message)
        elif message.lower()[:len('iot')] == 'iot':
            return rihanna_iot.selector(format_string(message).lower().strip())
        elif message.lower()[:len('ibn')] == 'ibn':
            return ri_windowsDocker.selector(format_string(message[3:]).lower().strip(), IBN)
        elif message.lower()[:len('recommend movies')] == 'recommend movies':
            return rihanna_movieRecommender.selector(format_string(message).lower().strip()[len('recommend movies'):])
        elif message[-len('movies'):] == 'movies':
            return rihanna_movies.selector(format_string(message))
        elif message.lower()[:len('amazon')] == 'amazon':
            return rihanna_amazon.selector(format_string(message).lower().strip())
        elif message.lower()[:len('youtube')] == 'youtube':
            msg = format_string(message).lower().strip()
            return ri_youtube.selector(msg)
        elif message.lower()[:len('itunes')] == 'itunes':
            msg = format_string(message).lower().strip()[len('itunes '):]
            return rihanna_itunes.selector(msg)
        elif message.lower()[:len('sound cloud')] == 'sound cloud':
            msg = format_string(message).lower().strip()
            return rihanna_sound_cloud.selector(msg)
        elif (message.lower()[:len('spotify ')] == 'spotify') or (message.lower()[-len('spotify'):] == 'spotify'):
            msg = format_string(message).lower().strip()
            return rihanna_spotify.selector(msg)
        elif message.lower()[:len('dictionary')] == 'dictionary':
            return rihanna_dict.selector(format_string(message).lower().strip())
        elif message.lower()[:len('job search')] == 'job search':
            return rihanna_job.selector(format_string(message).lower().strip())
        elif message.lower()[:len('docker')] == 'docker':
            return ri_docker.selector(message.lower().strip())
        elif message.lower()[:len('game play')] == 'game play':
            return rihanna_games.selector(format_string(message).lower().strip())
        elif message.strip().lower() in laugh:
            return funny()
        else:
            message = rihanna_spell.auto_correct(format_string(message).lower().strip())
    else:
        message = rihanna_spell.auto_correct(message.lower().strip())

    # Main Decision Thread
    if email['run'] != 0:
        email['run'] += 1
        return email_thread(message)

    elif rihanna_football.football_key['status'] == 1:
        return rihanna_football.football_switch(message)

    elif {"send", "email"} - set(message.split()) == set():
        email['run'] += 1
        return email_thread(message)

    elif (len(message) == 1) or message.isdigit():
        return rihanna_one_char.main(message)

    elif message == 'sing a song':
        return sing_song()

    elif message.split()[0] == 'sensor':
        return ri_sensor.selector(message)

    elif message == 'i love you':
        return add_heart_effect()

    elif message == 'i hate you':
        return remove_heart_effect()

    elif message[:len('word cloud')] == 'word cloud':
        return rihanna_wc.selector(message)

    elif message[:len('nhs')] == 'nhs':
        return rihanna_nhs.selector(message)

    elif message[:len('hot 100')] == 'hot 100':
        return hot100.selector(message)

    elif message[:len('show image')] == 'show image':
        return ri_image.selector(message)

    elif message[:len('windows')] == 'windows':
        return rihanna_windows.selector(message)

    elif ("twitter" in message) or ("tweet" in message):
        return rihanna_tweet.twitter(message)

    elif message in break_words:
        reply = stop_words()
        if config.lang_code != 'en':
            reply = rihanna_dict.translate_sentence_code(reply, config.lang_code)
            config.lang_code = 'en'
        return reply

    elif message == 'why':
        reply = "Sorry, I cant tell you. Its a secret"
        if config.lang_code != 'en':
            reply = rihanna_dict.translate_sentence_code(reply, config.lang_code)
            config.lang_code = 'en'
        return reply

    elif message == 'what is your name':
        reply = "My name is Rihanna"
        if config.lang_code != 'en':
            reply = rihanna_dict.translate_sentence_code(reply, config.lang_code)
            config.lang_code = 'en'
            return reply
        return {'display': reply, 'say': reply}

    elif message in _date:
        reply = rihanna_time.rihanna_date()
        return reply

    elif message in _time:
        reply = rihanna_time.rihanna_time()
        return reply

    elif message[0:7] == 'what is':
        try:
            reply = wikipedia.summary(message.strip()[7:], sentences=1)
            if config.lang_code != 'en':
                reply = rihanna_dict.translate_sentence_code(reply, config.lang_code)
                config.lang_code = 'en'
                return reply
            return {'display': reply, 'say': reply}

        except:
            reply = "{}? hmm.. I know what it is but I can not tell you".format(message.strip()[7:])
            if config.lang_code != 'en':
                reply = rihanna_dict.translate_sentence_code(reply, config.lang_code)
                config.lang_code = 'en'
                return reply
            return {'display': reply, 'say': reply}

    elif {"bbc", "news"} - set(message.split()) == set():
        reply = rihanna_news.bbc()
        return reply

    elif message[:len('news')] == 'news':
        return ri_news.selector(message)

    elif message == 'weather forecast today':
        reply = weather('london,uk')
        # # rihanna_voice(reply)
        # if config.lang_code != 'en':
        #     reply = rihanna_dict.translate_sentence_code(reply, config.lang_code)
        #     config.lang_code = 'en'
        #     return reply
        return reply

    elif "facebook" in message:
        return rihanna_facebook.fb(message)

    elif message[:8] == 'football':
        reply = rihanna_football.football(message)
        return reply

    elif message[:9] == 'calculate':
        data = message.strip()[10:]
        reply = calc.calculate(data)
        return reply

    elif message[:6] == 'google':
        search = message.strip()[7:]
        display = google_search(search)
        reply = "Googling . . ."
        if config.lang_code != 'en':
            reply = rihanna_dict.translate_sentence_code(reply, config.lang_code)
            config.lang_code = 'en'
        return reply

    elif message[:3] == 'tfl':
        return rihanna_tfl.tfl(message)

    elif message[0:16] == 'weather forecast':
        reply = weather(message.strip()[16:].strip())
        # if config.lang_code != 'en':
        #     reply = rihanna_dict.translate_sentence_code(reply, config.lang_code)
        #     config.lang_code = 'en'
        #     return reply
        return reply

    elif message[0:4] == 'play':
        critic = ['that is a lovely song', 'that is a terrible song', "don't like that song",
                  "that's my jam", "someone turn the music up", "you have a terrible song taste"]
        # rihanna_voice('Searching for {}'.format(message.strip()[5:]))
        play_song(message.strip()[5:])
        reply = critic[r.randrange(len(critic))]
        if config.lang_code != 'en':
            reply = rihanna_dict.translate_sentence_code(reply, config.lang_code)
            config.lang_code = 'en'
            return reply
        return {'display': reply, 'say': reply}

    elif len([i for i in calc.opp_code if i in message]) > 0:
        reply = calc.calculate(message)
        return reply

    elif message == 'lock screen':
        import ctypes
        ctypes.windll.user32.LockWorkStation()

    elif message != 'Bye':
        reply = str(bot.get_response(message)).replace('-', '')

        if config.lang_code != 'en':
            reply = rihanna_dict.translate_sentence_code(reply, config.lang_code)
            config.lang_code = 'en'
            return reply
        return {'display': reply, 'say': reply}


def get_response(usrText):
    bot = ChatBot('Bot',
                  storage_adapter='chatterbot.storage.SQLStorageAdapter',
                  logic_adapters=[
                      {
                          'import_path': 'chatterbot.logic.BestMatch'
                      },
                      {
                          'import_path': 'chatterbot.logic.LowConfidenceAdapter',
                          'threshold': 0.50,
                          'default_response': 'I am sorry. I am not allowed to give an answer to that question.'
                      }
                  ],
                  trainer='chatterbot.trainers.ListTrainer')
    bot.set_trainer(ListTrainer)
    while True:
        if usrText.strip() == 'click':
            text = rihanna_speak.speech_recog()
            print(f'speech: {text.strip()}')
            if text == 'sorry, could not recognize your voice':
                reply = str(text)
                return str({'display': reply, 'say': reply})
            else:
                result = rihanna(text.strip())
                if type(result).__name__ == 'dict':
                    result['user_said'] = text.strip()
                    reply = str(result)
                    return reply
                else:
                    print('Not dict', result)
                    result = f"{text};{result}"
                    reply = str(result)
                    return reply
                # return str({'user_said': text, 'reply': result, 'voice_check': 0, 'say': ''})

        elif usrText.strip() != 'Bye':
            result = rihanna(usrText)
            reply = str(result)
            # print(result)
            return reply  # reply should be string else it wont work
        elif usrText.strip() == 'Bye':
            reply = 'Bye'
            return str({'display': reply, 'say': reply})


if __name__ == '__main__':
    # Testing some functionalities
    print(google_search("when is the wilder fight date"))
    print(rihanna('journey duration from se18 3px to se1 5hp'))
    print(rihanna("amazon least price for external hard drive 2tb"))
    print(get_response("jugar drake va mal"))