chabot_final-code_only.py

import numpy as np #for data
import speech_recognition as sr #speech-text
from gtts import gTTS #text-speech
import os #for data -os
import datetime #for data
import transformers #language model

import time

class ChatBot(): #a chatbot class
    def __init__(self, name):
        print("----- starting up", name, "-----")
        self.name = name

    def speech_to_text(self): #speech to text method function
        recognizer = sr.Recognizer() #a recogniser instance
        with sr.Microphone() as mic:
            print("listening to...")
            audio = recognizer.listen(mic)
        try:
            self.text = recognizer.recognize_google(audio)
            print("me -->", self.text)
        except:
            print("me --> ERROR")
        finally:
            print("me --> ERROR")

    @staticmethod
    def text_to_speech(text):
        print("Ibot --> ", text)
        speaker = gTTS(text=text, lang='en', slow=False)
        speaker.save("res.mp3")
        statbuf = os.stat("res.mp3")
        mbytes = statbuf.st_size / 1024
        duration = mbytes / 200
        os.system("start res.mp3") #for macbook->afplay or for windows use->start
        # os.system("close res.mp3")
        time.sleep(int(50 * duration)) #import time
        os.remove("res.mp3")

    def wake_up(self, text):
        return True if self.name in text.lower() else False

#import datetime
    @staticmethod
    def action_time(self):
        return datetime.datetime.now().time().strftime('%H:%M')
    #and run the script after reading the above function to the AI class

# Run the AI
#if __name__ == "__main__":
#    ai = ChatBot(name="Ibot")
#    while True:
#        ai.speech_to_text()
#    ## waking up
#        if ai.wake_up(ai.text) is True:
#            res = "Hello I am Ibot the AI, what may I do for you?"
#         ## do any action
#        elif "time" in ai.text:
#            res = ai.action_time()
#         ## respond politely
#        elif any(i in ai.text for i in ["thank","thanks"]):
#            res = np.random.choice(
#                  ["you're welcome!","anytime!",
#                   "no problem!","cool!",
#                   "I'm here if you need me!","peace out!"])
#        ai.text_to_speech(res)
#

# Final -Running the AI
if __name__ == "__main__":

    ai = ChatBot(name="Ibot")
    nlp = transformers.pipeline("conversational", model="microsoft/DialoGPT-medium")
    os.environ["TOKENIZERS_PARALLELISM"] = "true"

    ex = True
    while ex:
        ai.speech_to_text()

        ## wake up
        if ai.wake_up(ai.text) is True:
            res = "Hello I am Ibot the AI, what may I do for you?"

        ## action time
        elif "time" in ai.text:
            res = ai.action_time()

        ## respond politely
        elif any(i in ai.text for i in ["thank", "thanks"]):
            res = np.random.choice(
                ["My pleasure", "you're welcome!", "anytime!", "no problem!", "cool!", "I'm here if you need me!", "don't mention"])

        elif any(i in ai.text for i in ["exit", "close"]):
            res = np.random.choice(["Have a good day", "Bye", "Goodbye", "Cheers!"])

            ex = False
        ## conversation
        else:
            if ai.text == "ERROR":
                res = "Sorry, come again?"
            else:
                chat = nlp(transformers.Conversation(ai.text), pad_token_id=50256)
                res = str(chat)
                res = res[res.find("bot >> ") + 6:].strip()

        ai.text_to_speech(res)
    print("----- Shutting down Ibot -----")