import numpy as np

import time
import os
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

checkpoint = "microsoft/DialoGPT-medium"

tokenizer = AutoTokenizer.from_pretrained(checkpoint)

model = AutoModelForCausalLM.from_pretrained(checkpoint)

{"model_id":"1a57186592b5425f8ac1b71a1816d40c","version_major":2,"version_min
or":0}

{"model_id":"f3a07af02cc543b8a202d8a148e9e326","version_major":2,"version_min
or":0}

{"model_id":"324f2626dc0e44988d1bbce09b8d78a9","version_major":2,"version_min
or":0}

{"model_id":"b517ddd83b2c4e2092def32ec5f05594","version_major":2,"version_min
or":0}

{"model_id":"0e4cdb3b7d784b2293d5399227a24338","version_major":2,"version_min
or":0}

class ChatBot():

def __init__(self):

self.chat_history_ids = None

self.bot_input_ids = None

self.end_chat = False

self.welcome()

def welcome(self):
print("Initializing ChatBot ...")

time.sleep(2)
print('Type "bye" or "quit" or "exit" to end chat \n')

time.sleep(3)

greeting = np.random.choice([
"Welcome, I am ChatBot, here for your kind service",
"Hey, Great day! I am your virtual assistant",
 "Hello, it's my pleasure meeting you",
"Hi, I am a ChatBot. Let's chat!"
])
print("ChatBot >> " + greeting)

def user_input(self):

text = input("User >> ")

if text.lower().strip() in ['bye', 'quit', 'exit']:

self.end_chat=True

print('ChatBot >> See you soon! Bye!')

time.sleep(1)
print('\nQuitting ChatBot ...')
else:

self.new_user_input_ids = tokenizer.encode(text +
tokenizer.eos_token, \
return_tensors='pt')

def bot_response(self):

if self.chat_history_ids is not None:

self.bot_input_ids = torch.cat([self.chat_history_ids,
self.new_user_input_ids], dim=-1)
else:

self.bot_input_ids = self.new_user_input_ids

self.chat_history_ids = model.generate(self.bot_input_ids,
max_length=1000, \

pad_token_id=tokenizer.eos_token_id)

response = tokenizer.decode(self.chat_history_ids[:,
self.bot_input_ids.shape[-1]:][0], \
skip_special_tokens=True)

if response == "":
response = self.random_response()

print('ChatBot >> '+ response)

def random_response(self):
i = -1
 response = tokenizer.decode(self.chat_history_ids[:,
self.bot_input_ids.shape[i]:][0], \
skip_special_tokens=True)

while response == '':

i = i-1
response = tokenizer.decode(self.chat_history_ids[:,
self.bot_input_ids.shape[i]:][0], \
skip_special_tokens=True)

if response.strip() == '?':
reply = np.random.choice(["I don't know",
"I am not sure"])

else:
reply = np.random.choice(["Great",
"Fine. What's up?",
"Okay"
])
return reply

bot = ChatBot()

while True:

bot.user_input()

if bot.end_chat:
break

bot.bot_response()

Initializing ChatBot ...

Type "bye" or "quit" or "exit" to end chat

ChatBot >> Hey, Great day! I am your virtual assistant

User >> How are you?

ChatBot >> I'm good, how are you?

User >> I am fine. My name is Hanzla.

ChatBot >> Nice to meet you Hanzla.

User >> Nice to meet you too. I study in Delhi India

ChatBot >> Nice to meet you too.

User >> what is your favourite book?

ChatBot >> I don't read books.

User >> It was nice to meet you. Bye!

ChatBot >> Bye!

User >> exit

ChatBot >> See you soon! Bye!

Quitting ChatBot ...