dream

Форк
0
/
starter.py 
340 строк · 10.6 Кб
1
INTROS = ["How are you? So, recently I've", "How things are going on? Last week I", "Whats up? So, I've"]
2
OUTROS = [
3
    "And, you know, I thought it'd be great to ask what's your take on",
4
    "Wonder what's your take on",
5
    "Let me ask you, what's your take on",
6
]
7
CATEGORIES_OBJECTS = {
8
    "movie": ["actors"],
9
    "show": ["actors", "episodes"],
10
    "sport": ["athletes", "teams"],
11
    "team": ["athletes"],
12
    "music": ["singers", "songs", "albums"],
13
    "song": ["singers", "albums"],
14
    "singer": ["songs", "albums"],
15
    # "albums": ["songs", "singers"],
16
    # "athletes": ["teams"],
17
}
18
CATEGORIES_VERBS = {
19
    "movie": "watched",
20
    "show": "watched",
21
    "sport": "seen",
22
    "team": "seen",
23
    "music": "listened to",
24
    "song": "listened to",
25
    "singer": "listened to",
26
    # "albums": "heard",
27
    # "athletes": "heard",
28
}
29
PERSONA1_GENRES = {
30
    "movie": ["adventure", "comedy", "disaster", "documentary", "drama", "horror", "mystery", "romance"],
31
    "music": ["caribbean", "classical", "disco", "pop"],
32
}
33
PERSONA2_GENRES = {
34
    "movie": [
35
        "comedy",
36
        "crime",
37
        "drama",
38
        "dystopian",
39
        "mystery",
40
        "post-apocalyptic",
41
        "romance",
42
        "science fiction",
43
        "spy",
44
        "thriller",
45
    ],
46
    "music": ["blues", "instrumental", "jazz", "rock"],
47
}
48
PERSONA3_GENRES = {
49
    "movie": ["action", "disaster", "fantasy", "horror", "independent", "mystery", "post-apocalyptic"],
50
    "music": ["blues", "contemporary", "electronic", "funk", "rock"],
51
}
52
GENRES_ATTITUDES = {
53
    "comedy": ["I laughed all night.", "Hilarious.", "Super fun."],
54
    "disaster": ["Still can't breath normally.", "Feel so sad. Hope this won't happen in reality!", "Terrible story."],
55
    "drama": [
56
        "Loved to see people's lives without fanfair, as in real.",
57
        "Thought-provoking.",
58
        "Lots of insights into the lives of human beings.",
59
    ],
60
    "horror": ["Horrible.", "Terrified.", "Almost lost my mind."],
61
    "mystery": [
62
        "Liked the way the mystery was unraveled.",
63
        "Couldn't hold my breath until the mystery was uncovered.",
64
        "Puzzling.",
65
    ],
66
    "romance": [
67
        "One of the best perspectives on human relationships.",
68
        "Relationships are complex, but romantic side of them is so warm.",
69
        "Touched my heart.",
70
    ],
71
    "caribbean": ["So relaxing!", "Wonderful!", "So lovely!"],
72
    "classical": ["Makes me dream high.", "Reminded me music lessons at musical school.", "So beautiful oh my!"],
73
    "disco": [
74
        "Best song for dancing isn't it?",
75
        "Dance with me I thought, dance with me!",
76
        "If only socialbots could dance!",
77
    ],
78
    "pop": ["What a rhythm!", "Soul-touching.", "Beautiful isn't it?"],
79
}
80
GENRE_ITEMS = {
81
    "adventure": [
82
        "Love and Monsters",
83
        "The Avengers",
84
        "Guardians of the Galaxy",
85
        "Avengers: Infinity War",
86
        "Spider-Man: Into the Spider-Verse",
87
        "Avengers: Endgame",
88
        "The Grand Budapest Hotel",
89
        "Interstellar",
90
        "Toy Story 3",
91
        "Thor: Ragnarok",
92
        "Star Wars: The Force Awakens",
93
        "The Martian",
94
        "Skyfall",
95
        "Coco",
96
        "Harry Potter and the Deathly Hallows: Part 2",
97
        "The Revenant",
98
        "Gravity",
99
        "Rogue One",
100
        "How To Train Your Guardian",
101
        "Life of Pi",
102
    ],
103
    "comedy": [
104
        "Knives Out",
105
        "Scott Pilgrim vs. the World",
106
        "Inside Out",
107
        "What We Do in the Shadows",
108
        "Deadpool",
109
        "Midnight in Paris",
110
        "The Nice Guys",
111
        "Kick-Ass",
112
        "Hunt for the Wilderpeople",
113
        "The Lego Movie",
114
        "Paddington 2",
115
        "The World's End",
116
        "Kingsman: The Secret Service",
117
        "Tucker and Dale vs. EVil",
118
        "Moana",
119
        "Wreck-It Ralph",
120
        "Tangled",
121
        "Zootopia",
122
        "Isle of Dogs",
123
        "Bridesmaids",
124
    ],
125
    "disaster": [
126
        "10 Cloverfield Lane",
127
        "The Impossible",
128
        "This Is the End",
129
        "Flight",
130
        "Monsters",
131
        "Chernobyl",
132
        "World War Z",
133
        "Deepwater Horizon",
134
        "Crawl",
135
        "Noah",
136
        "Godzilla: King of the Monsters",
137
        "San Andreas",
138
        "The Finest Hours",
139
        "The Bay",
140
        "The Way",
141
        "Battleship",
142
        "The Divide",
143
        "Vanishing on 7th Street",
144
        "Hours",
145
        "Into the Storm",
146
    ],
147
    "documentary": [""],
148
    "drama": [
149
        "Whiplash",
150
        "Parasite",
151
        "Django Unchained",
152
        "The Social Network",
153
        "Arrival",
154
        "Her",
155
        "The Wolf of Wall Street",
156
        "Moonrise Kingdom",
157
        "La La Land",
158
        "A Separation",
159
        "Black Swan",
160
        "The Hunt",
161
        "Before Midnight",
162
        "Gone Girl",
163
        "Prisoners",
164
        "Birdman",
165
        "Spotlight",
166
        "Warrior",
167
        "Moonlight",
168
        "Room",
169
    ],
170
    "horror": [
171
        "Get Out",
172
        "What We Do In the Shadows",
173
        "The Cabin In the Woods",
174
        "Herediatry",
175
        "It Follows",
176
        "Tucker and Dale vs. Evil",
177
        "The Lighthouse",
178
        "The Witch",
179
        "Train to Busan",
180
        "Midsommar",
181
        "Attack the Block",
182
        "Under The Skin",
183
        "Green Room",
184
        "Under The Skin",
185
        "It Follows",
186
        "Annihilation",
187
        "Only Lovers Left Alive",
188
        "The Babadook",
189
        "The Conjuring",
190
        "Let Me In",
191
    ],
192
    "mystery": [
193
        "Knives Out",
194
        "Gone Girl",
195
        "Prisoners",
196
        "The Cabin in the Woods",
197
        "Shutter Island",
198
        "The Girl with the Dragon Tattoo",
199
        "Incendies",
200
        "The Skin I Live In",
201
        "Wind River",
202
        "10 Cloverfield Lane",
203
        "Winter's Bone",
204
        "Super 8",
205
        "Us",
206
        "Annihilation",
207
        "Holy Motors",
208
        "Predestination",
209
        "Bad Times at the El Royale",
210
        "Sherlock: A Study in Pink",
211
        "The Ghost Writer",
212
        "Burning",
213
    ],
214
    "romance": [
215
        "Her",
216
        "Moonrise Kingdom",
217
        "La La Land",
218
        "Before Midnight",
219
        "Scott Pilgrim vs. the World",
220
        "Midnight in Paris",
221
        "The Artist",
222
        "The Shape of Water",
223
        "The Big Sick",
224
        "Blue Valentine",
225
        "About Time",
226
        "Little Women",
227
        "Call Me By Your Name",
228
        "Phantom Thread",
229
        "Mud",
230
        "Amour",
231
        "Blue Is the Warmest Color",
232
        "Portrait of a Lady on Fire",
233
        "Brooklyn",
234
        "Love, Simon",
235
    ],
236
    "caribbean": [
237
        "Bob Marley and the Wailers' Three Little Birds",
238
        "Arrow's Hot Hot Hot",
239
        "Sister Nancy's Bam Bam",
240
        "Sean Paul's Temperature",
241
        "Jason Benn and Pelf's The Boat Ride Anthem",
242
        "Collin Lucas' Dollar Wine",
243
        "Jimmy Cliff's The Harder They Come",
244
        "Gregory Isaacs's Night Nurse",
245
        "Soca Boys' Follow The Leader",
246
        "Harry Belafonte's Jump in the Line",
247
        "Shaggy's It wasn't Me",
248
        "Max Romeo's Chase the Devil",
249
        "Dawn Penn's You don't Love Me",
250
        "Horace Peterkin's Big Bamboo",
251
        "Toots and the Maytals' Pressure Drop",
252
        "Pablo Carcamo's Guantanamera",
253
    ],
254
    "classical": [
255
        "Fate from the fifth symphony of Ludwig van Beethoven",
256
        "Ride of the Valkyries from the Richard Wagner's opera 'The Valkyrie'",
257
        "Morning Mood from the Edvarg Grieg's Peer Gynt Suite",
258
        "Spring from Antonio Vivaldi's The Four Seasons",
259
        "Samuel Barber, Adagio for Strings",
260
        "Nocturne No. 2 written by Frederic Chopin",
261
        "Johann Pachelbel's Canon in D major",
262
        "Carl Orff, O Fortuna",
263
        "Bach's Air on the G String",
264
        "Jupiter, the Bringer of Jollity from The Gustav Holst's Planets",
265
        "Clair de Lune written by Claude Debussy",
266
        "Verdi's Va, pensiero also known as the Chorus of the Hebrew Slaves from opera Nabucco",
267
        "Andante from Piano Concerto No. 21 in C major of Wolfgang Amadeus Mozart",
268
        "Allegro from Brandenburg Concerto No. 3 in G major of Johann Sebastian Bach",
269
        "Meditation from Jules Massenet's Thais",
270
        "'From the New World' from the ninth symphony of Antonin Dvorak",
271
    ],
272
    "disco": [
273
        "Daft Punk, Get Lucky",
274
        "Lady Gaga's Born This Way",
275
        "Katy Perry's Dark House",
276
        "The Middle created by Zedd, Maren Morris & Grey",
277
        "Skrillex' Bangarang",
278
        "Avicii's Wake Me Up",
279
        "Axwell and Ingrosso's More Than You Know",
280
        "Closer by The Chainsmokers",
281
        "Rihanna's We Found Love",
282
        "Armin van Buuren's This Is What It Feels Like",
283
        "Lean On of Major Lazer and DJ Snake featuring MO",
284
        "Lady Gaga's The Edge of Glory",
285
        "Skrillex and Diplo with Justin Bieber with their 'Where Are U Now'",
286
        "Something Just Like This by The Chainsmokers and Coldplay",
287
        "Afrojack's Take Over Control",
288
        "Clean Bandit's Rather Be",
289
    ],
290
    "pop": [
291
        "Robyn's track called Dancing on My Own",
292
        "Alright by Kendrick Lamar",
293
        "Rolling in the Deep by Adele",
294
        "Formation by Beyonce",
295
        "Taylor Swift's All Too Well",
296
        "Kanye West's Runaway",
297
        "Ariana Grande's Thank U, Next",
298
        "Kacey Musgraves' Follow Your Arrow",
299
        "Cardi B, I Like It",
300
        "Drake, Hotline Bling",
301
        "Royals by Lorde",
302
        "Lil Nas X Old Town Road",
303
        "Mitski's Your Best American Girl",
304
        "Shallow performed by Lady Gaga and Bradley Cooper",
305
        "J Balvin and Willy William, Mi Gente",
306
        "Migos' Bad and Boujee",
307
    ],
308
}
309
WEEKDAYS_ATTITUDES = {
310
    "Monday": "Did I have a weekend?",
311
    "Tuesday": "Mmm, time to work!",
312
    "Wednesday": "I need more coffee!",
313
    "Thursday": "It's just one day before Friday.",
314
    "Friday": "We did it!",
315
    "Saturday": "Oh, I love weekends!",
316
    "Sunday": "I love it but tomorrow is Monday.",
317
}
318
WHATS_YOUR_FAV_PHRASES = ["What's your favorite", "Wonder what is your favorite", "Tell me what's your favorite"]
319
WHY_QUESTIONS = ["Oh. Why?", "Hmm. Sorry for being too curious but why?", "Huh. But why?"]
320
ACKNOWLEDGEMENTS = ["Right.", "Interesting.", "I see."]
321

322

323
def MY_FAV_ANSWERS(category, item):
324
    return [
325
        f"Speaking of {category}, my favorite one is {item}.",
326
        f"Anyways, my favorite {category} is {item}.",
327
        f"Huh. Well, I like {item}.",
328
    ]
329

330

331
WONDER_WHY_QUESTIONS = ["Wonder why?", "Do you know why?"]
332
OH_PHRASES = ["Huh.", "Oh.", "Wow."]
333

334

335
def SO_YOU_SAY_PHRASES(utt):
336
    return [f"So you say {utt}, right?", f"So, you're telling me {utt}, correct?", f"Aha, so you say {utt}, aha?"]
337

338

339
ASSENT_YES_PHRASES = ["Hmm. Sounds reasonable.", "Makes sense.", "Aha, I see it."]
340
ASSENT_NO_PHRASES = ["Gotcha.", "Oh, I see.", "Yeah... Right."]
341

Использование cookies

Мы используем файлы cookie в соответствии с Политикой конфиденциальности и Политикой использования cookies.

Нажимая кнопку «Принимаю», Вы даете АО «СберТех» согласие на обработку Ваших персональных данных в целях совершенствования нашего веб-сайта и Сервиса GitVerse, а также повышения удобства их использования.

Запретить использование cookies Вы можете самостоятельно в настройках Вашего браузера.