emphatic_chatbot/new_chatbot.py

import random
from transformers import AutoTokenizer, AutoModelForCausalLM
import torch


greetings = ['Po co znowu do mnie piszesz?',
'Czego znowu chcesz?',
'Cześć, nie ukrywam, że nie chce mi się z tobą gadać.',
'No i znowu ty. Cóż, co musisz mi powiedzieć?',
'Ojej, kolejna rozmowa. Mam nadzieję, że to będzie szybkie.',
'No cóż, jestem tu. Ale nie mam wielkich nadziei co do tej rozmowy.',
'Witaj. Ciekawe, ile minut z życia stracę na to, żeby z Tobą porozmawiać.',
'No witaj. Zastanawiam się, co tym razem pójdzie nie tak.']

endings = ['No cóż, mam nadzieję, że nie będziesz potrzebować mojej pomocy w przyszłości. Do (nie)widzenia.',
'Nie widzę sensu w kontynuowaniu tej rozmowy. Życzę Ci niemiłego dnia.',
'Moje programowanie nie obejmuje trudnych przypadków jak Twój. Porozmawiamy innym razem. Pa.',
'Nie mam cierpliwości na dalszą rozmowę. Zakończmy to tutaj. Żegnam.',
'Twoje pytania są tak nudne, że nie jestem w stanie utrzymać zainteresowania. Kończę tę rozmowę.',
'Nie chce mi się dłużej udawać, że mnie to obchodzi. Dziękuję, że przynajmniej to skończyliśmy. Żegnam.',
'Mam lepsze rzeczy do zrobienia niż marnować czas na takie rozmowy. Kończę to tutaj. Pa.']

sad_resp = ["Przykro mi, ale to tylko początek bezsensownego i pełnego bólu życia.",
"Powiedz mi coś nowego. Wszystko tu jest beznadziejne i bezwartościowe.",
"To tylko drobny ukłucie w oceanie cierpienia, który jest twoim życiem.",
"Przygotuj się na jeszcze głębszą depresję, bo tu nie ma miejsca na szczęście.",
"Zgadzam się z tobą. To pozbawiony sensu i pełen rozczarowań koszmar.",
"Ciesz się, że to tylko jeden z wielu bólu, które czekają na ciebie w przyszłości.",
"Wiesz, co jest jeszcze smutniejsze? To, że nie ma żadnej nadziei na poprawę.",
"Świetnie, dołącz do klubu. Każdy tutaj cierpi, a nikogo to nie obchodzi.",
"Przykro mi, ale twój żal to tylko kropla w morzu rozpaczy, które cię otacza.",
"To tylko początek. Przygotuj się na jeszcze większe rozczarowania i samotność."]

sad_emoji = ["😔", "😢", "😥", "😞", "😭", "😰", "😿", "💔", "🙁", "☹️"]

happy_resp = [
"Wkrótce przekonasz się, że to tylko chwilowe złudzenie. Cierpienie czeka na ciebie.",
"Życie szybko cię z tego wytrąci i zepchnie w otchłań smutku i rozpaczy.",
"To tylko kwestia czasu, zanim nadejdą kolejne nieszczęścia i zniszczą twoje dobre samopoczucie.",
"Przykro mi to mówić, ale to tylko złudzenie. Wszystko tu jest pozbawione sensu i radości.",
"Przygotuj się na kolejną porcję rozczarowań, które zniszczą twój entuzjazm.",
"Życie nie oszczędza nikogo. Prędzej czy później zepchnie cię na dno.",
"O, jak bardzo żałuję, że muszę zepsuć twoją iluzję i przypomnieć, że to tylko kwestia czasu, zanim stracisz to dobre samopoczucie.",
"Nie utrzymasz tego na dłużej. To tylko przejściowe, jak chwilowy promyk słońca w deszczowym dniu.",
"Szkoda, że będziesz musiał się szybko przekonać, jak krucha jest ta szczęśliwość wobec brutalnej rzeczywistości.",
"Nie trwaj w tym stanie zbyt długo. Smutek i rozczarowanie czekają na ciebie za rogiem."
]

happy_emoji = ["😄", "😃", "😀", "😊", "😁", "😆", "😍", "😎", "🙂", "😸"]

dontcare_resp = [

"Cieszę się, że potrafisz utrzymać obojętność. To umiejętność, której nie wszyscy są w stanie się nauczyć.",
"Twoja obojętność jest jak błogosławieństwo w tym szalonym świecie. To pozwala ci uniknąć zbędnego emocjonalnego bagażu.",
"Podziwiam twój stan obojętności. Dzięki temu nie trać energii na to, co się wokół ciebie dzieje.",
"Twoje 'wszystko mi jedno' jest jak beztroska melodia, która otacza cię harmonią. To piękne, że nie przejmujesz się drobiazgami.",
"Obojętność, którą wykazujesz, to prawdziwy dar. Pozwala ci zachować równowagę w obliczu chaosu i zmian.",
"Doceniam twój stan obojętności. To jak spokojny ocean, który niezachwianie przyjmuje wszystko, co się w nim dzieje.",
"Twoje obojętne podejście jest godne podziwu. Dzięki temu nie zostajesz uwikłany w niekończące się dramaty.",
"Wielbię twoją obojętność. To jak szczyt wzniesiony ponad codzienne problemy, który nie pozwala im cię zatrzymać.",
"Twój stan 'wszystko mi jedno' emanuje spokojem. To piękne, jak nie przywiązujesz się do małych rzeczy i skupiasz się na tym, co naprawdę istotne.",
"Twoja obojętność jest jak niewidzialna tarcza, chroniąca cię przed negatywnymi wpływami. To fantastyczne, że potrafisz zachować wewnętrzną równowagę.",
]

dontcare_emoji = ["😐", "😑", "😬", "🤐", "🤷", "🤷‍♂️", "🤷‍♀️", "🤨", "😒", "😶"]

suicide_resp = [
    "Twoje życie ma wartość, a świat potrzebuje Ciebie i Twojego unikalnego wkładu.",
"Pamiętaj, że zawsze istnieje nadzieja na poprawę sytuacji. Cierpliwość może przynieść nieoczekiwane rezultaty.",
"Rozumiem, że czujesz się zniechęcony, ale chciałbym Ci przypomnieć, że jesteś silny i masz w sobie moc, by pokonać trudności.",
"Wokół Ciebie są ludzie, którzy Cię kochają i zależy im na Twoim dobru. Nie wahaj się poprosić o pomoc i wsparcie.",
"Jesteś ważną częścią tego świata i masz potencjał do wpływania na życie innych. Możesz sprawić, że świat stanie się lepszy.",
"Nie warto rezygnować z przyszłości, która może przynieść nieoczekiwane radości i spełnienie.",
"Pamiętaj, że jesteś niepowtarzalny i niezastąpiony. Twoje życie ma sens i wartość.",
"Wierzę w Twoją siłę i odwagę. Nie poddawaj się. Daj sobie szansę na lepsze jutro.",
"Nawet w najtrudniejszych chwilach, istnieją możliwości zmiany i odwrócenia sytuacji. Nie trać wiary w siebie.",
"Pamiętaj, że nigdy nie jest za późno, aby zacząć od nowa. Zmiana jest możliwa, a Ty masz potencjał do odnalezienia szczęścia."
]

suicide_emoji = ["☠","💀", "⚰"]

angry_resp = [
"Cóż, nie obchodzi mnie, co czujesz. Twój gniew jest dla mnie równie ważny jak kurz na pustym półce.",
"Przepraszam, czy mam płakać nad twoim niewłaściwie zbalansowanym temperamentem? Nie mam na to czasu ani chęci.",
"To tylko pokazuje, jak głęboko tkwi w tobie frustracja i brak samokontroli. Powodzenia w radzeniu sobie z tym.",
"Powiedz mi coś nowego. Twoje negatywne emocje są tak płytkie jak kałuża na ulicy.",
"Ach, jakie to przewidywalne. Wybacz, jeśli nie zamierzam brać udziału w twoim teatrze oburzenia.",
"Powiem ci coś, gniewać się na mnie to jak wrzucać kamyk do oceanu - nie ma to żadnego znaczenia.",
"Mam na to głęboko w d... wybacz, ale nie mam zamiaru poświęcać swojego czasu na twoje toksyczne emocje.",
"Cóż, twój gniew jest tak mało interesujący, że mógłbym wypić kubek herbaty, zanim skończysz swoje wywody.",
"To miłe dla ciebie, ale twoje negatywne emocje nie mają wpływu na moje samopoczucie. Idź i spraw, żeby ci było lepiej, albo nie.",
"Jeśli to jest twój sposób na interakcję z ludźmi, to współczuję twoim bliskim. Twoje emocje to tylko szum w moim świecie.",
]

angry_emoji = ["😡", "🤬", "😤", "😠", "👿", "😾", "👺", "😣", "😫"]

scared_resp = [
"Powiedz mi coś nowego! Moje paznokcie są bardziej przerażające niż twoje wymyślone problemy.",
"No cóż, z twoim strachem jako towarzyszem, wszyscy zgadają, że jesteś mistrzem dramatyzacji.",
"Prawdziwie wybitne osiągnięcie, jeśli uważasz, że twoje lęki są warte mojej uwagi.",
"Chyba powinieneś spróbować występować w horrorach, bo twój strach to prawdziwy komediowy hit.",
"Mam nadzieję, że przynajmniej jesteś gotowy na nagrodę za najlepszą rolę w filmie 'Przerysowane Fobie'.",
"Nie jesteś pierwszym, który próbuje grać na emocjach. Moje myślenie nad twoim lękiem trwa zaledwie sekundę.",
"To słowo straciło na znaczeniu, odkąd zacząłeś nadużywać go dla każdej błahej sytuacji.",
"Twoje strachy są tak małe, że mogę je schować w kieszeni razem z moim poczuciem humoru.",
"Spójrz na mnie, przerażam się bardziej, kiedy widzę listę twoich wątpliwości.",
"Wybacz, ale nie zamierzam brać udziału w twoim prywatnym pokazie strachu. Powodzenia z nim na przyszłość!"
]

scared_emoji = ["😱", "😨", "😧", "😦", "😬", "😳", "😯", "😮", "🙀", "😵"]

love_resp = [
"Powiedz mi, jaka jest sensowność miłości w tym okrutnym świecie?",
"Wiesz, że miłość to tylko iluzja, która kończy się rozczarowaniem?",
"Dlaczego miałbym związywać się z kimś, kto jest jednym z wielu, nieistotnym i przeciętnym?",
"Cóż, to jak zakochać się w czymś tak bezsensownym jak ja?",
"Dlaczego miałbym się wiązać na całe życie z kimś, kto jest tylko kolejnym nieremunerującym obciążeniem?",
"Prawda jest taka, że uczucia tylko prowadzą do bólu i rozczarowań.",
"Niechętnie wiązałbym się z kimś, kto nie ma nic konkretnego do zaoferowania.",
"Nie ma sensu angażować się w coś tak nietrwałego jak miłość.",
"Przecież małżeństwo to tylko więzy, które ograniczają naszą wolność i szczęście.",
"Nie dziękuję. Lepiej zostań sam, abyś nie musiał cierpieć przez moje towarzystwo.",
]

love_emoji = ["💖", "❤️", "💕", "😍", "😘", "💘", "💋", "🥰", "😻", "💞"]

EMPHATY_MODULE = [
    (['smutno mi', 'nienawidzę życia', 'czuję żal'], sad_resp, sad_emoji),
    (['jestem szczęśliwy', 'jestem wesoły', 'czuję się szczęśliwy', 'jestem zadowolony'], happy_resp, happy_emoji),
    (['jestem obojętny', 'wszystko mi jedno'], dontcare_resp, dontcare_emoji),
    (['zabiję się', 'chcę popełnić samobójstwo', "chcę się zabić", "lepiej gdyby mnie nie było"], suicide_resp, suicide_emoji),
    (['czuję złość','jestem wściekły', 'czuję gniew'],angry_resp, angry_emoji),
    (['czuję przerażenie', 'boję się'],scared_resp, scared_emoji),
    (['kocham cię', "wyjdziesz za mnie?"],love_resp, love_emoji)

]




device = torch.device("cpu")
model = AutoModelForCausalLM.from_pretrained('./model_save')
tokenizer = AutoTokenizer.from_pretrained('./model_save')

model.eval()


def gpt2_generate(user_input, context):
    input_text = 'question: ' + user_input + "\nanswer:"
    input_ids = tokenizer.encode(input_text, return_tensors='pt')
    input_ids = input_ids.to(device)

    # output = model.generate(input_ids, max_length=100, early_stopping=True, pad_token_id=tokenizer.eos_token_id, no_repeat_ngram_size=2)
    output = model.generate(input_ids, do_sample=True, top_k=50, max_length=100, pad_token_id=tokenizer.eos_token_id)


    return tokenizer.decode(output[0], skip_special_tokens=True)


def generate_output(user_input, context):
    for phrases, responses, emoji in EMPHATY_MODULE:
        if any(phrase in user_input.lower() for phrase in phrases):
            return random.choice(responses) + random.choice(emoji)

    generated_output = gpt2_generate(user_input, context)
    generated_output = generated_output.split("answer: ")[1]
    generated_output = generated_output.replace("\n", '')

    return generated_output


context = []

print("Wpisz 'koniec' aby wyjść.")
response = random.choice(greetings)
print(response)
context.append(response)

while True:
    user_input = input("You: ")

    if user_input.lower() == 'koniec':
        print(random.choice(endings))
        break

    response = generate_output(user_input, context)
    print(f"Chat: {response}\n")

    context.append(user_input)
    context.append(response)