111 lines
No EOL
4.4 KiB
Python
111 lines
No EOL
4.4 KiB
Python
#plugin for sparkytron3000
|
|
import os
|
|
import random
|
|
import time
|
|
from discord.ext import commands
|
|
from cogs.base_cog.bot_base_cog import BotBaseCog
|
|
|
|
|
|
class Meme(BotBaseCog):
|
|
|
|
def __init__(self, bot):
|
|
super().__init__(bot)
|
|
super().setup(__class__.__name__)
|
|
|
|
async def answer_question(self, 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 self.http_session.post(url, headers=headers, json=data) as resp:
|
|
response_data = await resp.json()
|
|
response = response_data['choices'][0]['message']['content']
|
|
return response
|
|
except:
|
|
return "error occurred in meme"
|
|
|
|
async def generate_random_meme(self, topic):
|
|
async with self.http_session.get('https://api.imgflip.com/get_memes') as resp:
|
|
response_data = await resp.json()
|
|
response = response_data['data']['memes']
|
|
memepics = [{'name':image['name'],'url':image['url'],'id':image['id']} for image in response]
|
|
|
|
#Pick a meme format
|
|
memenumber = random.randint(1,99)
|
|
meme_name = response[memenumber-1]['name']
|
|
panel_count = response[memenumber-1]['box_count']
|
|
panel_text = await self.answer_question("Create text for a meme. The meme is " + meme_name + ". It has " + str(panel_count) + " panels. 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]")
|
|
|
|
id = memenumber
|
|
imgflip_username = os.getenv('imgflip_username')
|
|
imgflip_password = os.getenv('imgflip_password')
|
|
params = {
|
|
'username':imgflip_username,
|
|
'password':imgflip_password,
|
|
'template_id':memepics[id-1]['id']
|
|
}
|
|
boxes = []
|
|
text = panel_text.split('\n')
|
|
for x in range(len(text)):
|
|
if text[x].strip() != "":
|
|
item = text[x][3:]
|
|
if len(params)-3 < panel_count:
|
|
dictionary = {"text":item, "color": "#ffffff", "outline_color": "#000000"}
|
|
boxes.append(dictionary)
|
|
|
|
for i, box in enumerate(boxes):
|
|
params[f"boxes[{i}][text]"] = box["text"]
|
|
params[f"boxes[{i}][color]"] = box["color"]
|
|
params[f"boxes[{i}][outline_color]"] = box["outline_color"]
|
|
|
|
URL = 'https://api.imgflip.com/caption_image'
|
|
|
|
try:
|
|
async with self.http_session.post(URL, params=params) as resp:
|
|
response = await resp.json()
|
|
image_link = response['data']['url']
|
|
except:
|
|
self.logger.exception("Error occurred in meme")
|
|
try:
|
|
#------------------------------------Saving Image Using Aiohttp---------------------------------#
|
|
filename = memepics[id-1]['name']
|
|
async with self.http_session.get(image_link) as response:
|
|
folder = "tmp/meme/"
|
|
filename = f"{folder}{time.time_ns()}.jpg"
|
|
|
|
with open(filename, "wb") as file:
|
|
while True:
|
|
chunk = await response.content.read(1024) # Read the response in chunks
|
|
if not chunk:
|
|
break
|
|
file.write(chunk)
|
|
except:
|
|
self.logger.exception("Something's Wrong with the aiohttp in meme So try again")
|
|
return image_link
|
|
|
|
@commands.command(
|
|
description="Meme",
|
|
help="Generates a meme based on input. Usage: !meme (topic)",
|
|
brief="Generate a meme"
|
|
)
|
|
async def meme(self, ctx, *topic):
|
|
self.logger.info(f"Meme command called by {ctx.author.name}")
|
|
if topic:
|
|
topic = ' '.join(topic)
|
|
else:
|
|
topic = await self.answer_question("Respond with a topic that will be used to generate a meme. Be brief.")
|
|
await ctx.send(f'Generating {topic} meme')
|
|
link = await self.generate_random_meme(topic)
|
|
await ctx.send(link)
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(Meme(bot)) |