mirror of
https://github.com/Paillat-dev/presentator.git
synced 2026-01-01 16:56:20 +00:00
221 lines
12 KiB
Python
221 lines
12 KiB
Python
import openai
|
|
import discord
|
|
from discord import Intents
|
|
from discord.commands import slash_command, option
|
|
from discord.ext import commands
|
|
import re
|
|
import os
|
|
import asyncio
|
|
import logging
|
|
import datetime
|
|
import base64
|
|
import requests
|
|
import zipfile
|
|
from zipfile import ZipFile
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
use_images = os.getenv("USE_IMAGES")
|
|
cooldown = os.getenv("COOLDOWN")
|
|
if use_images != "No": import imagesGeneration
|
|
logging.basicConfig(level=logging.INFO)
|
|
imageint = ""
|
|
if use_images != "No": imageint = "To add an image illustration , use  at the beginning of the slide, just after \"---\". Use only .png. It's not possible to add technical images but only illustrations. The images are generated by an ai, the name of the file should be a detailed quite long description of the image wanted. You always need to specify the description of the image."
|
|
intstructions = f'''Here is a presentation with marp. It's not possible to make slides longer than 200 characters. to separate slides,
|
|
"
|
|
|
|
---
|
|
|
|
"
|
|
then go at the line. The presentatio should be for everybody, all technical words and concepts, explained. {imageint} The presentation is minimum 20 slides long. You can use bulletpoints. Use markdown formatting (titles, etc...). The presentation has also a conclusion.'''
|
|
bot = discord.Bot()
|
|
|
|
styles = ["default", "gaia", "uncover", "default-dark", "gaia-dark", "uncover-dark", "olive"]
|
|
languages = ["english", "french", "spanish", "german", "italian", "portuguese", "russian", "chinese", "japanese", "korean", "arabic"]
|
|
darkstyles = ["default-dark", "gaia-dark", "uncover-dark"]
|
|
customstyles = ["olive"]
|
|
async def get_style(ctx: discord.AutocompleteContext):
|
|
"""Returns a list of colors that begin with the characters entered so far."""
|
|
return [style for style in styles if style.startswith(ctx.value.lower())]
|
|
async def get_ln(ctx: discord.AutocompleteContext):
|
|
return [language for language in languages if language.startswith(ctx.value.lower())]
|
|
|
|
@bot.slash_command(name="private_present", description="Generate a presentation with marp, private command for user 707196665668436019")
|
|
@option(name="subject", description="The subject of the presentation", required=True)
|
|
@option(name="style", description="The style of the presentation", required=False, autocomplete=get_style)
|
|
@option(name="center", description="Center the text", required=False)
|
|
@option(name="language", description="The language of the presentation", required=False, autocomplete=get_ln)
|
|
@option(name="indications", description="The indications for the presentation", required=False)
|
|
#command wprks only in dm and only for user 707196665668436019
|
|
@commands.is_owner()
|
|
async def private_present(ctx: discord.ApplicationContext, subject: str, style: str = "default", center: bool = True, language: str = "english", indications: str = ""):
|
|
await present(ctx, subject, style, language, indications)
|
|
|
|
@bot.slash_command(name="present", description="Generate a presentation with marp")
|
|
#we create a function that takes the subject of the presentation and the style of the presentation as arguments, and that
|
|
@option(name="subject", description="The subject of the presentation", required=True)
|
|
@option(name="style", description="The style of the presentation", required=False, autocomplete=get_style)
|
|
@option(name="center", description="Center the text", required=False)
|
|
@option(name="language", description="The language of the presentation", required=False, autocomplete=get_ln)
|
|
@option(name="indications", description="The indications for the presentation", required=False)
|
|
# a cooldown of duration cooldown seconds, except if the user is 707196665668436019
|
|
#@commands.cooldown(1, int(cooldown), commands.BucketType.user)
|
|
@commands.cooldown(1, int(cooldown), commands.BucketType.guild)
|
|
async def normal_present(ctx: discord.ApplicationContext, subject: str, style: str = "default", center: bool = True, language: str = "english", indications: str = ""):
|
|
await present(ctx, subject, style, language, indications)
|
|
async def present(ctx: discord.ApplicationContext, subject: str, style: str = "default", center: bool = True, language: str = "english", indications: str = ""):
|
|
await ctx.defer()
|
|
date = datetime.datetime.now()
|
|
date = date.strftime("%Y-%m-%d-%H-%M-%S")
|
|
#if the style is dark
|
|
dark = False
|
|
if style in darkstyles:
|
|
style = style.replace("-dark", "")
|
|
dark = True
|
|
marp = f'''---
|
|
marp: true
|
|
theme: {styles[styles.index(style)]}
|
|
class:
|
|
'''
|
|
if dark: marp = marp + f" - invert\n---"
|
|
if center: marp = marp + " - lead\n---"
|
|
else: marp = marp + "\n---"
|
|
# if style in customstyles:
|
|
# marp = f"/* @theme {style} */\n" + marp
|
|
# print(marp)
|
|
prompt = f"{intstructions} {indications} The subject of the presentation is: {subject} The Language is: {language} <|endofprompt|> \n {marp}"
|
|
subject2 = subject
|
|
forbidden = ["\\", "/", "?", "!", ":", ";", "(", ")", "[", "]", "{", "}", "'", '"', "=", "+", "*", "&", "^", "%", "$", "#", "@", "`", "~", "|", "<", ">", ",", ".", "?", " "]
|
|
for i in forbidden:
|
|
if i in subject: subject = subject.replace(i, "-")
|
|
#we save teh subject in base64 in a variable
|
|
#if dosen't exist, create a directory called "userid" where the userid is the id of the user who called the command
|
|
uid = str(ctx.author.id)
|
|
if not os.path.exists("./data/"+uid):
|
|
os.mkdir("./data/"+uid)
|
|
datenow = datetime.datetime.now()
|
|
datenow = datenow.strftime("%Y-%m-%d-%H-%M-%S")
|
|
response = await openai.Completion.acreate(
|
|
engine="text-davinci-003",
|
|
prompt=prompt,
|
|
temperature=0.6,
|
|
max_tokens=1024,
|
|
top_p=1,
|
|
frequency_penalty=0,
|
|
presence_penalty=0,
|
|
stop=["<|endofprompt|>"]
|
|
)
|
|
#we save the output in a variable
|
|
output = response["choices"][0]["text"]
|
|
present = marp + output
|
|
##we save the output in a file called "subject.md"
|
|
matches = re.finditer(r'!\[.*?\]\((.*?)\)', present)
|
|
image_filenames = []
|
|
for match in matches:
|
|
image_filenames.append(match.group(1))
|
|
#we create a text file with the image names and a md file for the presentation with utf8 encoding
|
|
if len(subject) > 15: subject = subject[:15]
|
|
b64 = base64.urlsafe_b64encode(subject.encode("utf-8"))
|
|
os.mkdir(f"./data/{uid}/{b64}{datenow}")
|
|
path = f"./data/{uid}/{b64}{datenow}"
|
|
with open(f"{path}/{subject}-images.txt", "w", encoding="utf8") as f:
|
|
for image in image_filenames:
|
|
f.write(image + "\n")
|
|
with open(f"{path}/{subject}.md", "w", encoding="utf8") as f: f.write(present)
|
|
if len(image_filenames) > 0 and use_images!="no":
|
|
#now we first remove the extension from the image filenames by removing the last 4 characters
|
|
image_filenames = [image[:-4] for image in image_filenames]
|
|
print(image_filenames)
|
|
for images in image_filenames:
|
|
print ("generating image " + images + "with " + str(use_images))
|
|
r = await imagesGeneration.generate(images, f"{os.getcwd()}\\data\\{uid}\\{b64}{datenow}\\", str(use_images), apikey)
|
|
if str(use_images) == "sd": os.rename(f"{os.getcwd()}\\.\\data\\{uid}\\{b64}{datenow}\\{images}_0.png", f"{os.getcwd()}\\data\\{uid}\\{b64}{datenow}\\{images}.png")
|
|
if str(use_images) == "dalle":
|
|
image_url = r['data'][0]['url']
|
|
img_data = requests.get(image_url).content
|
|
with open(f'{path}/{images}.png', 'wb') as handler:
|
|
handler.write(img_data)
|
|
await asyncio.sleep(15) #wait 15 seconds to avoid rate limiting
|
|
cmd = f"--pdf --allow-local-files {path}/{subject}.md"
|
|
if style in customstyles: cmd = cmd + f" --theme ./themes/{style}.css"
|
|
if os.path.exists("./marp.exe"):
|
|
os.system(f"marp.exe {cmd}")
|
|
else:
|
|
cmd = cmd.replace("'", "\\'")
|
|
os.system(f"./marp {cmd}")
|
|
cmd = f" --image png -o {path}/{subject}.png --allow-local-files {path}/{subject}.md"
|
|
if style in customstyles: cmd = cmd + f" --theme ./themes/{style}.css"
|
|
if os.path.exists("./marp.exe"):
|
|
os.system(f"marp.exe {cmd}")
|
|
else:
|
|
cmd = cmd.replace("'", "\\'")
|
|
os.system(f"./marp {cmd}")
|
|
cmd = f" --html --allow-local-files {path}/{subject}.md"
|
|
if style in customstyles: cmd = cmd + f" --theme ./themes/{style}.css"
|
|
if os.path.exists("./marp.exe"):
|
|
os.system(f"marp.exe {cmd}")
|
|
else:
|
|
cmd = cmd.replace("'", "\\'")
|
|
os.system(f"./marp {cmd}")
|
|
cmd = f" --pptx --allow-local-files {path}/{subject}.md"
|
|
if style in customstyles: cmd = cmd + f" --theme ./themes/{style}.css"
|
|
if os.path.exists("./marp.exe"):
|
|
os.system(f"marp.exe {cmd}")
|
|
else:
|
|
cmd = cmd.replace("'", "\\'")
|
|
os.system(f"./marp {cmd}")
|
|
#now, we create a zip file with all the files
|
|
zipObj = ZipFile(f"{path}/{subject}.zip", 'w')
|
|
zipObj.write(f"{path}/{subject}.md", f"{subject}.md")
|
|
zipObj.write(f"{path}/{subject}.html", f"{subject}.html")
|
|
zipObj.write(f"{path}/{subject}.pdf", f"{subject}.pdf")
|
|
zipObj.write(f"{path}/{subject}.png", f"{subject}.png")
|
|
zipObj.write(f"{path}/{subject}.pptx", f"{subject}.pptx")
|
|
with open(f"{path}/{subject}-images.txt", "r", encoding="utf8") as f:
|
|
for image in f.readlines():
|
|
zipObj.write(f"{path}/{image.strip()}", f"{image.strip()}")
|
|
zipObj.close()
|
|
embed = discord.Embed(title=subject2, description="Thanks for using presentator bot. You will find your presentation in the attached zip file in the following formats: markdown, html, pdf, pptx, and the presentation' images. If you want to modify your presentation you can use the markdown file. More information about how to modify the file [HERE](https://marp.app).", color=discord.Color.brand_red())
|
|
files = [discord.File(f"{path}/{subject}.zip"), discord.File(f"{path}/{subject}.png")]
|
|
embed.set_image(url=f"attachment://{subject}.png")
|
|
await ctx.respond(embed=embed, files=files)
|
|
|
|
@bot.slash_command(name="list", description="List all the presentations you have created")
|
|
async def list(ctx: discord.ApplicationContext):
|
|
embed = discord.Embed(title="Presentations", description="Here is the list of all the presentations you have created. You can download the presentation in different formats (pdf, markdown, html) by doing `/get` \"*presentation id*\". The images are generated by an ai. If you want to modify your presentation you can use the markdown file. More information about how to modify the file [HERE](https://marp.app).", color=0x00ff00)
|
|
liste = await get_presentations(str(ctx.author.id))
|
|
for key in liste:
|
|
embed.add_field(name=f"{liste[key]}", value=f"</get:1063051827010084925> `{key}`", inline=False)
|
|
await ctx.respond(embed=embed, ephemeral=True)
|
|
|
|
async def get_presentations(uid):
|
|
folders = os.listdir(f"./data/{uid}")
|
|
names = {}
|
|
for folder in folders:
|
|
name = base64.urlsafe_b64decode(folder[2:-20]).decode("utf-8")
|
|
names[folder] = name
|
|
return names
|
|
|
|
@bot.slash_command(name="get", description="Get a presentation")
|
|
@option(name="pid", description="The id of the presentation", required=True)
|
|
async def get(ctx: discord.ApplicationContext, pid: str):
|
|
uid = str(ctx.author.id)
|
|
liste = await get_presentations(uid)
|
|
if pid in liste:
|
|
files = [discord.File(f"./data/{uid}/{pid}/{liste[pid]}.pdf"), discord.File(f"./data/{uid}/{pid}/{liste[pid]}.md"), discord.File(f"./data/{uid}/{pid}/{liste[pid]}.html")]
|
|
await ctx.respond(files=files, ephemeral=True)
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
print("Bot is ready")
|
|
if not os.path.exists("data"):
|
|
os.mkdir("data")
|
|
@bot.event
|
|
async def on_application_command_error(ctx, error):
|
|
#if there is an error we send a message to the user
|
|
await ctx.respond(f"An error occured: {error}", ephemeral=True)
|
|
|
|
#get the openai key drom he key.env file
|
|
token = os.getenv("TOKEN")
|
|
apikey = os.getenv("OPENAI")
|
|
openai.api_key = apikey
|
|
bot.run(token) |