This commit is contained in:
Paillat
2023-01-12 14:33:30 +01:00
parent 83e41c9546
commit 23fc5d7c88
2 changed files with 68 additions and 22 deletions

3
.gitignore vendored
View File

@@ -152,3 +152,6 @@ cython_debug/
#.idea/
token.env
key.env
707196665668436019
1020254043597193226

83
main.py
View File

@@ -1,4 +1,5 @@
import openai
# from openai import api_key
import discord
from discord import Intents
from discord.commands import slash_command, option
@@ -6,34 +7,54 @@ import re
import os
import asyncio
import logging
import datetime
import base64
logging.basicConfig(level=logging.INFO)
#we open the
intstructions = '''Instruction: Here is a presentation with marp. It's not possible to make slides longer than 200 characters. to separate slides,
intstructions = '''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. To add an image illustration , use ![bg left:50% 100%](a-description-of-the-image.png) at the beginning of the slide, just after "---"-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 description of the image wanted. For example " ![bg left:50% 100%](a-man-wearing-a hat-ryding-a-bicicle.png)" but son't need to show a person necessairly. The presentatio should be for everybody, all technical words and concepts, explained. The presentation is minimum 20 slides long. You can use bulletpoints. Use markdown formatting (titles, etc...)'''
then go at the line. To add an image illustration , use ![bg left:50% 100%](a-description-of-the-image.png) at the beginning of the slide, just after "---"-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 description of the image wanted. For example " ![bg left:50% 100%](a-man-wearing-a hat-ryding-a-bicicle.png)" but son't need to show a person necessairly. The presentatio should be for everybody, all technical words and concepts, explained. 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"]
languages = ["english", "french", "spanish", "german", "italian", "portuguese", "russian", "chinese", "japanese", "korean", "arabic"]
darkstyles = ["default-dark", "gaia-dark", "uncover-dark"]
async def get_style(ctx: discord.AutocompleteContext):
"""Returns a list of colors that begin with the characters entered so far."""
return [color for color in styles if color.startswith(ctx.value.lower())]
async def get_ln(ctx: discord.AutocompleteContext):
"""Returns a list of colors that begin with the characters entered so far."""
return [color for color in languages if color.startswith(ctx.value.lower())]
@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)
async def present(ctx: discord.ApplicationContext, subject: str, style: str = "default"):
@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)
async def present(ctx: discord.ApplicationContext, subject: str, style: str = "default", language: str = "english", indications: str = ""):
await ctx.defer()
prompt = intstructions + "The subject of the presentation is: " + subject + "<|endofprompt|>"
date = datetime.datetime.now()
date = date.strftime("%Y-%m-%d-%H-%M-%S")
prompt = f"{intstructions} {indications} The subject of the presentation is: {subject} The Language is: {language} <|endofprompt|> \n"
#replace the spaces in the szbject with dashes
subject2 = subject
subject = subject.replace(" ", "-")
response = openai.Completion.create(
#we save teh subject in base64 in a variable
b64 = base64.urlsafe_b64encode(subject.encode("utf-8"))
#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(uid):
os.mkdir(uid)
datenow = datetime.datetime.now()
datenow = datenow.strftime("%Y-%m-%d-%H-%M-%S")
os.mkdir(f"{uid}/{b64}{datenow}")
response = await openai.Completion.acreate(
engine="text-davinci-003",
prompt=prompt,
temperature=0.7,
@@ -64,38 +85,60 @@ class:
image_filenames = []
for match in matches:
image_filenames.append(match.group(1))
print(image_filenames)
#we create a text file with the image names and a md file for the presentation with utf8 encoding
with open(subject + ".md", "w", encoding="utf8") as f:
with open(f"./{uid}/{b64}{datenow}/{subject}.md", "w", encoding="utf8") as f:
f.write(present)
with open(subject + "-images.txt", "w") as f:
with open(f"./{uid}/{b64}{datenow}/{subject}-images.txt", "w", encoding="utf8") as f:
for image in image_filenames:
f.write(image + "\n")
#we execute the command to convert the markdown file to a pdf and html file and also generate the first slide image
cmd = f"marp --pdf --allow-local-files {subject}.md"
cmd = f"marp --pdf --allow-local-files ./{uid}/{b64}{datenow}/{subject}.md"
os.system(cmd)
print(cmd)
cmd = f"marp --image png -o {subject}.png --allow-local-files {subject}.md"
cmd = f"marp --image png -o ./{uid}/{b64}{datenow}/{subject}.png --allow-local-files ./{uid}/{b64}{datenow}/{subject}.md"
os.system(cmd)
print(cmd)
cmd = f"marp --html --allow-local-files {subject}.md"
cmd = f"marp --html --allow-local-files ./{uid}/{b64}{datenow}/{subject}.md"
os.system(cmd)
print(cmd)
#we create an embed with the first slide imageand send it with the pdf file and the markdown file
embed = discord.Embed(title=subject2, description="Thanks for using presentator bot. You can download the presentation in different formats (pdf, markdown, html). 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)
files = [discord.File(f"{subject}.pdf"), discord.File(f"{subject}.md"), discord.File(f"{subject}.html"), discord.File(f"{subject}.png")]
files = [discord.File(f"./{uid}/{b64}{datenow}/{subject}.pdf"), discord.File(f"./{uid}/{b64}{datenow}/{subject}.md"), discord.File(f"./{uid}/{b64}{datenow}/{subject}.html"), discord.File(f"./{uid}/{b64}{datenow}/{subject}.png")]
embed.set_image(url=f"attachment://{subject}.png")
#now we send the embed and all the 4 files (pdf, markdown, html, png) at the same time
await ctx.respond(embed=embed, files=files)
#we delete the markdown file and the text file with the image names
#wait for 10 seconds
await asyncio.sleep(20)
os.remove(f"{subject}.md")
os.remove(f"{subject}-images.txt")
os.remove(f"{subject}.png")
os.remove(f"{subject}.pdf")
os.remove(f"{subject}.html")
@bot.slash_command(name="list", description="List all the presentations you have created")
async def list(ctx: discord.ApplicationContext):
#we create an embed with the list of presentations
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)
#we get the list of presentations
liste = await get_presentations(str(ctx.author.id))
#we add the list of presentations to the embed
for key in liste:
embed.add_field(name=f"{liste[key]}", value=f"</get:1063051827010084925> `{key}`", inline=False)
#now we send the embed
await ctx.respond(embed=embed, ephemeral=True)
async def get_presentations(uid):
folders = os.listdir(f"./{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)
#we get the list of presentations
liste = await get_presentations(uid)
#we check if the presentation id is in the list
if pid in liste:
#if it is we send the pdf, markdown and html files
files = [discord.File(f"./{uid}/{pid}/{liste[pid]}.pdf"), discord.File(f"./{uid}/{pid}/{liste[pid]}.md"), discord.File(f"./{uid}/{pid}/{liste[pid]}.html")]
await ctx.respond(files=files, ephemeral=True)
#when the bot is ready we print a message
@bot.event
async def on_ready():