| 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 |