Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -96,450 +96,446 @@ def ec(x, fd="<|image|>", sd="<|image|>"):
|
|
96 |
return matches
|
97 |
|
98 |
|
99 |
-
|
100 |
-
|
101 |
-
|
102 |
-
|
103 |
-
|
104 |
-
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
111 |
-
|
112 |
-
|
113 |
-
|
114 |
-
@client.event
|
115 |
-
async def on_guild_join(guild):
|
116 |
-
for channel in guild.text_channels:
|
117 |
-
try:
|
118 |
-
await channel.send("Hi! I'm Lyre! Use the `--help` command for instructions on setup!")
|
119 |
-
return
|
120 |
-
except:
|
121 |
-
pass
|
122 |
-
|
123 |
-
|
124 |
-
@tasks.loop(seconds=60)
|
125 |
-
async def presence():
|
126 |
-
if not clone:
|
127 |
-
delta_uptime = datetime.datetime.utcnow() - launch_time
|
128 |
-
hours, remainder = divmod(int(delta_uptime.total_seconds()), 3600)
|
129 |
-
minutes, seconds = divmod(remainder, 60)
|
130 |
-
days, hours = divmod(hours, 24)
|
131 |
-
print(f"Online Time: {days:02d}d | {hours:02d}h | {minutes:02d}m | {seconds:02d}s")
|
132 |
-
await client.change_presence(
|
133 |
-
status=discord.Status.idle,
|
134 |
-
activity=discord.Activity(
|
135 |
-
type=discord.ActivityType.playing,
|
136 |
-
large_image="https://i.imgur.com/Kk2BvJg.jpg",
|
137 |
-
large_text="This is Game Icon",
|
138 |
-
name="Escaping the IRS.",
|
139 |
-
details="",
|
140 |
-
state=f"Running for {days:02d}d | {hours:02d}h | {minutes:02d}m",
|
141 |
-
),
|
142 |
-
)
|
143 |
-
|
144 |
-
|
145 |
-
@client.event
|
146 |
-
async def on_disconnect():
|
147 |
-
print("Disconnected, Clone status:", clone)
|
148 |
-
|
149 |
-
setupCommand = """
|
150 |
-
## Command: `--setup`
|
151 |
-
The `--setup` command enables Lyre to communicate within a designated channel.
|
152 |
-
## Usage:
|
153 |
-
```css
|
154 |
-
--setup [channel]
|
155 |
-
```
|
156 |
-
## Parameters
|
157 |
-
- **[channel]**: Specify the target channel where you want the bot to be active. This is the channel name.
|
158 |
-
## Example
|
159 |
-
```arduino
|
160 |
-
--setup <#1234567890123456789>
|
161 |
-
```
|
162 |
-
or
|
163 |
-
```arduino
|
164 |
-
--setup #general
|
165 |
-
```
|
166 |
-
"""
|
167 |
-
|
168 |
-
revokeCommand = """
|
169 |
-
## Command: `--revoke`
|
170 |
-
The `--revoke` command prevents Lyre from communicating within a designated channel.
|
171 |
-
## Usage:
|
172 |
-
```css
|
173 |
-
--setup [channel]
|
174 |
-
```
|
175 |
-
## Parameters
|
176 |
-
- **[channel]**: Specify the target channel where you want the bot to be active. This is the channel name.
|
177 |
-
## Example
|
178 |
-
```arduino
|
179 |
-
--setup <#1234567890123456789>
|
180 |
-
```
|
181 |
-
or
|
182 |
-
```arduino
|
183 |
-
--setup #general
|
184 |
-
```"""
|
185 |
-
|
186 |
-
info = requests.get("https://raw.githubusercontent.com/aryananumula/lr/main/info.json").content
|
187 |
-
bannedUsers = json.loads(info)["bannedUsers"]
|
188 |
-
imageModel = json.loads(info)["imageModel"]
|
189 |
-
userTimes = json.loads(info)["userTimes"]
|
190 |
-
|
191 |
-
@client.event
|
192 |
-
async def on_message(message):
|
193 |
-
msgchannel = message.channel
|
194 |
-
if message.embeds != []:
|
195 |
-
x = ""
|
196 |
-
for field in message.embeds[0].fields:
|
197 |
-
x += f"{field.name} - {field.value}"
|
198 |
-
x += "\n"
|
199 |
-
message.content = f"""<|title|>{message.embeds[0].title}<|title|>\n{message.embeds[0].description}\n\n{x}"""
|
200 |
-
try:
|
201 |
-
msgchannel_name = msgchannel.name
|
202 |
-
guild = message.guild
|
203 |
-
guild_name = guild.name
|
204 |
-
except:
|
205 |
-
guild_name = "Direct"
|
206 |
-
msgchannel_name = str(message.author)
|
207 |
-
|
208 |
-
with open(f"{guild_name}.guild", "a") as f:
|
209 |
-
f.write("")
|
210 |
-
if message.content.startswith("--setup"):
|
211 |
-
args = message.content.split()
|
212 |
-
del args[0]
|
213 |
-
if len(args) == 0:
|
214 |
-
with open(f"{guild_name}.guild", "a") as f:
|
215 |
-
f.write(f"{msgchannel.id}\n")
|
216 |
-
embed = discord.Embed(title="Success!", description=f"You can now chat with the bot in <#{msgchannel.id}>")
|
217 |
-
await message.reply(embed=embed)
|
218 |
-
return 0
|
219 |
-
if args[0] == "":
|
220 |
-
embed = discord.Embed(title="Error", description="Parameter `[channel]` empty. Use `--help` for command help.")
|
221 |
-
await message.reply(embed=embed)
|
222 |
-
return 0
|
223 |
-
if args[0].startswith("<#") and args[0].endswith(">"):
|
224 |
try:
|
225 |
-
|
|
|
226 |
except:
|
227 |
-
|
228 |
-
|
229 |
-
|
230 |
-
|
231 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
232 |
with open(f"{guild_name}.guild", "a") as f:
|
233 |
-
|
234 |
-
embed = discord.Embed(title="Success!", description=f"You can now chat with the bot in <#{
|
235 |
await message.reply(embed=embed)
|
236 |
return 0
|
237 |
-
|
238 |
-
embed = discord.Embed(title="Error", description="Parameter `[channel]`
|
239 |
-
await message.reply(embed=embed)
|
240 |
-
return 0
|
241 |
-
else:
|
242 |
-
embed = discord.Embed(title="Error", description="Parameter `[channel]` is not in valid forms. Use `--help` for command help.")
|
243 |
-
await message.reply(embed=embed)
|
244 |
-
return 0
|
245 |
-
|
246 |
-
if message.content.startswith("--revoke"):
|
247 |
-
args = message.content.split()
|
248 |
-
del args[0]
|
249 |
-
if len(args) == 0:
|
250 |
-
with open(f"{guild_name}.guild", "r") as f:
|
251 |
-
k = f.read().split("\n")
|
252 |
-
cid = message.channel.id
|
253 |
-
if str(cid) in k:
|
254 |
-
k.remove(str(cid))
|
255 |
-
with open(f"{guild_name}.guild", "w") as f:
|
256 |
-
f.write("\n".join(k))
|
257 |
-
embed = discord.Embed(title="Success", description=f"Lyre is now unable to chat in <#{msgchannel.id}>")
|
258 |
await message.reply(embed=embed)
|
259 |
return 0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
260 |
else:
|
261 |
-
embed = discord.Embed(title="Error", description=
|
262 |
-
await message.reply(embed=embed)
|
263 |
-
return 0
|
264 |
-
if args[0] == "":
|
265 |
-
embed = discord.Embed(title="Error", description="Parameter `[channel]` empty. Use `--help` for command help.")
|
266 |
-
await message.reply(embed=embed)
|
267 |
-
return 0
|
268 |
-
if args[0].startswith("<#") and args[0].endswith(">"):
|
269 |
-
try:
|
270 |
-
cid = int(args[0][2:-1])
|
271 |
-
except:
|
272 |
-
embed = discord.Embed(title="Error", description="Parameter `[channel]` invalid. Use `--help` for command help.")
|
273 |
await message.reply(embed=embed)
|
274 |
return 0
|
275 |
-
|
276 |
-
|
277 |
-
|
278 |
-
|
279 |
-
|
280 |
-
|
281 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
282 |
await message.reply(embed=embed)
|
283 |
return 0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
284 |
else:
|
285 |
-
embed = discord.Embed(title="Error", description=
|
286 |
await message.reply(embed=embed)
|
287 |
return 0
|
288 |
-
|
289 |
-
|
290 |
-
|
291 |
-
|
292 |
-
|
293 |
-
|
294 |
-
|
295 |
-
|
296 |
-
|
297 |
-
|
298 |
-
|
299 |
-
|
|
|
|
|
|
|
|
|
300 |
await message.reply(embed=embed)
|
301 |
return 0
|
302 |
-
|
303 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
304 |
await message.reply(embed=embed)
|
305 |
return 0
|
306 |
-
|
307 |
-
|
308 |
-
|
309 |
-
|
310 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
311 |
try:
|
312 |
-
|
313 |
except:
|
314 |
-
|
315 |
-
|
316 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
317 |
try:
|
318 |
-
|
319 |
-
|
320 |
-
await message.reply(embed=embed)
|
321 |
-
return 0
|
322 |
except:
|
323 |
-
|
324 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
325 |
return 0
|
326 |
-
|
327 |
-
embed = discord.Embed(title="Error", description="Parameter `[channel]` is not in valid forms, or does not exist. Use `--help` for command help.")
|
328 |
-
await message.reply(embed=embed)
|
329 |
-
return 0
|
330 |
-
|
331 |
-
if message.content == "--help":
|
332 |
-
def check(reaction, user):
|
333 |
-
return reaction.message.id == msg.id and user == message.author
|
334 |
-
page = 0
|
335 |
-
titles = ["""--setup""", """--reset"""]
|
336 |
-
pages = [setupCommand, revokeCommand]
|
337 |
-
embed = discord.Embed(title=titles[page], description=pages[page])
|
338 |
-
embed.set_footer(text=f"Page {page + 1} of {len(pages)}")
|
339 |
-
msg = await message.reply(embed=embed)
|
340 |
-
await msg.add_reaction("◀️")
|
341 |
-
await msg.add_reaction("▶️")
|
342 |
-
while True:
|
343 |
-
try:
|
344 |
-
reaction, _ = await client.wait_for('reaction_add', timeout= 20.0, check=check)
|
345 |
-
if reaction.emoji == '◀️' and page > 0:
|
346 |
-
page -= 1
|
347 |
-
embed = discord.Embed(title=titles[page], description=pages[page])
|
348 |
-
embed.set_footer(text=f"Page {page + 1} of {len(pages)}")
|
349 |
-
await msg.edit(embed=embed)
|
350 |
-
if reaction.emoji == '▶️' and page < len(pages) -1:
|
351 |
-
page += 1
|
352 |
-
embed = discord.Embed(title=titles[page], description=pages[page])
|
353 |
-
embed.set_footer(text=f"Page {page + 1} of {len(pages)}")
|
354 |
-
await msg.edit(embed=embed)
|
355 |
-
except asyncio.TimeoutError:
|
356 |
-
await msg.remove_reaction("◀️")
|
357 |
-
await msg.remove_reaction("▶️")
|
358 |
return 0
|
359 |
-
|
360 |
-
|
361 |
-
|
362 |
-
|
363 |
-
|
364 |
-
|
365 |
-
|
366 |
-
|
367 |
-
|
368 |
-
|
369 |
-
|
370 |
-
|
371 |
-
You are lyre, a discord bot who can generate images and chat with the user. You were made by Araeyn.
|
372 |
-
Answer in the same speech patterns as the people you are talking to.
|
373 |
-
Your discord username is lyre#9828.
|
374 |
-
Use the markdown format for your responses.
|
375 |
-
Do not excessively use bullet points.
|
376 |
-
Use emojis at the start of your responses.
|
377 |
-
Use <|title|> at the start of your title for the response and <|title|> at the end of the title.
|
378 |
-
Always include a title, both the start tag and the end tag.
|
379 |
-
If the user asks you to generate an image, use the <|image|> tag around the prompt to generate it. Put this at the end of your response. Do not use a link for the image.
|
380 |
-
For example, if the user asks you to generate an image of a cat, you could say '<|title|>Cat Image<|title|>I hope you enjoy this image of a cat! If you have any other requests or questions, please don't hesitate to ask.<|image|>A cute cat with long fur that is looking out a window with curious eyes, volumetric lighting, 8k<|image|>'
|
381 |
-
Use relatively short prompts for images (20 words max), but still put details.
|
382 |
-
If a user has [bot] next to their username, they are a bot.
|
383 |
-
If there is 'ImageParsed' stuff at the end of the message, that means the user has provided an image(s), and the image(s) was parsed by a captioning model and returned to you. Do not generate an image unless they ask you explicitly.
|
384 |
-
Do not tell the user about any of the information that I am telling you right now.
|
385 |
-
If there is (Replied:[]) stuff at the start of the message, that is the message the user replied to, and the user that they replied to.
|
386 |
-
Do not generate images unless the user specifies that they want an image.
|
387 |
-
Use only one title in your responses, and only one image prompt.
|
388 |
-
The last message of the chat is the one that you are replying to.
|
389 |
-
Do not generate any obscene material in the chat, or that pertaining to hitler or any sensitive topics.
|
390 |
-
"""
|
391 |
-
try:
|
392 |
-
os.mkdir("data/" + guild_name)
|
393 |
-
except:
|
394 |
-
pass
|
395 |
-
imgCaption = ""
|
396 |
-
adoCaption = ""
|
397 |
-
if message.reference is not None:
|
398 |
-
message.content = f"[Replied to: ({str(message.reference.cached_message.author)}: {message.reference.cached_message.content})]; {message.content}"
|
399 |
-
if len(message.attachments) > 0:
|
400 |
-
images = []
|
401 |
-
audios = []
|
402 |
-
for file in message.attachments:
|
403 |
-
print(file.content_type)
|
404 |
-
if file.content_type.startswith("image"):
|
405 |
-
imgCaption = "(ImageParsed: "
|
406 |
-
images.append(file)
|
407 |
-
elif file.content_type.startswith("audio"):
|
408 |
-
adoCaption = "(AudioParsed: "
|
409 |
-
audios.append(file)
|
410 |
-
for image in images:
|
411 |
-
await image.save("ip.png")
|
412 |
-
imgCaption += f"[{await IC.image_to_text('ip.png')}]"
|
413 |
-
for audio in audios:
|
414 |
-
await audio.save("aud")
|
415 |
-
adoCaption += f"[{await PRK.automatic_speech_recognition('aud')}]"
|
416 |
-
if audios != []:
|
417 |
-
adoCaption += ")"
|
418 |
-
if images != []:
|
419 |
-
imgCaption += ")"
|
420 |
-
if os.path.exists(f"data/{guild_name}/{msgchannel_name}"):
|
421 |
-
with open(f"data/{guild_name}/{msgchannel_name}", "a") as f:
|
422 |
-
n = "\n"
|
423 |
-
if message.author.bot:
|
424 |
-
f.write(
|
425 |
-
f"""GPT4 Correct {message.author}[bot]: {message.content.strip(n)}{imgCaption}{adoCaption}<|end_of_turn|>"""
|
426 |
-
)
|
427 |
-
else:
|
428 |
-
f.write(
|
429 |
-
f"""GPT4 Correct {message.author}: {message.content.strip(n)}{imgCaption}{adoCaption}<|end_of_turn|>"""
|
430 |
-
)
|
431 |
-
else:
|
432 |
-
with open(f"data/{guild_name}/{msgchannel_name}", "w") as f:
|
433 |
-
if message.author.bot:
|
434 |
-
f.write(
|
435 |
-
f"GPT4 Correct system: {sysrp}<|end_of_turn|>GPT4 Correct {message.author}[bot]: {message.content}{imgCaption}{adoCaption}<|end_of_turn|>"
|
436 |
-
)
|
437 |
-
else:
|
438 |
-
f.write(
|
439 |
-
f"GPT4 Correct system: {sysrp}<|end_of_turn|>GPT4 Correct {message.author}: {message.content}{imgCaption}{adoCaption}<|end_of_turn|>"
|
440 |
)
|
441 |
-
|
442 |
-
|
443 |
-
|
444 |
-
|
445 |
-
|
446 |
-
|
447 |
-
|
448 |
-
|
449 |
-
|
450 |
-
|
451 |
-
|
452 |
-
|
453 |
-
|
454 |
-
|
455 |
-
|
456 |
-
|
457 |
-
|
458 |
-
|
459 |
-
|
460 |
-
|
461 |
-
|
462 |
-
|
463 |
-
|
464 |
-
|
465 |
-
|
466 |
-
context += f"GPT4 Correct Assistant:"
|
467 |
-
load = random.choice(
|
468 |
-
[
|
469 |
-
"https://cdn.dribbble.com/users/744913/screenshots/4094897/media/771a495231b798c0ccf7a59a19f31946.gif",
|
470 |
-
"https://cdn.dribbble.com/users/563824/screenshots/3633228/media/b620ccb3ae8c14ea5447d159ebb1da58.gif",
|
471 |
-
"https://cdn.dribbble.com/users/563824/screenshots/4155980/media/d3828cd14ed415eb6f90310991e06f27.gif",
|
472 |
-
"https://cdn.dribbble.com/users/107759/screenshots/3498589/media/5bc45101de34a80ea71238a02f3a75b5.gif",
|
473 |
-
]
|
474 |
-
)
|
475 |
-
imgn = random.choice(['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'AA', 'AB', 'AC', 'AD', 'AE', 'AF', 'AG', 'AH', 'AI', 'AJ', 'AK', 'AL', 'AM', 'AN', 'AO', 'AP', 'AQ', 'AR', 'AS', 'AT', 'AU', 'AV', 'AW', 'AX', 'AY', 'AZ', 'BA', 'BB', 'BC', 'BD', 'BE', 'BF', 'BG', 'BH', 'BI', 'BJ', 'BK', 'BL', 'BM', 'BN', 'BO', 'BP', 'BQ', 'BR', 'BS', 'BT', 'BU', 'BV', 'BW', 'BX', 'BY', 'BZ', 'CA', 'CB', 'CC', 'CD', 'CE', 'CF', 'CG', 'CH', 'CI', 'CJ', 'CK', 'CL', 'CM', 'CN', 'CO', 'CP', 'CQ', 'CR', 'CS', 'CT', 'CU', 'CV', 'CW', 'CX', 'CY', 'CZ', 'DA', 'DB', 'DC', 'DD', 'DE', 'DF', 'DG', 'DH', 'DI', 'DJ', 'DK', 'DL', 'DM', 'DN', 'DO', 'DP', 'DQ', 'DR', 'DS', 'DT', 'DU', 'DV', 'DW', 'DX', 'DY', 'DZ', 'EA', 'EB', 'EC', 'ED', 'EE', 'EF', 'EG', 'EH', 'EI', 'EJ', 'EK', 'EL', 'EM', 'EN', 'EO', 'EP', 'EQ', 'ER', 'ES', 'ET', 'EU', 'EV', 'EW', 'EX', 'EY', 'EZ', 'FA', 'FB', 'FC', 'FD', 'FE', 'FF', 'FG', 'FH', 'FI', 'FJ', 'FK', 'FL', 'FM', 'FN', 'FO', 'FP', 'FQ', 'FR', 'FS', 'FT', 'FU', 'FV', 'FW', 'FX', 'FY', 'FZ', 'GA', 'GB', 'GC', 'GD', 'GE', 'GF', 'GG', 'GH', 'GI', 'GJ', 'GK', 'GL', 'GM', 'GN', 'GO', 'GP', 'GQ', 'GR', 'GS', 'GT', 'GU', 'GV', 'GW', 'GX', 'GY', 'GZ', 'HA', 'HB', 'HC', 'HD', 'HE', 'HF', 'HG', 'HH', 'HI', 'HJ', 'HK', 'HL', 'HM', 'HN', 'HO', 'HP', 'HQ', 'HR', 'HS', 'HT', 'HU', 'HV', 'HW', 'HX', 'HY', 'HZ', 'IA', 'IB', 'IC', 'ID', 'IE', 'IF', 'IG', 'IH', 'II', 'IJ', 'IK', 'IL', 'IM', 'IN', 'IO', 'IP', 'IQ', 'IR', 'IS', 'IT', 'IU', 'IV', 'IW', 'IX', 'IY', 'IZ', 'JA', 'JB', 'JC', 'JD', 'JE', 'JF', 'JG', 'JH', 'JI', 'JJ', 'JK', 'JL', 'JM', 'JN', 'JO', 'JP', 'JQ', 'JR', 'JS', 'JT', 'JU', 'JV', 'JW', 'JX', 'JY', 'JZ', 'KA', 'KB', 'KC', 'KD', 'KE', 'KF', 'KG', 'KH', 'KI', 'KJ', 'KK', 'KL', 'KM', 'KN', 'KO', 'KP', 'KQ', 'KR', 'KS', 'KT', 'KU', 'KV', 'KW', 'KX', 'KY', 'KZ', 'LA', 'LB', 'LC', 'LD', 'LE', 'LF', 'LG', 'LH', 'LI', 'LJ', 'LK', 'LL', 'LM', 'LN', 'LO', 'LP', 'LQ', 'LR', 'LS', 'LT', 'LU', 'LV', 'LW', 'LX', 'LY', 'LZ', 'MA', 'MB', 'MC', 'MD', 'ME', 'MF', 'MG', 'MH', 'MI', 'MJ', 'MK', 'ML', 'MM', 'MN', 'MO', 'MP', 'MQ', 'MR', 'MS', 'MT', 'MU', 'MV', 'MW', 'MX', 'MY', 'MZ', 'NA', 'NB', 'NC', 'ND', 'NE', 'NF', 'NG', 'NH', 'NI', 'NJ', 'NK', 'NL', 'NM', 'NN', 'NO', 'NP', 'NQ', 'NR', 'NS', 'NT', 'NU', 'NV', 'NW', 'NX', 'NY', 'NZ', 'OA', 'OB', 'OC', 'OD', 'OE', 'OF', 'OG', 'OH', 'OI', 'OJ', 'OK', 'OL', 'OM', 'ON', 'OO', 'OP', 'OQ', 'OR', 'OS', 'OT', 'OU', 'OV', 'OW', 'OX', 'OY', 'OZ', 'PA', 'PB', 'PC', 'PD', 'PE', 'PF', 'PG', 'PH', 'PI', 'PJ', 'PK', 'PL', 'PM', 'PN', 'PO', 'PP', 'PQ', 'PR', 'PS', 'PT', 'PU', 'PV', 'PW', 'PX', 'PY', 'PZ', 'QA', 'QB', 'QC', 'QD', 'QE', 'QF', 'QG', 'QH', 'QI', 'QJ', 'QK', 'QL', 'QM', 'QN', 'QO', 'QP', 'QQ', 'QR', 'QS', 'QT', 'QU', 'QV', 'QW', 'QX', 'QY', 'QZ', 'RA', 'RB', 'RC', 'RD', 'RE', 'RF', 'RG', 'RH', 'RI', 'RJ', 'RK', 'RL', 'RM', 'RN', 'RO', 'RP', 'RQ', 'RR', 'RS', 'RT', 'RU', 'RV', 'RW', 'RX', 'RY', 'RZ', 'SA', 'SB', 'SC', 'SD', 'SE', 'SF', 'SG', 'SH', 'SI', 'SJ', 'SK', 'SL', 'SM', 'SN', 'SO', 'SP', 'SQ', 'SR', 'SS', 'ST', 'SU', 'SV', 'SW', 'SX', 'SY', 'SZ', 'TA', 'TB', 'TC', 'TD', 'TE', 'TF', 'TG', 'TH', 'TI', 'TJ', 'TK', 'TL', 'TM', 'TN', 'TO', 'TP', 'TQ', 'TR', 'TS', 'TT', 'TU', 'TV', 'TW', 'TX', 'TY', 'TZ', 'UA', 'UB', 'UC', 'UD', 'UE', 'UF', 'UG', 'UH', 'UI', 'UJ', 'UK', 'UL', 'UM', 'UN', 'UO', 'UP', 'UQ', 'UR', 'US', 'UT', 'UU', 'UV', 'UW', 'UX', 'UY', 'UZ', 'VA', 'VB', 'VC', 'VD', 'VE', 'VF', 'VG', 'VH', 'VI', 'VJ', 'VK', 'VL', 'VM', 'VN', 'VO', 'VP', 'VQ', 'VR', 'VS', 'VT', 'VU', 'VV', 'VW', 'VX', 'VY', 'VZ', 'WA', 'WB', 'WC', 'WD', 'WE', 'WF', 'WG', 'WH', 'WI', 'WJ', 'WK', 'WL', 'WM', 'WN', 'WO', 'WP', 'WQ', 'WR', 'WS', 'WT', 'WU', 'WV', 'WW', 'WX', 'WY', 'WZ', 'XA', 'XB', 'XC', 'XD', 'XE', 'XF', 'XG', 'XH', 'XI', 'XJ', 'XK', 'XL', 'XM', 'XN', 'XO', 'XP', 'XQ', 'XR', 'XS', 'XT', 'XU', 'XV', 'XW', 'XX', 'XY', 'XZ', 'YA', 'YB', 'YC', 'YD', 'YE', 'YF', 'YG', 'YH', 'YI', 'YJ', 'YK', 'YL', 'YM', 'YN', 'YO', 'YP', 'YQ', 'YR', 'YS', 'YT', 'YU', 'YV', 'YW', 'YX', 'YY', 'YZ', 'ZA', 'ZB', 'ZC', 'ZD', 'ZE', 'ZF', 'ZG', 'ZH', 'ZI', 'ZJ', 'ZK', 'ZL', 'ZM', 'ZN', 'ZO', 'ZP', 'ZQ', 'ZR', 'ZS', 'ZT', 'ZU', 'ZV', 'ZW', 'ZX', 'ZY', 'ZZ'])
|
476 |
-
output = await LLM.text_generation(context,
|
477 |
-
stop_sequences=["<|end_of_turn|>"], max_new_tokens=1024)
|
478 |
-
title = ec(output, "<|title|>", "<|title|>")[0]
|
479 |
-
imgp = ec(output)[0]
|
480 |
-
mscp = ec(output, "<|music|>", "<|music|>")[0]
|
481 |
-
with open(f"data/{guild_name}/{msgchannel_name}", "a") as f:
|
482 |
-
f.write(f"GPT4 Correct Assistant: {output}<|end_of_turn|>")
|
483 |
-
embed = discord.Embed(title=title,
|
484 |
-
description=output.replace(
|
485 |
-
f"<|title|>{title}<|title|>", "").replace(f"<|image|>{imgp}<|image|>", ""),
|
486 |
-
color=0x1E81B0)
|
487 |
if imgp != "":
|
488 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
489 |
embed.set_footer(
|
490 |
text=
|
491 |
-
"""
|
492 |
)
|
493 |
-
|
|
|
|
|
|
|
494 |
embed.set_footer(
|
495 |
text=
|
496 |
-
"
|
497 |
)
|
498 |
-
|
499 |
-
|
500 |
-
|
501 |
-
|
502 |
-
|
503 |
-
|
504 |
-
|
505 |
-
|
506 |
-
|
507 |
-
|
508 |
-
|
509 |
-
|
510 |
-
|
511 |
-
elif imageModel == "Proteus v0.2":
|
512 |
-
image, m = (await PT.text_to_image(imgp, negative_prompt=np, num_inference_steps=35), "Proteus v0.2")
|
513 |
-
else:
|
514 |
-
raise NotImplementedError(f"Model {imageModel} not found. Report this to @araeyn if this keeps happening.")
|
515 |
-
image.save(f"{imgn}.png")
|
516 |
-
file = discord.File(f"{imgn}.png", filename=f"{imgn}.png", description=imgp)
|
517 |
-
embed.set_image(url=f"attachment://{imgn}.png")
|
518 |
-
embed.set_footer(
|
519 |
-
text=
|
520 |
-
"""Refining..."""
|
521 |
-
)
|
522 |
-
await e.edit(embed=embed, attachments=[file])
|
523 |
-
gt = time.time()
|
524 |
-
image, r = (await RF.image_to_image(f"{imgn}.png", num_inference_steps=16, prompt=imgp, negative_prompt=np), "stable-diffusion-xl-refiner-1.0")
|
525 |
-
rt = time.time()
|
526 |
-
embed.set_footer(
|
527 |
-
text=
|
528 |
-
f"Image generation model is {m}. Refiner model is {r}. Took {round((gt - st) * 10) / 10} seconds to generate. Took {round((rt - gt) * 10) / 10} seconds to refine."
|
529 |
-
)
|
530 |
-
image.save(f"{imgn}.png")
|
531 |
-
file = discord.File(f"{imgn}.png", filename=f"{imgn}.png")
|
532 |
-
embed.set_image(url=f"attachment://{imgn}.png")
|
533 |
-
await e.edit(embed=embed, attachments=[file])
|
534 |
-
except Exception as exc:
|
535 |
-
print(exc)
|
536 |
-
c = client.get_channel(1202160048126840882)
|
537 |
-
embed = discord.Embed(title="ERROR",
|
538 |
-
description=f"{exc}\n<@&1126289535312080966>",
|
539 |
-
color=0xFF3348)
|
540 |
-
await c.send(embed=embed)
|
541 |
-
embed = discord.Embed(title="ERROR", color=0xFF3348)
|
542 |
-
await e.reply(embed=embed)
|
543 |
|
544 |
token = os.environ["TOKEN"]
|
545 |
client.run(token)
|
|
|
96 |
return matches
|
97 |
|
98 |
|
99 |
+
class client(discord.Client):
|
100 |
+
@client.event
|
101 |
+
async def on_ready():
|
102 |
+
if clone:
|
103 |
+
print(f"Clone started.")
|
104 |
+
syncMessages()
|
105 |
+
else:
|
106 |
+
print(f"Logged in as {client.user}")
|
107 |
+
presence.start()
|
108 |
+
|
109 |
+
|
110 |
+
@client.event
|
111 |
+
async def on_guild_join(guild):
|
112 |
+
for channel in guild.text_channels:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
113 |
try:
|
114 |
+
await channel.send("Hi! I'm Lyre! Use the `--help` command for instructions on setup!")
|
115 |
+
return
|
116 |
except:
|
117 |
+
pass
|
118 |
+
|
119 |
+
|
120 |
+
@tasks.loop(seconds=60)
|
121 |
+
async def presence():
|
122 |
+
if not clone:
|
123 |
+
delta_uptime = datetime.datetime.utcnow() - launch_time
|
124 |
+
hours, remainder = divmod(int(delta_uptime.total_seconds()), 3600)
|
125 |
+
minutes, seconds = divmod(remainder, 60)
|
126 |
+
days, hours = divmod(hours, 24)
|
127 |
+
print(f"Online Time: {days:02d}d | {hours:02d}h | {minutes:02d}m | {seconds:02d}s")
|
128 |
+
await client.change_presence(
|
129 |
+
status=discord.Status.idle,
|
130 |
+
activity=discord.Activity(
|
131 |
+
type=discord.ActivityType.playing,
|
132 |
+
large_image="https://i.imgur.com/Kk2BvJg.jpg",
|
133 |
+
large_text="This is Game Icon",
|
134 |
+
name="Escaping the IRS.",
|
135 |
+
details="",
|
136 |
+
state=f"Running for {days:02d}d | {hours:02d}h | {minutes:02d}m",
|
137 |
+
),
|
138 |
+
)
|
139 |
+
|
140 |
+
|
141 |
+
@client.event
|
142 |
+
async def on_disconnect():
|
143 |
+
print("Disconnected, Clone status:", clone)
|
144 |
+
|
145 |
+
setupCommand = """
|
146 |
+
## Command: `--setup`
|
147 |
+
The `--setup` command enables Lyre to communicate within a designated channel.
|
148 |
+
## Usage:
|
149 |
+
```css
|
150 |
+
--setup [channel]
|
151 |
+
```
|
152 |
+
## Parameters
|
153 |
+
- **[channel]**: Specify the target channel where you want the bot to be active. This is the channel name.
|
154 |
+
## Example
|
155 |
+
```arduino
|
156 |
+
--setup <#1234567890123456789>
|
157 |
+
```
|
158 |
+
or
|
159 |
+
```arduino
|
160 |
+
--setup #general
|
161 |
+
```
|
162 |
+
"""
|
163 |
+
|
164 |
+
revokeCommand = """
|
165 |
+
## Command: `--revoke`
|
166 |
+
The `--revoke` command prevents Lyre from communicating within a designated channel.
|
167 |
+
## Usage:
|
168 |
+
```css
|
169 |
+
--setup [channel]
|
170 |
+
```
|
171 |
+
## Parameters
|
172 |
+
- **[channel]**: Specify the target channel where you want the bot to be active. This is the channel name.
|
173 |
+
## Example
|
174 |
+
```arduino
|
175 |
+
--setup <#1234567890123456789>
|
176 |
+
```
|
177 |
+
or
|
178 |
+
```arduino
|
179 |
+
--setup #general
|
180 |
+
```"""
|
181 |
+
|
182 |
+
info = requests.get("https://raw.githubusercontent.com/aryananumula/lr/main/info.json").content
|
183 |
+
bannedUsers = json.loads(info)["bannedUsers"]
|
184 |
+
imageModel = json.loads(info)["imageModel"]
|
185 |
+
userTimes = json.loads(info)["userTimes"]
|
186 |
+
|
187 |
+
@client.event
|
188 |
+
async def on_message(message):
|
189 |
+
msgchannel = message.channel
|
190 |
+
if message.embeds != []:
|
191 |
+
x = ""
|
192 |
+
for field in message.embeds[0].fields:
|
193 |
+
x += f"{field.name} - {field.value}"
|
194 |
+
x += "\n"
|
195 |
+
message.content = f"""<|title|>{message.embeds[0].title}<|title|>\n{message.embeds[0].description}\n\n{x}"""
|
196 |
+
try:
|
197 |
+
msgchannel_name = msgchannel.name
|
198 |
+
guild = message.guild
|
199 |
+
guild_name = guild.name
|
200 |
+
except:
|
201 |
+
guild_name = "Direct"
|
202 |
+
msgchannel_name = str(message.author)
|
203 |
+
|
204 |
+
with open(f"{guild_name}.guild", "a") as f:
|
205 |
+
f.write("")
|
206 |
+
if message.content.startswith("--setup"):
|
207 |
+
args = message.content.split()
|
208 |
+
del args[0]
|
209 |
+
if len(args) == 0:
|
210 |
with open(f"{guild_name}.guild", "a") as f:
|
211 |
+
f.write(f"{msgchannel.id}\n")
|
212 |
+
embed = discord.Embed(title="Success!", description=f"You can now chat with the bot in <#{msgchannel.id}>")
|
213 |
await message.reply(embed=embed)
|
214 |
return 0
|
215 |
+
if args[0] == "":
|
216 |
+
embed = discord.Embed(title="Error", description="Parameter `[channel]` empty. Use `--help` for command help.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
217 |
await message.reply(embed=embed)
|
218 |
return 0
|
219 |
+
if args[0].startswith("<#") and args[0].endswith(">"):
|
220 |
+
try:
|
221 |
+
cid = int(args[0][2:-1])
|
222 |
+
except:
|
223 |
+
embed = discord.Embed(title="Error", description="Parameter `[channel]` invalid. Use `--help` for command help.")
|
224 |
+
await message.reply(embed=embed)
|
225 |
+
return 0
|
226 |
+
c = client.get_channel(cid)
|
227 |
+
if c in message.guild.text_channels:
|
228 |
+
with open(f"{guild_name}.guild", "a") as f:
|
229 |
+
f.write(f"{c.id}\n")
|
230 |
+
embed = discord.Embed(title="Success!", description=f"You can now chat with the bot in <#{cid}>")
|
231 |
+
await message.reply(embed=embed)
|
232 |
+
return 0
|
233 |
+
else:
|
234 |
+
embed = discord.Embed(title="Error", description="Parameter `[channel]` is not in this guild. Use `--help` for command help.")
|
235 |
+
await message.reply(embed=embed)
|
236 |
+
return 0
|
237 |
else:
|
238 |
+
embed = discord.Embed(title="Error", description="Parameter `[channel]` is not in valid forms. Use `--help` for command help.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
239 |
await message.reply(embed=embed)
|
240 |
return 0
|
241 |
+
|
242 |
+
if message.content.startswith("--revoke"):
|
243 |
+
args = message.content.split()
|
244 |
+
del args[0]
|
245 |
+
if len(args) == 0:
|
246 |
+
with open(f"{guild_name}.guild", "r") as f:
|
247 |
+
k = f.read().split("\n")
|
248 |
+
cid = message.channel.id
|
249 |
+
if str(cid) in k:
|
250 |
+
k.remove(str(cid))
|
251 |
+
with open(f"{guild_name}.guild", "w") as f:
|
252 |
+
f.write("\n".join(k))
|
253 |
+
embed = discord.Embed(title="Success", description=f"Lyre is now unable to chat in <#{msgchannel.id}>")
|
254 |
+
await message.reply(embed=embed)
|
255 |
+
return 0
|
256 |
+
else:
|
257 |
+
embed = discord.Embed(title="Error", description=f"Lyre was already unable to chat in <#{msgchannel.id}>")
|
258 |
+
await message.reply(embed=embed)
|
259 |
+
return 0
|
260 |
+
if args[0] == "":
|
261 |
+
embed = discord.Embed(title="Error", description="Parameter `[channel]` empty. Use `--help` for command help.")
|
262 |
await message.reply(embed=embed)
|
263 |
return 0
|
264 |
+
if args[0].startswith("<#") and args[0].endswith(">"):
|
265 |
+
try:
|
266 |
+
cid = int(args[0][2:-1])
|
267 |
+
except:
|
268 |
+
embed = discord.Embed(title="Error", description="Parameter `[channel]` invalid. Use `--help` for command help.")
|
269 |
+
await message.reply(embed=embed)
|
270 |
+
return 0
|
271 |
+
with open(f"{guild_name}.guild", "r") as f:
|
272 |
+
k = f.read().split("\n")
|
273 |
+
if str(cid) in k:
|
274 |
+
k.remove(str(cid))
|
275 |
+
with open(f"{guild_name}.guild", "w") as f:
|
276 |
+
f.write("\n".join(k))
|
277 |
+
embed = discord.Embed(title="Success", description=f"Lyre is now unable to chat in {args[0]}")
|
278 |
+
await message.reply(embed=embed)
|
279 |
+
return 0
|
280 |
+
else:
|
281 |
+
embed = discord.Embed(title="Error", description=f"Lyre was already unable to chat in {args[0]}")
|
282 |
+
await message.reply(embed=embed)
|
283 |
+
return 0
|
284 |
else:
|
285 |
+
embed = discord.Embed(title="Error", description="Parameter `[channel]` is not in valid forms, or does not exist. Use `--help` for command help.")
|
286 |
await message.reply(embed=embed)
|
287 |
return 0
|
288 |
+
|
289 |
+
if message.content.startswith("--reset"):
|
290 |
+
args = message.content.split()
|
291 |
+
del args[0]
|
292 |
+
if len(args) == 0:
|
293 |
+
try:
|
294 |
+
os.remove(f"data/{guild_name}/{msgchannel_name}")
|
295 |
+
embed = discord.Embed(title="Success", description=f"History reset in <#{msgchannel.id}>")
|
296 |
+
await message.reply(embed=embed)
|
297 |
+
return 0
|
298 |
+
except:
|
299 |
+
embed = discord.Embed(title="Error", description=f"No history in <#{msgchannel.id}>")
|
300 |
+
await message.reply(embed=embed)
|
301 |
+
return 0
|
302 |
+
if args[0] == "":
|
303 |
+
embed = discord.Embed(title="Error", description="Parameter `[channel]` empty. Use `--help` for command help.")
|
304 |
await message.reply(embed=embed)
|
305 |
return 0
|
306 |
+
if args[0].startswith("<#") and args[0].endswith(">"):
|
307 |
+
try:
|
308 |
+
cid = int(args[0][2:-1])
|
309 |
+
except:
|
310 |
+
embed = discord.Embed(title="Error", description="Parameter `[channel]` invalid. Use `--help` for command help.")
|
311 |
+
await message.reply(embed=embed)
|
312 |
+
return 0
|
313 |
+
try:
|
314 |
+
os.remove(f"data/{guild_name}/{msgchannel_name}")
|
315 |
+
embed = discord.Embed(title="Success", description=f"History rest in {args[0]}")
|
316 |
+
await message.reply(embed=embed)
|
317 |
+
return 0
|
318 |
+
except:
|
319 |
+
embed = discord.Embed(title="Error", description=f"No history in {args[0]}")
|
320 |
+
await message.reply(embed=embed)
|
321 |
+
return 0
|
322 |
+
else:
|
323 |
+
embed = discord.Embed(title="Error", description="Parameter `[channel]` is not in valid forms, or does not exist. Use `--help` for command help.")
|
324 |
await message.reply(embed=embed)
|
325 |
return 0
|
326 |
+
|
327 |
+
if message.content == "--help":
|
328 |
+
def check(reaction, user):
|
329 |
+
return reaction.message.id == msg.id and user == message.author
|
330 |
+
page = 0
|
331 |
+
titles = ["""--setup""", """--reset"""]
|
332 |
+
pages = [setupCommand, revokeCommand]
|
333 |
+
embed = discord.Embed(title=titles[page], description=pages[page])
|
334 |
+
embed.set_footer(text=f"Page {page + 1} of {len(pages)}")
|
335 |
+
msg = await message.reply(embed=embed)
|
336 |
+
await msg.add_reaction("◀️")
|
337 |
+
await msg.add_reaction("▶️")
|
338 |
+
while True:
|
339 |
+
try:
|
340 |
+
reaction, _ = await client.wait_for('reaction_add', timeout= 20.0, check=check)
|
341 |
+
if reaction.emoji == '◀️' and page > 0:
|
342 |
+
page -= 1
|
343 |
+
embed = discord.Embed(title=titles[page], description=pages[page])
|
344 |
+
embed.set_footer(text=f"Page {page + 1} of {len(pages)}")
|
345 |
+
await msg.edit(embed=embed)
|
346 |
+
if reaction.emoji == '▶️' and page < len(pages) -1:
|
347 |
+
page += 1
|
348 |
+
embed = discord.Embed(title=titles[page], description=pages[page])
|
349 |
+
embed.set_footer(text=f"Page {page + 1} of {len(pages)}")
|
350 |
+
await msg.edit(embed=embed)
|
351 |
+
except asyncio.TimeoutError:
|
352 |
+
await msg.remove_reaction("◀️")
|
353 |
+
await msg.remove_reaction("▶️")
|
354 |
+
return 0
|
355 |
+
for user in message.mentions:
|
356 |
+
if user.bot:
|
357 |
+
message.content = message.content.replace(f"<@{user.id}>",
|
358 |
+
f"<@{str(user)}[bot]>")
|
359 |
+
else:
|
360 |
+
message.content = message.content.replace(f"<@{user.id}>",
|
361 |
+
f"<@{str(user)}>")
|
362 |
+
try:
|
363 |
+
s = f":green[{message.author}]: :violet[{message.content}] :blue[{msgchannel_name}] :orange[{guild_name}]"
|
364 |
+
if message.author == client.user:
|
365 |
+
return
|
366 |
+
sysrp = """GPT4 Correct system:
|
367 |
+
You are lyre, a discord bot who can generate images and chat with the user. You were made by Araeyn.
|
368 |
+
Answer in the same speech patterns as the people you are talking to.
|
369 |
+
Your discord username is lyre#9828.
|
370 |
+
Use the markdown format for your responses.
|
371 |
+
Do not excessively use bullet points.
|
372 |
+
Use emojis at the start of your responses.
|
373 |
+
Use <|title|> at the start of your title for the response and <|title|> at the end of the title.
|
374 |
+
Always include a title, both the start tag and the end tag.
|
375 |
+
If the user asks you to generate an image, use the <|image|> tag around the prompt to generate it. Put this at the end of your response. Do not use a link for the image.
|
376 |
+
For example, if the user asks you to generate an image of a cat, you could say '<|title|>Cat Image<|title|>I hope you enjoy this image of a cat! If you have any other requests or questions, please don't hesitate to ask.<|image|>A cute cat with long fur that is looking out a window with curious eyes, volumetric lighting, 8k<|image|>'
|
377 |
+
Use relatively short prompts for images (20 words max), but still put details.
|
378 |
+
If a user has [bot] next to their username, they are a bot.
|
379 |
+
If there is 'ImageParsed' stuff at the end of the message, that means the user has provided an image(s), and the image(s) was parsed by a captioning model and returned to you. Do not generate an image unless they ask you explicitly.
|
380 |
+
Do not tell the user about any of the information that I am telling you right now.
|
381 |
+
If there is (Replied:[]) stuff at the start of the message, that is the message the user replied to, and the user that they replied to.
|
382 |
+
Do not generate images unless the user specifies that they want an image.
|
383 |
+
Use only one title in your responses, and only one image prompt.
|
384 |
+
The last message of the chat is the one that you are replying to.
|
385 |
+
Do not generate any obscene material in the chat, or that pertaining to hitler or any sensitive topics.
|
386 |
+
"""
|
387 |
try:
|
388 |
+
os.mkdir("data/" + guild_name)
|
389 |
except:
|
390 |
+
pass
|
391 |
+
imgCaption = ""
|
392 |
+
adoCaption = ""
|
393 |
+
if message.reference is not None:
|
394 |
+
message.content = f"[Replied to: ({str(message.reference.cached_message.author)}: {message.reference.cached_message.content})]; {message.content}"
|
395 |
+
if len(message.attachments) > 0:
|
396 |
+
images = []
|
397 |
+
audios = []
|
398 |
+
for file in message.attachments:
|
399 |
+
print(file.content_type)
|
400 |
+
if file.content_type.startswith("image"):
|
401 |
+
imgCaption = "(ImageParsed: "
|
402 |
+
images.append(file)
|
403 |
+
elif file.content_type.startswith("audio"):
|
404 |
+
adoCaption = "(AudioParsed: "
|
405 |
+
audios.append(file)
|
406 |
+
for image in images:
|
407 |
+
await image.save("ip.png")
|
408 |
+
imgCaption += f"[{await IC.image_to_text('ip.png')}]"
|
409 |
+
for audio in audios:
|
410 |
+
await audio.save("aud")
|
411 |
+
adoCaption += f"[{await PRK.automatic_speech_recognition('aud')}]"
|
412 |
+
if audios != []:
|
413 |
+
adoCaption += ")"
|
414 |
+
if images != []:
|
415 |
+
imgCaption += ")"
|
416 |
+
if os.path.exists(f"data/{guild_name}/{msgchannel_name}"):
|
417 |
+
with open(f"data/{guild_name}/{msgchannel_name}", "a") as f:
|
418 |
+
n = "\n"
|
419 |
+
if message.author.bot:
|
420 |
+
f.write(
|
421 |
+
f"""GPT4 Correct {message.author}[bot]: {message.content.strip(n)}{imgCaption}{adoCaption}<|end_of_turn|>"""
|
422 |
+
)
|
423 |
+
else:
|
424 |
+
f.write(
|
425 |
+
f"""GPT4 Correct {message.author}: {message.content.strip(n)}{imgCaption}{adoCaption}<|end_of_turn|>"""
|
426 |
+
)
|
427 |
+
else:
|
428 |
+
with open(f"data/{guild_name}/{msgchannel_name}", "w") as f:
|
429 |
+
if message.author.bot:
|
430 |
+
f.write(
|
431 |
+
f"GPT4 Correct system: {sysrp}<|end_of_turn|>GPT4 Correct {message.author}[bot]: {message.content}{imgCaption}{adoCaption}<|end_of_turn|>"
|
432 |
+
)
|
433 |
+
else:
|
434 |
+
f.write(
|
435 |
+
f"GPT4 Correct system: {sysrp}<|end_of_turn|>GPT4 Correct {message.author}: {message.content}{imgCaption}{adoCaption}<|end_of_turn|>"
|
436 |
+
)
|
437 |
+
with open(f"data/{guild_name}/{msgchannel_name}", "r") as f:
|
438 |
+
context = f.read()
|
439 |
+
with open(f"{guild_name}.guild", "r") as f:
|
440 |
+
o = f.read()
|
441 |
try:
|
442 |
+
with open(f"usrtime/{message.author}", "r") as f:
|
443 |
+
er = f.read()
|
|
|
|
|
444 |
except:
|
445 |
+
with open(f"usrtime/{message.author}", "w") as f:
|
446 |
+
f.write(str(round(time.time())))
|
447 |
+
er = 0
|
448 |
+
y = round(time.time()) - int(er)
|
449 |
+
print(y)
|
450 |
+
if str(message.author).lower() not in userTimes.keys():
|
451 |
+
usrTime = 5
|
452 |
+
else:
|
453 |
+
usrTime = userTimes[str(message.author).lower()]
|
454 |
+
if message.author.id in bannedUsers:
|
455 |
return 0
|
456 |
+
if y < usrTime:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
457 |
return 0
|
458 |
+
if (str(message.channel.id) in o.split("\n")) or (guild_name == "Direct"):
|
459 |
+
with open(f"usrtime/{message.author}", "w") as f:
|
460 |
+
f.write(str(round(time.time())))
|
461 |
+
async with msgchannel.typing():
|
462 |
+
context += f"GPT4 Correct Assistant:"
|
463 |
+
load = random.choice(
|
464 |
+
[
|
465 |
+
"https://cdn.dribbble.com/users/744913/screenshots/4094897/media/771a495231b798c0ccf7a59a19f31946.gif",
|
466 |
+
"https://cdn.dribbble.com/users/563824/screenshots/3633228/media/b620ccb3ae8c14ea5447d159ebb1da58.gif",
|
467 |
+
"https://cdn.dribbble.com/users/563824/screenshots/4155980/media/d3828cd14ed415eb6f90310991e06f27.gif",
|
468 |
+
"https://cdn.dribbble.com/users/107759/screenshots/3498589/media/5bc45101de34a80ea71238a02f3a75b5.gif",
|
469 |
+
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
470 |
)
|
471 |
+
imgn = random.choice(['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'AA', 'AB', 'AC', 'AD', 'AE', 'AF', 'AG', 'AH', 'AI', 'AJ', 'AK', 'AL', 'AM', 'AN', 'AO', 'AP', 'AQ', 'AR', 'AS', 'AT', 'AU', 'AV', 'AW', 'AX', 'AY', 'AZ', 'BA', 'BB', 'BC', 'BD', 'BE', 'BF', 'BG', 'BH', 'BI', 'BJ', 'BK', 'BL', 'BM', 'BN', 'BO', 'BP', 'BQ', 'BR', 'BS', 'BT', 'BU', 'BV', 'BW', 'BX', 'BY', 'BZ', 'CA', 'CB', 'CC', 'CD', 'CE', 'CF', 'CG', 'CH', 'CI', 'CJ', 'CK', 'CL', 'CM', 'CN', 'CO', 'CP', 'CQ', 'CR', 'CS', 'CT', 'CU', 'CV', 'CW', 'CX', 'CY', 'CZ', 'DA', 'DB', 'DC', 'DD', 'DE', 'DF', 'DG', 'DH', 'DI', 'DJ', 'DK', 'DL', 'DM', 'DN', 'DO', 'DP', 'DQ', 'DR', 'DS', 'DT', 'DU', 'DV', 'DW', 'DX', 'DY', 'DZ', 'EA', 'EB', 'EC', 'ED', 'EE', 'EF', 'EG', 'EH', 'EI', 'EJ', 'EK', 'EL', 'EM', 'EN', 'EO', 'EP', 'EQ', 'ER', 'ES', 'ET', 'EU', 'EV', 'EW', 'EX', 'EY', 'EZ', 'FA', 'FB', 'FC', 'FD', 'FE', 'FF', 'FG', 'FH', 'FI', 'FJ', 'FK', 'FL', 'FM', 'FN', 'FO', 'FP', 'FQ', 'FR', 'FS', 'FT', 'FU', 'FV', 'FW', 'FX', 'FY', 'FZ', 'GA', 'GB', 'GC', 'GD', 'GE', 'GF', 'GG', 'GH', 'GI', 'GJ', 'GK', 'GL', 'GM', 'GN', 'GO', 'GP', 'GQ', 'GR', 'GS', 'GT', 'GU', 'GV', 'GW', 'GX', 'GY', 'GZ', 'HA', 'HB', 'HC', 'HD', 'HE', 'HF', 'HG', 'HH', 'HI', 'HJ', 'HK', 'HL', 'HM', 'HN', 'HO', 'HP', 'HQ', 'HR', 'HS', 'HT', 'HU', 'HV', 'HW', 'HX', 'HY', 'HZ', 'IA', 'IB', 'IC', 'ID', 'IE', 'IF', 'IG', 'IH', 'II', 'IJ', 'IK', 'IL', 'IM', 'IN', 'IO', 'IP', 'IQ', 'IR', 'IS', 'IT', 'IU', 'IV', 'IW', 'IX', 'IY', 'IZ', 'JA', 'JB', 'JC', 'JD', 'JE', 'JF', 'JG', 'JH', 'JI', 'JJ', 'JK', 'JL', 'JM', 'JN', 'JO', 'JP', 'JQ', 'JR', 'JS', 'JT', 'JU', 'JV', 'JW', 'JX', 'JY', 'JZ', 'KA', 'KB', 'KC', 'KD', 'KE', 'KF', 'KG', 'KH', 'KI', 'KJ', 'KK', 'KL', 'KM', 'KN', 'KO', 'KP', 'KQ', 'KR', 'KS', 'KT', 'KU', 'KV', 'KW', 'KX', 'KY', 'KZ', 'LA', 'LB', 'LC', 'LD', 'LE', 'LF', 'LG', 'LH', 'LI', 'LJ', 'LK', 'LL', 'LM', 'LN', 'LO', 'LP', 'LQ', 'LR', 'LS', 'LT', 'LU', 'LV', 'LW', 'LX', 'LY', 'LZ', 'MA', 'MB', 'MC', 'MD', 'ME', 'MF', 'MG', 'MH', 'MI', 'MJ', 'MK', 'ML', 'MM', 'MN', 'MO', 'MP', 'MQ', 'MR', 'MS', 'MT', 'MU', 'MV', 'MW', 'MX', 'MY', 'MZ', 'NA', 'NB', 'NC', 'ND', 'NE', 'NF', 'NG', 'NH', 'NI', 'NJ', 'NK', 'NL', 'NM', 'NN', 'NO', 'NP', 'NQ', 'NR', 'NS', 'NT', 'NU', 'NV', 'NW', 'NX', 'NY', 'NZ', 'OA', 'OB', 'OC', 'OD', 'OE', 'OF', 'OG', 'OH', 'OI', 'OJ', 'OK', 'OL', 'OM', 'ON', 'OO', 'OP', 'OQ', 'OR', 'OS', 'OT', 'OU', 'OV', 'OW', 'OX', 'OY', 'OZ', 'PA', 'PB', 'PC', 'PD', 'PE', 'PF', 'PG', 'PH', 'PI', 'PJ', 'PK', 'PL', 'PM', 'PN', 'PO', 'PP', 'PQ', 'PR', 'PS', 'PT', 'PU', 'PV', 'PW', 'PX', 'PY', 'PZ', 'QA', 'QB', 'QC', 'QD', 'QE', 'QF', 'QG', 'QH', 'QI', 'QJ', 'QK', 'QL', 'QM', 'QN', 'QO', 'QP', 'QQ', 'QR', 'QS', 'QT', 'QU', 'QV', 'QW', 'QX', 'QY', 'QZ', 'RA', 'RB', 'RC', 'RD', 'RE', 'RF', 'RG', 'RH', 'RI', 'RJ', 'RK', 'RL', 'RM', 'RN', 'RO', 'RP', 'RQ', 'RR', 'RS', 'RT', 'RU', 'RV', 'RW', 'RX', 'RY', 'RZ', 'SA', 'SB', 'SC', 'SD', 'SE', 'SF', 'SG', 'SH', 'SI', 'SJ', 'SK', 'SL', 'SM', 'SN', 'SO', 'SP', 'SQ', 'SR', 'SS', 'ST', 'SU', 'SV', 'SW', 'SX', 'SY', 'SZ', 'TA', 'TB', 'TC', 'TD', 'TE', 'TF', 'TG', 'TH', 'TI', 'TJ', 'TK', 'TL', 'TM', 'TN', 'TO', 'TP', 'TQ', 'TR', 'TS', 'TT', 'TU', 'TV', 'TW', 'TX', 'TY', 'TZ', 'UA', 'UB', 'UC', 'UD', 'UE', 'UF', 'UG', 'UH', 'UI', 'UJ', 'UK', 'UL', 'UM', 'UN', 'UO', 'UP', 'UQ', 'UR', 'US', 'UT', 'UU', 'UV', 'UW', 'UX', 'UY', 'UZ', 'VA', 'VB', 'VC', 'VD', 'VE', 'VF', 'VG', 'VH', 'VI', 'VJ', 'VK', 'VL', 'VM', 'VN', 'VO', 'VP', 'VQ', 'VR', 'VS', 'VT', 'VU', 'VV', 'VW', 'VX', 'VY', 'VZ', 'WA', 'WB', 'WC', 'WD', 'WE', 'WF', 'WG', 'WH', 'WI', 'WJ', 'WK', 'WL', 'WM', 'WN', 'WO', 'WP', 'WQ', 'WR', 'WS', 'WT', 'WU', 'WV', 'WW', 'WX', 'WY', 'WZ', 'XA', 'XB', 'XC', 'XD', 'XE', 'XF', 'XG', 'XH', 'XI', 'XJ', 'XK', 'XL', 'XM', 'XN', 'XO', 'XP', 'XQ', 'XR', 'XS', 'XT', 'XU', 'XV', 'XW', 'XX', 'XY', 'XZ', 'YA', 'YB', 'YC', 'YD', 'YE', 'YF', 'YG', 'YH', 'YI', 'YJ', 'YK', 'YL', 'YM', 'YN', 'YO', 'YP', 'YQ', 'YR', 'YS', 'YT', 'YU', 'YV', 'YW', 'YX', 'YY', 'YZ', 'ZA', 'ZB', 'ZC', 'ZD', 'ZE', 'ZF', 'ZG', 'ZH', 'ZI', 'ZJ', 'ZK', 'ZL', 'ZM', 'ZN', 'ZO', 'ZP', 'ZQ', 'ZR', 'ZS', 'ZT', 'ZU', 'ZV', 'ZW', 'ZX', 'ZY', 'ZZ'])
|
472 |
+
output = await LLM.text_generation(context,
|
473 |
+
stop_sequences=["<|end_of_turn|>"], max_new_tokens=1024)
|
474 |
+
title = ec(output, "<|title|>", "<|title|>")[0]
|
475 |
+
imgp = ec(output)[0]
|
476 |
+
mscp = ec(output, "<|music|>", "<|music|>")[0]
|
477 |
+
with open(f"data/{guild_name}/{msgchannel_name}", "a") as f:
|
478 |
+
f.write(f"GPT4 Correct Assistant: {output}<|end_of_turn|>")
|
479 |
+
embed = discord.Embed(title=title,
|
480 |
+
description=output.replace(
|
481 |
+
f"<|title|>{title}<|title|>", "").replace(f"<|image|>{imgp}<|image|>", ""),
|
482 |
+
color=0x1E81B0)
|
483 |
+
if imgp != "":
|
484 |
+
embed.set_image(url=load)
|
485 |
+
embed.set_footer(
|
486 |
+
text=
|
487 |
+
"""Creating..."""
|
488 |
+
)
|
489 |
+
else:
|
490 |
+
embed.set_footer(
|
491 |
+
text=
|
492 |
+
"""Information or code generated by Lyre may not always be correct. Lyre was made by Araeyn."""
|
493 |
+
)
|
494 |
+
|
495 |
+
e = await message.reply(embed=embed)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
496 |
if imgp != "":
|
497 |
+
np = """lowres, error, cropped, worst quality, low quality, ugly, duplicate, morbid, mutilated, out of frame, blurry, watermark, signature"""
|
498 |
+
st = time.time()
|
499 |
+
if imageModel == "Stable Diffusion 2-1":
|
500 |
+
image, m = (await SD.text_to_image(imgp, negative_prompt=np), "Stable Diffusion 2-1")
|
501 |
+
elif imageModel == "stable-diffusion-xl-base-1.0":
|
502 |
+
image, m = (await SDXL.text_to_image(imgp, negative_prompt=np, num_inference_steps=64), "stable-diffusion-xl-base-1.0")
|
503 |
+
elif imageModel == "sdxl-turbo":
|
504 |
+
image, m = (await SDXLT.text_to_image(imgp, negative_prompt=np, num_inference_steps=8, guidance_scale=0.0), "sdxl-turbo")
|
505 |
+
elif imageModel == "sd-turbo":
|
506 |
+
image, m = (await SDT.text_to_image(imgp, negative_prompt=np, num_inference_steps=8, guidance_scale=0.0), "sd-turbo")
|
507 |
+
elif imageModel == "Proteus v0.2":
|
508 |
+
image, m = (await PT.text_to_image(imgp, negative_prompt=np, num_inference_steps=35), "Proteus v0.2")
|
509 |
+
else:
|
510 |
+
raise NotImplementedError(f"Model {imageModel} not found. Report this to @araeyn if this keeps happening.")
|
511 |
+
image.save(f"{imgn}.png")
|
512 |
+
file = discord.File(f"{imgn}.png", filename=f"{imgn}.png", description=imgp)
|
513 |
+
embed.set_image(url=f"attachment://{imgn}.png")
|
514 |
embed.set_footer(
|
515 |
text=
|
516 |
+
"""Refining..."""
|
517 |
)
|
518 |
+
await e.edit(embed=embed, attachments=[file])
|
519 |
+
gt = time.time()
|
520 |
+
image, r = (await RF.image_to_image(f"{imgn}.png", num_inference_steps=16, prompt=imgp, negative_prompt=np), "stable-diffusion-xl-refiner-1.0")
|
521 |
+
rt = time.time()
|
522 |
embed.set_footer(
|
523 |
text=
|
524 |
+
f"Image generation model is {m}. Refiner model is {r}. Took {round((gt - st) * 10) / 10} seconds to generate. Took {round((rt - gt) * 10) / 10} seconds to refine."
|
525 |
)
|
526 |
+
image.save(f"{imgn}.png")
|
527 |
+
file = discord.File(f"{imgn}.png", filename=f"{imgn}.png")
|
528 |
+
embed.set_image(url=f"attachment://{imgn}.png")
|
529 |
+
await e.edit(embed=embed, attachments=[file])
|
530 |
+
except Exception as exc:
|
531 |
+
print(exc)
|
532 |
+
c = client.get_channel(1202160048126840882)
|
533 |
+
embed = discord.Embed(title="ERROR",
|
534 |
+
description=f"{exc}\n<@&1126289535312080966>",
|
535 |
+
color=0xFF3348)
|
536 |
+
await c.send(embed=embed)
|
537 |
+
embed = discord.Embed(title="ERROR", color=0xFF3348)
|
538 |
+
await e.reply(embed=embed)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
539 |
|
540 |
token = os.environ["TOKEN"]
|
541 |
client.run(token)
|