gitea/conjurer
1
0
Fork 0
mirror of https://github.com/migatu/conjurer.git synced 2026-07-14 21:38:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

saving repeatable promtp

Browse Source
This commit is contained in:
mtuszowski
2024-11-28 11:24:22 +01:00
parent 9823a94025
commit 3d5d8a2c14
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ai_commands.py
+6 -2
View File
@@ -5,7 +5,7 @@ import sys
from datetime import datetime from datetime import datetime
from enum import Enum from enum import Enum
from typing import Optional from typing import Optional
from pathlib import Path
import discord import discord
import openai import openai
import requests import requests
@@ -356,9 +356,13 @@ class Events(commands.Cog):
self.logger.debug("Wynikowy obrazek pod url: %s", image_url) self.logger.debug("Wynikowy obrazek pod url: %s", image_url)
response = requests.get(image_url, timeout=360) response = requests.get(image_url, timeout=360)
temp_file_name = message.content + ".png" temp_file_name = message.content + ".png"
temp_file_name = GRAPHICS_PATH + message.content + ".png" temp_file_name = GRAPHICS_PATH + message.content + ".png"
num = 0
while (Path(temp_file_name)).exists():
temp_file_name = GRAPHICS_PATH + message.content + str(num) + ".png"
num += 1
with open(temp_file_name, "wb") as dalle_file: with open(temp_file_name, "wb") as dalle_file:
dalle_file.write(response.content) dalle_file.write(response.content)
self.logger.info("Koniec procedury obrazkowej.") self.logger.info("Koniec procedury obrazkowej.")