DialoGPT-medium / Enhanced Roleplay AI with Memory and Interactivity.msgpack
Karneus's picture
Redid Enhanced Roleplay AI with Memory and Interactivity
b229ee8 verified
raw
history blame
1.03 kB
from transformers import pipeline, set_seed
# Initialize the conversation pipeline
set_seed(42)
roleplay_bot = pipeline('conversational', model='microsoft/DialoGPT-medium')
# Memory to store past interactions
memory = []
def update_memory(user_input, bot_response):
memory.append({"user": user_input, "bot": bot_response})
def get_memory_context():
context = ""
for interaction in memory[-5:]: # limiting memory to last 5 interactions for simplicity
context += f"User: {interaction['user']}\nBot: {interaction['bot']}\n"
return context
def interact(user_input):
context = get_memory_context()
input_with_context = context + f"User: {user_input}\n"
bot_response = roleplay_bot(input_with_context)[0]['generated_text'].split('\n')[-1]
update_memory(user_input, bot_response)
return bot_response
# Example interaction
user_input = "Hi! How are you?"
print("User:", user_input)
bot_response = interact(user_input)
print("Bot:", bot_response)
# Continue with more interactions as needed