Paillat-dev/presentator
1
0
Fork 0
mirror of https://github.com/Paillat-dev/presentator.git synced 2026-01-02 01:06:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update main.py

Browse Source
This commit is contained in:
Paillat
2023-01-22 23:17:56 +01:00
parent e44001f8ec
commit af984608d7
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
code/main.py
View File

@@ -11,6 +11,7 @@ import datetime
import base64 import base64
import requests import requests
from dotenv import load_dotenv from dotenv import load_dotenv
load_dotenv()
use_images = os.getenv("USE_IMAGES") use_images = os.getenv("USE_IMAGES")
if use_images != "No": import imagesGeneration if use_images != "No": import imagesGeneration
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
@@ -94,10 +95,10 @@ class:
print(image_filenames) print(image_filenames)
for images in image_filenames: for images in image_filenames:
#we download the image #we download the image
print ("generating image" + images) print ("generating image " + images + "with " + str(use_images))
r = await imagesGeneration.generate(images, f"{os.getcwd()}\\data\\{uid}\\{b64}{datenow}\\", use_images, apikey) r = await imagesGeneration.generate(images, f"{os.getcwd()}\\data\\{uid}\\{b64}{datenow}\\", str(use_images), apikey)
if 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) == "sd": os.rename(f"{os.getcwd()}\\.\\data\\{uid}\\{b64}{datenow}\\{images}_0.png", f"{os.getcwd()}\\data\\{uid}\\{b64}{datenow}\\{images}.png")
if use_images == "dalle": if str(use_images) == "dalle":
image_url = r['data'][0]['url'] image_url = r['data'][0]['url']
img_data = requests.get(image_url).content img_data = requests.get(image_url).content
with open(f'./data/{uid}/{b64}{datenow}/{images}.png', 'wb') as handler: with open(f'./data/{uid}/{b64}{datenow}/{images}.png', 'wb') as handler:
@@ -159,9 +160,7 @@ async def on_ready():
if not os.path.exists("data"): if not os.path.exists("data"):
os.mkdir("data") os.mkdir("data")
#get the openai key drom he key.env file #get the openai key drom he key.env file
load_dotenv()
token = os.getenv("TOKEN") token = os.getenv("TOKEN")
apikey = os.getenv("OPENAI") apikey = os.getenv("OPENAI")
openai.api_key = apikey openai.api_key = apikey
print(token)
bot.run(token) bot.run(token)