import discord from discord.ext import commands, tasks from discord.utils import get import shutil import json import requests import openai import random import time import os import io import base64 import asyncio import nacl import sys import subprocess import math from PIL import Image, PngImagePlugin from dotenv import load_dotenv import threading import matplotlib.pyplot as plt import aiohttp import aioftp #config STABLE_DIFFUSION_URL = "http://127.0.0.1:7861" #env vars START load_dotenv() imgflip_username = os.getenv('imgflip_username') imgflip_password = os.getenv('imgflip_password') openai.api_key = os.getenv('openai.api_key') discord_token = os.getenv('discord_token') eleven_labs_api_key = os.getenv('eleven_labs_api_key') ftp_server = os.getenv('ftp_server') ftp_username = os.getenv('ftp_username') ftp_password = os.getenv('ftp_password') ftp_ai_images = os.getenv('ftp_ai_images') ftp_ai_memes = os.getenv('ftp_ai_memes') ftp_ai_webpage = os.getenv('ftp_ai_webpage') ftp_public_html = os.getenv('ftp_public_html') #env vars END #discord stuff START intents = discord.Intents.default() intents.message_content = True bot = commands.Bot(command_prefix='!', intents=intents) #discord stuff END #delete save_image function once it has been used. this is temporary @bot.command() async def save_image(ctx): try: if ctx.message.content.startswith("!describe "): file_url = ctx.message.content.split(" ", maxsplit=1)[1] elif ctx.message.attachments: file_url = ctx.message.attachments[0].url else: print("No image linked or attached.") return except Exception as error: await handle_error(error) print("Couldn't find image.") return r = requests.get(file_url, stream=True) imageName = "blank_image.png" if os.path.exists(imageName): await ctx.send("Saved Image!") else: await ctx.send("Something went wrong.") @bot.command() async def moderate(ctx, filename): await ctx.send("Reminder, this currently tool works by replacing the filename on the ftp server with a black image. The description will remain the same and may need to be altered.") await upload_ftp("blank_image.png", os.getenv('ftp_ai_webpage'), filename) await ctx.send("Image replaced") async def upload_ftp(local_filename, server_folder, server_filename): client = aioftp.Client() await client.connect(ftp_server) await client.login(ftp_username, ftp_password) await client.change_directory(server_folder) await client.upload(local_filename, server_folder+server_filename, write_into=True) await client.quit() async def handle_error(error): print(error) current_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) log_line = current_time + ': ' + error with open("databases/error_log.txt", 'a') as f: f.write(log_line) return error async def upload_ftp_ai_images(filename, prompt): html_file = "phixxy.com/ai-images/index.html" html_insert = '''

''' img_list = [] server_folder = os.getenv('ftp_ai_images') client = aioftp.Client() await client.connect(ftp_server) await client.login(ftp_username, ftp_password) await client.change_directory(server_folder) server_files = await client.list() try: file_count = int(len(server_files)) except: file_count = 0 new_file_name = str(file_count) + ".png" await client.upload(filename, new_file_name, write_into=True) print("Uploaded", new_file_name) with open(html_file, 'r') as f: html_data = f.read() html_insert = html_insert.replace("", new_file_name) html_insert = html_insert.replace("", prompt) html_data = html_data.replace("", html_insert) with open(html_file, "w") as f: f.writelines(html_data) await client.upload(html_file, "index.html", write_into=True) await client.quit() def banana(orange): print("Orange you glad I didn't say banana?") def create_channel_config(filepath): config_dict = { "personality":"average", "channel_topic":"casual", "chat_enabled":False, "commands_enabled":True, "chat_history_len":5, "look_at_images":False, "react_to_msgs":False, "ftp_enabled":False } with open(filepath,"w") as f: json.dump(config_dict,f) print("Wrote config variables to file.") async def get_channel_config(channel_id): filepath = "channels/config/{0}.json".format(str(channel_id)) if not os.path.exists(filepath): create_channel_config(filepath) with open(filepath, "r") as f: config_dict = json.loads(f.readline()) return config_dict async def answer_question(topic, model="gpt-3.5-turbo"): headers = { 'Content-Type': 'application/json', 'Authorization': f'Bearer {os.getenv("openai.api_key")}', } data = { "model": model, "messages": [{"role": "user", "content": topic}] } url = "https://api.openai.com/v1/chat/completions" try: async with aiohttp.ClientSession() as session: async with session.post(url, headers=headers, json=data) as resp: response_data = await resp.json() response = response_data['choices'][0]['message']['content'] return response except Exception as error: return await handle_error(error) def extract_key_value_pairs(input_str): output_str = input_str key_value_pairs = {} tokens = input_str.split(', ') for token in tokens: if '=' in token: key, value = token.split('=') key_value_pairs[key] = value output_str = output_str.replace(token+', ', '') # Remove the key-value pair from the output string return key_value_pairs, output_str def combine_dicts(dict1, dict2): #prioritizes dict2 args combined_dict = {} for key in dict1: combined_dict[key] = dict1[key] for key in dict2: combined_dict[key] = dict2[key] return combined_dict def my_open_img_file(path): img = Image.open(path) w, h = img.size encoded = "" with io.BytesIO() as output: img.save(output, format="PNG") contents = output.getvalue() encoded = str(base64.b64encode(contents), encoding='utf-8') img.close() return encoded async def look_at(ctx, look=False): metadata = "" if look: for attachment in ctx.attachments: if attachment.url.endswith(('.jpg', '.png')): print("image seen") # see http://127.0.0.1:7860/docs for info, must be running stable diffusion with --api r = requests.get(attachment.url, stream=True) imageName = "tmp/" + str(len(os.listdir('tmp/'))) + '.png' with open(imageName, 'wb') as out_file: print('Saving image: ' + imageName) shutil.copyfileobj(r.raw, out_file) img_link = my_open_img_file(imageName) try: url = STABLE_DIFFUSION_URL payload = {"image": img_link} response = requests.post(url=f'{url}/sdapi/v1/interrogate', json=payload) r = response.json() description = r.get("caption") description = description.split(',')[0] metadata += f"\n" except Exception as error: await handle_error(error) return "ERROR: CLIP may not be running. Could not look at image." return metadata def edit_channel_config(channel_id, key, value): config_file = "channels/config/" + str(channel_id) + ".json" #if not os.path.exists(config_file): # create_channel_config(config_file) with open(config_file, 'r') as f: config_data = json.load(f) config_data[key] = value with open(config_file, "w") as f: json.dump(config_data, f) async def react_to_msg(ctx, react): if True: if not random.randint(0,10) and ctx.author.id != 1097302679836971038: system_msg = "Send only an emoji as a discord reaction to the following chat message" message = ctx.content[0] headers = { 'Content-Type': 'application/json', 'Authorization': f'Bearer {os.getenv("openai.api_key")}', } data = { "model": "gpt-3.5-turbo", "messages": [{"role": "system", "content": system_msg}, {"role": "user", "content": message}] } url = "https://api.openai.com/v1/chat/completions" try: async with aiohttp.ClientSession() as session: async with session.post(url, headers=headers, json=data) as resp: response_data = await resp.json() reaction = response_data['choices'][0]['message']['content'] await ctx.add_reaction(reaction.strip()) except Exception as error: print("Some error happened while trying to react to a message") await handle_error(error) async def log_chat_and_get_history(ctx, logfile, channel_vars): metadata = await look_at(ctx, channel_vars["look_at_images"]) log_line = metadata log_line += ctx.content log_line = ctx.author.name + ": " + log_line +"\n" chat_history = "" print("Logging: " + log_line, end="") with open(logfile, "a", encoding="utf-8") as f: f.write(log_line) with open(logfile, "r", encoding="utf-8") as f: for line in (f.readlines() [-int(channel_vars["chat_history_len"]):]): chat_history += line return chat_history async def chat_response(ctx, channel_vars, chat_history_string): async with ctx.channel.typing(): await asyncio.sleep(1) prompt = f"You are a {channel_vars['personality']} chat bot named Sparkytron 3000 created by @phixxy.com. Your personality should be {channel_vars['personality']}. You are currently in a {channel_vars['channel_topic']} chatroom. The message history is: {chat_history_string}" headers = { 'Content-Type': 'application/json', 'Authorization': f'Bearer {os.getenv("openai.api_key")}', } data = { "model": "gpt-3.5-turbo", "messages": [{"role": "user", "content": prompt}] } url = "https://api.openai.com/v1/chat/completions" try: async with aiohttp.ClientSession() as session: async with session.post(url, headers=headers, json=data) as resp: response_data = await resp.json() response = response_data['choices'][0]['message']['content'] if "Sparkytron 3000:" in response[0:17]: response = response.replace("Sparkytron 3000:", "") max_len = 1999 if len(response) > max_len: messages=[response[y-max_len:y] for y in range(max_len, len(response)+max_len,max_len)] else: messages=[response] for message in messages: await ctx.channel.send(message) except Exception as error: await handle_error(error) async def folder_setup(): folder_names = ["tmp", "channels", "users", "channels/config", "channels/logs", "databases", "databases/currency", "databases/currency/players"] for folder_name in folder_names: if not os.path.exists(folder_name): os.mkdir(folder_name) async def delete_all_files(path): for filename in os.listdir(path): if os.path.isdir(path+filename): shutil.rmtree(path+filename) elif os.path.isfile(path+filename): os.remove(path+filename) @tasks.loop(seconds=1) # Run the task every second async def task_loop(): current_time = time.localtime() #Run daily tasks if current_time.tm_hour == 17 and current_time.tm_min == 0 and current_time.tm_sec == 0: bot_stuff = bot.get_channel(544408659174883328) output = 'The following tasks failed:\n```' failed_tasks = [] await bot_stuff.send("<@242018983241318410> The current time is 5pm. Running daily tasks!") try: await blog(bot_stuff) except Exception as error: await handle_error(error) failed_tasks.append("Blogpost") try: await delete_all_files("tmp/") except Exception as error: await handle_error(error) failed_tasks.append("Delete tmp/") if failed_tasks != []: for failed_task in failed_tasks: output += failed_task + '\n' output += '```' await bot_stuff.send(output) else: await bot_stuff.send("All daily tasks successfully ran!") @bot.event async def on_ready(): print('We have logged in as {0.user}'.format(bot)) #stuff to do if first run await folder_setup() await delete_all_files("tmp/") task_loop.start() @bot.command() async def update(ctx): if ctx.author.id == 242018983241318410: output = subprocess.run(["git","pull"],capture_output=True) if output.stderr: await ctx.send("Update Failed") await ctx.send(output.stderr.decode('utf-8')) else: await ctx.send(output.stdout.decode('utf-8')) else: await ctx.send("You don't have permission to do this.") @bot.command() async def currency(ctx, arg1=None, arg2=None, arg3=None, arg4=None): def read_db(filepath): with open(filepath,"r") as fileobj: db_content = json.load(fileobj) #print(db_content,type(db_content)) return db_content def save_to_db(filepath,db_content): with open(filepath,"w") as fileobj: json.dump(db_content,fileobj,indent=4) def add_currency(filepath,amount): player_db = read_db(filepath) player_db["currency"] += amount save_to_db(filepath,player_db) return player_db def calc_level_from_xp(xp): level = min(100,math.floor(0.262615*xp**0.3220627)) return level def add_xp(filepath,player_db,time_spent): activity = player_db["status"]["current_activity"] starting_xp = player_db["skills"][activity]["xp"] starting_level = player_db["skills"][activity]["level"] if activity == "mining": equipment_level = player_db["equipment"]["pickaxe"]["level"] xp_gained = (time_spent) * equipment_level # SKILL LEVEL of XP / SEC player_db["skills"][activity]["xp"] += xp_gained new_xp = starting_xp + xp_gained new_level = calc_level_from_xp(new_xp) #calculate with curve here player_db["skills"][activity]["level"] = new_level levels_gained = new_level - starting_level summary = "" #summary should include xp gained, levels gained (only if any were gained), current level (or new level) summary += "You gained {} {} xp!".format(xp_gained, activity) if levels_gained > 0: #if levels gained > 0, then include levels gained in summary summary += "\nYou gained {} {} level(s)! You are now level {}.".format(levels_gained, activity, new_level) save_to_db(filepath, player_db) return player_db,summary def add_resources(filepath,player_db,time_spent): mining_resources = { "sapphire": { "value": 100, "amount": 0 }, "emerald": { "value": 250, "amount": 0 }, "ruby": { "value": 1000, "amount": 0 }, "diamond": { "value": 3000, "amount": 0 } } if player_db["status"]["current_activity"] == "mining": pick_power = player_db["equipment"]["pickaxe"]["power"] pick_level = player_db["equipment"]["pickaxe"]["level"] mining_level = player_db["skills"]["mining"]["level"] numerator = pick_power + pick_level + mining_level denominator = 1000 items_gained = [] time_summary = time.strftime("%H:%M:%S", time.gmtime(time_spent)) for second in range(0,time_spent): roll = random.randint(0,denominator) if roll <= numerator: #get a resource roll2 = random.randint(0,100) if roll2 <= 50: mining_resources["sapphire"]["amount"] += 1 elif roll2 <=80: mining_resources["emerald"]["amount"] += 1 elif roll2 <= 95: mining_resources["ruby"]["amount"] += 1 else: mining_resources["diamond"]["amount"] += 1 for item in mining_resources: mined_amount = mining_resources[item]["amount"] if item in player_db["items"]: player_db["items"][item]["amount"] += mined_amount items_gained.append(item.title()) items_gained.append(mined_amount) else: player_db["items"][item] = mining_resources[item] items_gained.append(item.title()) items_gained.append(mined_amount) save_to_db(filepath, player_db) summary = "You spent {} mining. You mined {} x{}, {} x{}, {} x{}, and {} x{}.".format(time_summary, *items_gained) return player_db,summary async def transfer_currency(filepath, player_db, player_id, amount): try: amount = int(amount) player2_filepath = "currency/players/" + str(player_id) + ".json" player2_db = read_db(player2_filepath) if player_db["currency"] >= amount: add_currency(filepath, -amount) add_currency(player2_filepath,amount) await ctx.send("Sent " + str(amount) + " sparks to " + str(player_id)) except FileNotFoundError: await ctx.send("They don't seem to be playing the game.") async def show_levels(player_db): output = '' for skill in player_db["skills"]: output += skill + ': ' + str(player_db["skills"][skill]["level"]) + '\n' await ctx.send(output) async def show_currency(player_db): output = 'Sparks: ' + str(player_db["currency"]) await ctx.send(output) async def show_items(player_db): output = '' for item in player_db["items"]: output += item + ': ' + str(player_db["items"][item]["amount"]) + '\n' await ctx.send(output) async def stop_activity(filepath,player_db): if player_db["status"]["current_activity"] == "idle": await ctx.send("You are currently idle. There is no activity to stop.") else: time_spent = int(time.time() - player_db["status"]["start_time"]) #integer in seconds player_db, xp_summary = add_xp(filepath,player_db,time_spent) player_db, resources_summary = add_resources(filepath,player_db,time_spent) await ctx.send(xp_summary) await ctx.send(resources_summary) player_db["status"]["current_activity"] = "idle" save_to_db(filepath,player_db) async def claim(filepath, player_db): if time.time() - player_db["status"]["last_claimed"] >= 86400: player_db = add_currency(filepath, 100) player_db["status"]["last_claimed"] = time.time() save_to_db(filepath,player_db) await ctx.send("You claimed 100 sparks!") else: await ctx.send("Sorry, you already claimed your sparks today.") async def mine(filepath, player_db): if player_db["status"]["current_activity"] == "idle": player_db["status"]["current_activity"] = "mining" player_db["status"]["start_time"] = time.time() save_to_db(filepath, player_db) await ctx.send("You start mining.") elif player_db["status"]["current_activity"] == "mining": await ctx.send("You are already mining!") else: await ctx.send("You must stop " + player_db["status"]["current_activity"] + " before you start mining!") async def gamble(filepath, player_db): pass working_dir = "databases/currency/" players_dir = "players/" sender_id = str(ctx.author.id) default_db = read_db("{0}{1}default.json".format(working_dir, players_dir)) filepath = '{0}{1}{2}.json'.format(working_dir, players_dir, sender_id) try: player_db = read_db(filepath) except FileNotFoundError: save_to_db(filepath,default_db) player_db = read_db(filepath) if arg1 == "claim": await claim(filepath, player_db) player_db = read_db(filepath) elif arg1 == "stop": await stop_activity(filepath, player_db) player_db = read_db(filepath) elif arg1 == "mine": await mine(filepath, player_db) player_db = read_db(filepath) elif arg1 == "levels": await show_levels(player_db) elif arg1 == "items": await show_items(player_db) elif (arg1 == "send" or arg1 == "give") and arg2 and arg3: await transfer_currency(filepath, player_db, arg2, arg3) player_db = read_db(filepath) else: await show_currency(player_db) @bot.command() async def meme(ctx): async def update_meme_webpage(filename): server_folder = os.getenv('ftp_ai_memes') client = aioftp.Client() await client.connect(ftp_server) await client.login(ftp_username, ftp_password) await client.change_directory(server_folder) server_files = await client.list() try: file_count = len(server_files) except: file_count = 0 new_file_name = str(file_count) + ".png" await client.upload(filename, new_file_name, write_into=True) print("Uploaded", new_file_name) with open("phixxy.com/ai-memes/index.html", 'r') as f: html_data = f.read() html_insert = '\n ' html_data = html_data.replace('',html_insert) with open("phixxy.com/ai-memes/index.html", "w") as f: f.writelines(html_data) await client.upload("phixxy.com/ai-memes/index.html", "index.html", write_into=True) #ftp.storbinary("STOR " + "index.html", open("phixxy.com/ai-memes/index.html", "rb")) await client.quit() async def generate_random_meme(topic): userAgent = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) \ AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3945.117 \ Safari/537.36' damn_data = requests.get('https://api.imgflip.com/get_memes').json()['data']['memes'] memepics = [{'name':image['name'],'url':image['url'],'id':image['id']} for image in damn_data] #Pick a meme format with 2 panels choosing_meme = True while choosing_meme: memenumber = random.randint(1,99) meme_name = damn_data[memenumber-1]['name'] panel_count = damn_data[memenumber-1]['box_count'] if panel_count == 2: print("Success") choosing_meme = False print("Trying to find a meme!") completion = openai.ChatCompletion.create(model="gpt-3.5-turbo",messages=[{"role": "user", "content": "Create text for a meme. The meme is " + meme_name + ". Only create one meme. Do not use emojis or hashtags! Use the topic: " + topic + ". Use the output format (DO NOT USE EXTRA NEWLINES AND DO NOT DESCRIBE THE PICTURE IN YOUR OUTPUT): \n1: [panel 1 text]\n2: [panel 2 text]"}]) response = completion.choices[0].message.content text = response.split('\n') text_boxes = [] for item in text: item = item[3:] text_boxes.append(item) id = memenumber txt1 = text_boxes[0] txt2 = text_boxes[1] URL = 'https://api.imgflip.com/caption_image' params = { 'username':imgflip_username, 'password':imgflip_password, 'template_id':memepics[id-1]['id'], 'text0':txt1, 'text1':txt2 } try: response = requests.request('POST',URL,params=params).json() #Printing Whether Creating A meme was A success print(f"Generated Meme = {response['success']}\nImage Link = {response['data']['url']}\nPage Link = {response['data']['page_url']}") image_link = response['data']['url'] except Exception as error: print("\nAaaaah An error occured when requesting Data..") await handle_error(error) try: #------------------------------------Saving Image Using Requests---------------------------------# filename = memepics[id-1]['name'] response = requests.get(f"{response['data']['url']}") folder = "memes/" uniqueness = random.randint(0,100000000) filename = folder + topic + str(uniqueness) + ".jpg" file = open(filename, "wb") file.write(response.content) file.close() print("Meme was Saved Successfuly") except Exception as error: await handle_error(error) print("Something's Wrong with the urllib So try again") return image_link, filename try: topic = ctx.message.content.split(" ", maxsplit=1)[1] link, filepath = await generate_random_meme(topic) channel_vars = await get_channel_config(ctx.channel.id) try: if channel_vars["ftp_enabled"]: await update_meme_webpage(filepath) except Exception as error: print("COULDN'T UPLOAD TO FTP!") await handle_error(error) await ctx.send(link) except Exception as error: await handle_error(error) await ctx.send('Something went wrong try again. Usage: !meme (topic)') @bot.command() async def rsgp(ctx, amount): cost_per_bil = 27 #1b rsgp to usd cost = (int(amount) * cost_per_bil / 1000000000) output = str(amount) + ' rsgp would cost: $' + str(cost) await ctx.send(output) @bot.command() async def suggest_blog(ctx, *args): message = ' '.join(args) if '\n' in message: await ctx.send("Send only one topic at a time.") return else: blogpost_file = "databases/blog_topics.txt" with open(blogpost_file, 'a') as f: f.writelines(message) await ctx.send("Saved suggestion!") @bot.command() async def blog(ctx): start_time = time.time() topic = '' filename = "phixxy.com/ai-blog/index.html" with open(filename, 'r', encoding="utf-8") as f: html_data = f.read() current_time = time.time() current_struct_time = time.localtime(current_time) date = time.strftime("%B %d, %Y", current_struct_time) if date in html_data: await ctx.send("I already wrote a blog post today!") return blogpost_file = "databases/blog_topics.txt" if os.path.isfile(blogpost_file): with open(blogpost_file, 'r') as f: blogpost_topics = f.read() f.seek(0) topic = f.readline() blogpost_topics = blogpost_topics.replace(topic, '') with open(blogpost_file, 'w') as f: f.write(blogpost_topics) if topic != '': await ctx.send("Writing blogpost") else: await ctx.send("No topic given for blogpost, generating one.") topic = await answer_question("Give me one topic for an absurd blogpost.") post_div = '''

''' title_prompt = 'generate an absurd essay title about ' + topic title = await answer_question(title_prompt, model="gpt-3.5-turbo") prompt = 'Write a satirical essay with a serious tone titled: "' + title + '". Do not label parts of the essay.' content = await answer_question(prompt, model="gpt-4") if title in content[:len(title)]: content = content.replace(title, '', 1) content = f"

{content}

" content = content.replace('\n\n', "

") content = content.replace("

", '') post_div = post_div.replace("", title) post_div = post_div.replace("", date) post_div = post_div.replace("", content) html_data = html_data.replace("", post_div) with open(filename, 'w', encoding="utf-8") as f: f.write(html_data) await upload_ftp(filename, "/media/sdq1/bottlecap/www/phixxy.com/public_html/ai-blog/", "index.html") run_time = time.time() - start_time print("It took " + str(run_time) + " seconds to generate the blog post!") output = "Blog Updated! (" + str(run_time) + " seconds) https://phixxy.com/ai-blog" await ctx.send(output) @bot.command() async def question(ctx): question = ctx.message.content.split(" ", maxsplit=1)[1] answer = await answer_question(question) chunks = [answer[i:i+1999] for i in range(0, len(answer), 1999)] for chunk in chunks: await ctx.send(chunk) @bot.command() async def question_gpt4(ctx): question = ctx.message.content.split(" ", maxsplit=1)[1] answer = await answer_question(question, "gpt-4") chunks = [answer[i:i+1999] for i in range(0, len(answer), 1999)] for chunk in chunks: await ctx.send(chunk) @bot.command() async def highscores(ctx, limit=0): filename = str(ctx.channel.id) + ".log" with open("channels/logs/" + filename, 'r', encoding="utf-8") as logfile: data = logfile.readlines() logfile.close() def is_username(user): for character in user: if character.isupper(): return False if not (character.isalpha() or character.isdigit() or character == '.' or character == '_'): return False return True user_message_counts = {} for line in data: try: user = line[0:line.find(':')] if is_username(user): if user not in user_message_counts and user != "" and len(user) <= 32: user_message_counts[user] = 1 else: user_message_counts[user] += 1 except Exception as error: await handle_error(error) def remove_dict_keys_if_less_than_x(dictionary,x): for key in dictionary: if dictionary[key] <= x: dictionary.pop(key) return remove_dict_keys_if_less_than_x(dictionary,x) return dictionary print(user_message_counts) remove_dict_keys_if_less_than_x(user_message_counts,limit) keys = list(user_message_counts.keys()) values = list(user_message_counts.values()) fig, ax = plt.subplots() bar_container = ax.barh(keys, values) ax.set_xlabel("Message Count") ax.set_ylabel("Username") ax.set_title("Messages Sent in " + ctx.channel.name) ax.bar_label(bar_container, label_type='center') plt.savefig(str(ctx.channel.id) + '_hiscores.png', dpi=1000, bbox_inches="tight") with open(str(ctx.channel.id) + '_hiscores.png', "rb") as fh: f = discord.File(fh, filename=str(ctx.channel.id) + '_hiscores.png') await ctx.send(file=f) @bot.command() async def highscores_server(ctx, limit=0): user_message_counts = {} data = [] for filename in os.listdir("channels/logs/"): with open("channels/logs/" + filename, 'r', encoding="utf-8") as logfile: data += logfile.readlines() logfile.close() def is_username(user): for character in user: if character.isupper(): return False if not (character.isalpha() or character.isdigit() or character == '.' or character == '_'): return False return True user_message_counts = {} for line in data: try: user = line[0:line.find(':')] if is_username(user): if user not in user_message_counts and user != "" and len(user) <= 32: user_message_counts[user] = 1 else: user_message_counts[user] += 1 except Exception as error: await handle_error(error) def remove_dict_keys_if_less_than_x(dictionary,x): for key in dictionary: if dictionary[key] <= x: dictionary.pop(key) return remove_dict_keys_if_less_than_x(dictionary,x) return dictionary print(user_message_counts) print("printed") user_message_counts = remove_dict_keys_if_less_than_x(user_message_counts,limit) keys = list(user_message_counts.keys()) values = list(user_message_counts.values()) fig, ax = plt.subplots() bar_container = ax.barh(keys, values) ax.set_xlabel("Message Count") ax.set_ylabel("Username") ax.set_title("Messages Sent in all channels I can see") ax.bar_label(bar_container, label_type='center') plt.savefig(str(ctx.channel.id) + '_hiscores.png', dpi=1000, bbox_inches="tight") with open(str(ctx.channel.id) + '_hiscores.png', "rb") as fh: f = discord.File(fh, filename=str(ctx.channel.id) + '_hiscores.png') await ctx.send(file=f) @bot.command() async def website(ctx): async def delete_local_pngs(local_folder): for filename in os.listdir(local_folder): if ".png" in filename: os.remove(local_folder + filename) async def delete_ftp_pngs(server_folder): client = aioftp.Client() await client.connect(ftp_server) await client.login(ftp_username, ftp_password) await client.change_directory(server_folder) for path, info in (await client.list()): if ".png" in path.name: print("Deleting", path.name) await client.remove(path.name) await client.quit() async def extract_image_tags(code): count = code.count("") + index1 + 1 img_tag = code[index1:index2] tags.append(img_tag) code = code[index2:] return tags async def extract_image_alt_text(tags): alt_texts = [] for tag in tags: index1 = tag.find("alt") + 5 index2 = tag[index1:].find("\"") + index1 alt_text = tag[index1:index2] alt_texts.append(alt_text) return alt_texts async def generate_images(local_folder, image_list): file_list = [] async with aiohttp.ClientSession() as session: for image in image_list: filename = image.replace(" ", "").lower() + ".png" url = STABLE_DIFFUSION_URL payload = {"prompt": image, "steps": 25} response = await session.post(url=f'{url}/sdapi/v1/txt2img', json=payload) r = await response.json() for i in r['images']: image = Image.open(io.BytesIO(base64.b64decode(i.split(",", 1)[0]))) png_payload = {"image": "data:image/png;base64," + i} response2 = await session.post(url=f'{url}/sdapi/v1/png-info', json=png_payload) pnginfo = PngImagePlugin.PngInfo() json_response = await response2.json() pnginfo.add_text("parameters", json_response.get("info")) image.save(local_folder + filename, pnginfo=pnginfo) file_list.append(filename) return file_list async def add_image_filenames(code, file_list): for filename in file_list: code = code.replace("src=\"\"", "src=\""+ filename + "\"", 1) return code async def upload_html_and_imgs(local_folder, server_folder): client = aioftp.Client() await client.connect(ftp_server) await client.login(ftp_username, ftp_password) await client.change_directory(server_folder) for filename in os.listdir(local_folder): if ".png" in filename: await client.upload(local_folder + filename, filename, write_into=True) #explicitly upload html files last! for filename in os.listdir(local_folder): if ".html" in filename: await client.upload(local_folder + filename, filename, write_into=True) await client.quit() server_folder = os.getenv('ftp_ai_webpage') server_archive_folder = "/media/sdq1/bottlecap/www/phixxy.com/public_html/webpage-archive/" local_archive_folder = "websites/" local_folder = "tmp/webpage/" working_file = local_folder + "index.html" if not os.path.exists(local_folder): os.mkdir(local_folder) try: await ctx.send("Please wait, this will take a long time! You will be able to view the website here: https://phixxy.com/ai-webpage/") with open(working_file, "w") as f: f.write("Generating Website

This webpage is currently being generated. The page will refresh once it is complete. Please be patient.

") await upload_ftp(working_file, server_folder, "index.html") topic = ctx.message.content.split(" ", maxsplit=1)[1] prompt = "Generate a webpage using html and inline css. The webpage topic should be " + topic + ". Feel free to add image tags with alt text. Leave the image source blank. The images will be added later." code = await answer_question(prompt) await delete_local_pngs(local_folder) await delete_ftp_pngs(server_folder) tags = await extract_image_tags(code) alt_texts = await extract_image_alt_text(tags) file_list = await generate_images(local_folder, alt_texts) code = await add_image_filenames(code, file_list) with open(working_file, 'w') as f: f.write(code) f.close() await upload_html_and_imgs(local_folder, server_folder) await ctx.send("Finished https://phixxy.com/ai-webpage/") except openai.error.APIConnectionError as error: await handle_error(error) await ctx.send("Failed, Try again.") @bot.command() async def feature(ctx): try: feature = ctx.message.content.split(" ", maxsplit=1)[1] with open("features.txt",'a') as f: f.writelines('\n' + feature) await ctx.send("Added " + feature) except Exception as error: await handle_error(error) with open("features.txt",'r') as f: features = f.read() await ctx.send(features) @bot.command() async def draw(ctx): try: if not os.path.isdir("tmp/draw/"): os.makedirs("tmp/draw/") if " " in ctx.message.content: amount = ctx.message.content.split(" ", maxsplit=1)[1] if int(amount) > 4: await ctx.send("No, that's too many.") return else: amount = 1 await ctx.send("Please be patient this may take some time!") prompt = "Give me 11 keywords i can use to generate art using AI. They should all be related to one piece of art. Please only respond with the keywords and no other text. Be sure to use keywords that really describe what the art portrays. Keywords should be comma separated with no other text!" completion = openai.ChatCompletion.create(model="gpt-3.5-turbo",messages=[{"role": "user", "content": prompt}]) prompt = completion.choices[0].message.content prompt = prompt.split('\n', maxsplit=1)[0] prompt = prompt.replace("AI, ", "") if "." in prompt: prompt = prompt.replace(".",",") prompt = prompt + " masterpiece, studio quality" else: prompt = prompt + ", masterpiece, studio quality" negative_prompt = "" url = STABLE_DIFFUSION_URL payload = {"prompt": prompt,"steps": 25, "negative_prompt": negative_prompt,"batch_size": amount} try: response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload) r = response.json() for i in r['images']: image = Image.open(io.BytesIO(base64.b64decode(i.split(",",1)[0]))) png_payload = {"image": "data:image/png;base64," + i} response2 = requests.post(url=f'{url}/sdapi/v1/png-info', json=png_payload) pnginfo = PngImagePlugin.PngInfo() pnginfo.add_text("parameters", response2.json().get("info")) my_filename = "tmp/draw/" + prompt + str(random.randint(0,10000000)) + ".png" image.save(my_filename, pnginfo=pnginfo) channel_vars = await get_channel_config(ctx.channel.id) if channel_vars["ftp_enabled"]: await upload_ftp_ai_images(my_filename, prompt) with open(my_filename, "rb") as fh: f = discord.File(fh, filename=my_filename) await ctx.send(file=f) await ctx.send(prompt) except Exception as error: await handle_error(error) await ctx.send("My image generation service may not be running.") except Exception as error: await handle_error(error) await ctx.send('Did you mean to use !imagine?. Usage: !draw (number)') @bot.command() async def chat(ctx, message): if "enable" in message: edit_channel_config(ctx.channel.id, "chat_enabled", True) await ctx.send("Chat Enabled") elif "disable" in message: edit_channel_config(ctx.channel.id, "chat_enabled", False) await ctx.send("Chat Disabled") else: await ctx.send("Usage: !chat (enable|disable)") @bot.command() async def reactions(ctx, message): if "enable" in message: edit_channel_config(ctx.channel.id, "react_to_msgs", True) await ctx.send("Reactions Enabled") elif "disable" in message: edit_channel_config(ctx.channel.id, "react_to_msgs", False) await ctx.send("Reactions Disabled") else: await ctx.send("Usage: !reactions (enable|disable)") @bot.command() async def memorylength(ctx, chat_history_len): if chat_history_len.isdigit(): edit_channel_config(ctx.channel.id, "chat_history_len", chat_history_len) await ctx.send("Memory changed to " + chat_history_len) else: await ctx.send("Memory unchanged, must be int.") @bot.command() async def viewimages(ctx, message): if "enable" in message: edit_channel_config(ctx.channel.id, "look_at_images", True) await ctx.send("Viewing Enabled") elif "disable" in message: edit_channel_config(ctx.channel.id, "look_at_images", False) await ctx.send("Viewing Disabled") else: await ctx.send("Usage: !viewimages (enable|disable)") @bot.command() async def enable_commands(ctx, message): if "disable" in message or "false" in message: edit_channel_config(ctx.channel.id, "commands_enabled", False) await ctx.send("Commands Disabled") else: edit_channel_config(ctx.channel.id, "commands_enabled", True) await ctx.send("Commands Enabled") @bot.command() async def topic(ctx, channel_topic): edit_channel_config(ctx.channel.id, "channel_topic", channel_topic) await ctx.send("Topic changed to " + channel_topic) @bot.command() async def python(ctx): try: code = ctx.message.content print(code.find("```"),code.rfind("```")) code = code[code.find("```")+3:code.rfind("```")] #Finds the code in codeblocks if "import" in code: await ctx.send("Imports not allowed") return 0 if "```" in code: code = code.replace("```", "") code = "import random\nimport math\n" + code if len(code) == 0: await ctx.send('Please provide some code to run') else: folder_path = "tmp/python_temp_scripts/" if not os.path.exists(folder_path): os.makedirs(folder_path) unique_num = str(len(os.listdir(folder_path))) filename = f"{folder_path}{unique_num}.py" with open(filename, "w") as f: f.write(code) try: try: response = subprocess.run(["python", filename], timeout=10, capture_output=True, check=True) except subprocess.TimeoutExpired: await ctx.send("Code took too long to run!") return 0 print("response", response.stdout.decode('utf-8')) if response.stdout.decode('utf-8') == "": await ctx.send("No Output") return 0 await ctx.send(response.stdout.decode('utf-8')) except subprocess.CalledProcessError as error: await ctx.send(error.stderr.decode('utf-8')) except Exception as error: await handle_error(error) await ctx.send("Usage: !python (codeblock)") @bot.command() async def ftp(ctx, message): if "enable" in message: edit_channel_config(ctx.channel.id, "ftp_enabled", True) await ctx.send("FTP Enabled") elif "disable" in message: edit_channel_config(ctx.channel.id, "ftp_enabled", False) await ctx.send("FTP Disabled") else: await ctx.send("Usage: !ftp (enable|disable)") @bot.command() async def personality(ctx): personality_type = ctx.message.content.split(" ", maxsplit=1)[1] edit_channel_config(ctx.channel.id, "personality", personality_type) await ctx.send("Personality changed to " + personality_type) @bot.command() async def change_model(ctx, model_choice='0'): url = STABLE_DIFFUSION_URL response = requests.get(url=f'{url}/sdapi/v1/options') config_json = response.json() current_model = config_json["sd_model_checkpoint"] output = 'Current Model: ' + current_model +'\n' if model_choice == '1': if current_model != "deliberate_v2.safetensors [9aba26abdf]": model_name = "deliberate_v2.safetensors [9aba26abdf]" else: await ctx.send("Already set to use DeliberateV2") return elif model_choice == '2': if current_model != "AnythingV5_v5PrtRE.safetensors [7f96a1a9ca]": model_name = "AnythingV5_v5PrtRE.safetensors [7f96a1a9ca]" else: await ctx.send("Already set to use AnythingV5") return elif model_choice == '3': if current_model != "Anything-V3.0.ckpt [8712e20a5d]": model_name = "Anything-V3.0.ckpt [8712e20a5d]" else: await ctx.send("Already set to use AnythingV3") return else: output += "1: DeliberateV2\n2: AnythingV5\n3: AnythingV3" await ctx.send(output) return payload = {"sd_model_checkpoint": model_name} response = requests.post(url=f'{url}/sdapi/v1/options', json=payload) output = "Changed model to: " + model_name await ctx.send(output) @bot.command() async def imagine(ctx): prompt = ctx.message.content.split(" ", maxsplit=1)[1] key_value_pairs, prompt = extract_key_value_pairs(prompt) #negative_prompt = "" negative_prompt = "badhandsv4, worst quality, lowres, EasyNegative, hermaphrodite, cropped, not in the frame, additional faces, jpeg large artifacts, jpeg small artifacts, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, disfigured, deformed, body out of frame, blurry, bad anatomy, blurred, watermark, grainy, signature, cut off, draft, not finished drawing, unfinished image, bad eyes, doll, 3d, cartoon, (bad eyes:1.2), (worst quality:1.2), (low quality:1.2), bad-image-v2-39000, (bad_prompt_version2:0.8), nude, badhandv4 By bad artist -neg easynegative ng_deepnegative_v1_75t verybadimagenegative_v1.3, (Worst Quality, Low Quality:1.4), Poorly Made Bad 3D, Lousy Bad Realistic, nsfw,(worst quality, low quality:1.4), (lip, nose, tooth, rouge, lipstick, eyeshadow:1.4), ( jpeg artifacts:1.4), (depth of field, bokeh, blurry, film grain, chromatic aberration, lens flare:1.0), (1boy, abs, muscular, rib:1.0), greyscale, monochrome, dusty sunbeams, trembling, motion lines, motion blur, emphasis lines, text, title, logo, signature, child, childlike, young, easynegative, (bad-hands-5:0.8), plain background, monochrome, poorly drawn face, poorly drawn hands, watermark, censored, (mutated hands and fingers), ugly, worst quality, low quality,, nsfw,(worst quality, low quality:1.4), (lip, nose, tooth, rouge, lipstick, eyeshadow:1.4), ( jpeg artifacts:1.4), (depth of field, bokeh, blurry, film grain, chromatic aberration, lens flare:1.0), (1boy, abs, muscular, rib:1.0), greyscale, monochrome, dusty sunbeams, trembling, motion lines, motion blur, emphasis lines, text, title, logo, signature, child, childlike, young" await ctx.send("Please be patient this may take some time! Generating: " + prompt + ".") url = STABLE_DIFFUSION_URL payload = { "prompt": prompt, "steps": 25, "negative_prompt": negative_prompt } payload = combine_dicts(payload, key_value_pairs) try: response = requests.post(url=f'{url}/sdapi/v1/txt2img', json=payload) r = response.json() for i in r['images']: if not os.path.isdir("users/" + str(ctx.author.id)): os.makedirs("users/" + str(ctx.author.id)) image = Image.open(io.BytesIO(base64.b64decode(i.split(",", 1)[0]))) png_payload = {"image": "data:image/png;base64," + i} response2 = requests.post(url=f'{url}/sdapi/v1/png-info', json=png_payload) pnginfo = PngImagePlugin.PngInfo() pnginfo.add_text("parameters", response2.json().get("info")) my_filename = "users/" + str(ctx.author.id) + '/' + prompt[0:15] + str(random.randint(0, 10000000)) + ".png" image.save(my_filename, pnginfo=pnginfo) channel_vars = await get_channel_config(ctx.channel.id) if channel_vars["ftp_enabled"]: await upload_ftp_ai_images(my_filename, prompt) with open(my_filename, "rb") as fh: f = discord.File(fh, filename=my_filename) await ctx.send(file=f) except Exception as error: await handle_error(error) await ctx.send("My image generation service may not be running.") @bot.command() async def describe(ctx): try: if ctx.message.content.startswith("!describe "): file_url = ctx.message.content.split(" ", maxsplit=1)[1] elif ctx.message.attachments: file_url = ctx.message.attachments[0].url else: print("No image linked or attached.") return except Exception as error: await handle_error(error) print("Couldn't find image.") return r = requests.get(file_url, stream=True) if not os.path.isdir(f"tmp/clip/{ctx.author.id}"): os.makedirs(f"tmp/clip/{ctx.author.id}") imageName = f"tmp/clip/{ctx.author.id}/{random.randint(0,10000000)}.png" with open(imageName, 'wb') as out_file: print(f"Saving image: {imageName}") shutil.copyfileobj(r.raw, out_file) img_link = my_open_img_file(imageName) try: url = STABLE_DIFFUSION_URL payload = {"image": img_link} response = requests.post(url=f"{url}/sdapi/v1/interrogate", json=payload) r = response.json() await ctx.send(r.get("caption")) except Exception as error: await handle_error(error) await ctx.send("My image generation service may not be running.") @bot.command() async def reimagine(ctx): #see http://127.0.0.1:7860/docs for info, must be running stable diffusion with --api try: try: file_url = ctx.message.content.split(" ", maxsplit=2)[1] prompt = ctx.message.content.split(" ", maxsplit=2)[2] except Exception as error: await handle_error(error) print("no linked image") try: file_url = ctx.message.attachments[0].url prompt = ctx.message.content.split(" ", maxsplit=1)[1] except Exception as error: await handle_error(error) print("no attached image") except Exception as error: await handle_error(error) print("couldn't find image") key_value_pairs, prompt = extract_key_value_pairs(prompt) r = requests.get(file_url, stream=True) if not os.path.isdir("tmp/reimagining/"+ str(ctx.author.id)): os.makedirs("tmp/reimagining/"+ str(ctx.author.id)) imageName = "tmp/reimagining/" + str(ctx.author.id) +"/"+ str(random.randint(0,10000000)) + '.png' with open(imageName, 'wb') as out_file: print('Saving image: ' + imageName) shutil.copyfileobj(r.raw, out_file) img_link = my_open_img_file(imageName) negative_prompt = "badhandsv4, worst quality, lowres, EasyNegative, hermaphrodite, cropped, not in the frame, additional faces, jpeg large artifacts, jpeg small artifacts, ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, disfigured, deformed, body out of frame, blurry, bad anatomy, blurred, watermark, grainy, signature, cut off, draft, not finished drawing, unfinished image, bad eyes, doll, 3d, cartoon, (bad eyes:1.2), (worst quality:1.2), (low quality:1.2), bad-image-v2-39000, (bad_prompt_version2:0.8), nude, badhandv4 By bad artist -neg easynegative ng_deepnegative_v1_75t verybadimagenegative_v1.3, (Worst Quality, Low Quality:1.4), Poorly Made Bad 3D, Lousy Bad Realistic, nsfw,(worst quality, low quality:1.4), (lip, nose, tooth, rouge, lipstick, eyeshadow:1.4), ( jpeg artifacts:1.4), (depth of field, bokeh, blurry, film grain, chromatic aberration, lens flare:1.0), (1boy, abs, muscular, rib:1.0), greyscale, monochrome, dusty sunbeams, trembling, motion lines, motion blur, emphasis lines, text, title, logo, signature, child, childlike, young, easynegative, (bad-hands-5:0.8), plain background, monochrome, poorly drawn face, poorly drawn hands, watermark, censored, (mutated hands and fingers), ugly, worst quality, low quality,, nsfw,(worst quality, low quality:1.4), (lip, nose, tooth, rouge, lipstick, eyeshadow:1.4), ( jpeg artifacts:1.4), (depth of field, bokeh, blurry, film grain, chromatic aberration, lens flare:1.0), (1boy, abs, muscular, rib:1.0), greyscale, monochrome, dusty sunbeams, trembling, motion lines, motion blur, emphasis lines, text, title, logo, signature, child, childlike, young" #negative_prompt = "" await ctx.send("Please be patient this may take some time! Generating: " + prompt + ".") try: url = STABLE_DIFFUSION_URL payload = {"init_images": [img_link], "prompt": prompt,"steps": 40, "negative_prompt": negative_prompt, "denoising_strength": 0.5} payload = combine_dicts(payload, key_value_pairs) response = requests.post(url=f'{url}/sdapi/v1/img2img', json=payload) r = response.json() for i in r['images']: if not os.path.isdir("tmp/reimagined/"+ str(ctx.author.id)): os.makedirs("tmp/reimagined/"+ str(ctx.author.id)) image = Image.open(io.BytesIO(base64.b64decode(i.split(",",1)[0]))) png_payload = {"image": "data:image/png;base64," + i} response2 = requests.post(url=f'{url}/sdapi/v1/png-info', json=png_payload) pnginfo = PngImagePlugin.PngInfo() pnginfo.add_text("parameters", response2.json().get("info")) my_filename = "tmp/reimagined/" + str(ctx.author.id) + '/' + prompt[0:15] + str(random.randint(0,10000000)) + ".png" image.save(my_filename, pnginfo=pnginfo) with open(my_filename, "rb") as fh: f = discord.File(fh, filename=my_filename) await ctx.send(file=f) except Exception as error: await handle_error(error) await ctx.send("My image generation service may not be running.") @bot.command() async def poll(ctx, question, *options: str): if len(options) > 9: await ctx.send("Error: You cannot have more than 9 options") return embed = discord.Embed(title=question, colour=discord.Colour(0x283593)) for i, option in enumerate(options): embed.add_field(name=f"Option {i+1}", value=option, inline=False) message = await ctx.send(embed=embed) numbers = {0: "\u0030\ufe0f\u20e3", 1: "\u0031\ufe0f\u20e3", 2: "\u0032\ufe0f\u20e3", 3: "\u0033\ufe0f\u20e3", 4: "\u0034\ufe0f\u20e3", 5: "\u0035\ufe0f\u20e3", 6: "\u0036\ufe0f\u20e3", 7: "\u0037\ufe0f\u20e3", 8: "\u0038\ufe0f\u20e3", 9: "\u0039\ufe0f\u20e3"} for i in range(len(options)): await message.add_reaction(numbers.get(i+1)) @bot.command() async def roll(ctx, dice_string): dice_parts = dice_string.split('d') num_dice = int(dice_parts[0]) if '+' in dice_parts[1]: die_parts = dice_parts[1].split('+') die_size = int(die_parts[0]) modifier = int(die_parts[1]) elif '-' in dice_parts[1]: die_parts = dice_parts[1].split('-') die_size = int(die_parts[0]) modifier = -int(die_parts[1]) else: die_size = int(dice_parts[1]) modifier = 0 rolls = [random.randint(1, die_size) for i in range(num_dice)] dice_str = ' + '.join([str(roll) for roll in rolls]) total = sum(rolls) + modifier await ctx.send(f'{dice_str} + {modifier} = {total}' if modifier != 0 else f'{dice_str} = {total}') @bot.command() async def kill(ctx): if ctx.author.id == 242018983241318410: exit() else: await ctx.channel.send("You don't have permission to do that.") @bot.command() async def reset(ctx): if ctx.author.id == 242018983241318410: python = sys.executable os.execl(python, python, *sys.argv) else: await ctx.channel.send("You don't have permission to do that.") @bot.event async def on_message(ctx): logfile = "channels/logs/{0}.log".format(str(ctx.channel.id)) channel_vars = await get_channel_config(ctx.channel.id) await react_to_msg(ctx, channel_vars["react_to_msgs"]) #emoji reactions chat_history_string = await log_chat_and_get_history(ctx, logfile, channel_vars) if channel_vars["commands_enabled"] or (ctx.author.id == 242018983241318410 and ctx.content[0] == "!"): await bot.process_commands(ctx) if not channel_vars["commands_enabled"]: await ctx.channel.send("This command only ran because you set it to allow to run even when commands are disabled") if channel_vars["chat_enabled"] and not ctx.author.bot: if ctx.content and ctx.content[0] != "!": await chat_response(ctx, channel_vars, chat_history_string) elif not ctx.content: await chat_response(ctx, channel_vars, chat_history_string) bot.run(discord_token)