diff --git a/bot.py b/bot.py index 8f61cbd..010d19c 100644 --- a/bot.py +++ b/bot.py @@ -49,7 +49,7 @@ Music_Config = TypedDict( "playing": bool, "ctx": Optional[str], "queue": List[str], - "requestor": List[str], + "requester": List[str], }, ) @@ -59,7 +59,7 @@ formatter = logging.Formatter( MASTER_TIMEOUT = datetime.now() INITIAL_TIME_WAIT = 5 MUZYKA: Music_Config = {"playing": False, - "ctx": None, "queue": [], "requestor": []} + "ctx": None, "queue": [], "requester": []} LOGFILE = "" NETRC_FILE = "" @@ -130,6 +130,7 @@ TOKEN = authTokens[2] REMOTE_HOST_NAME = "openai" authTokens = netrc_mod.authenticators(REMOTE_HOST_NAME) openai.api_key = authTokens[2] +openAIClient = openai.AsyncOpenAI(api_key=openai.api_key) REMOTE_HOST_NAME = "spotipy" authTokens = netrc_mod.authenticators(REMOTE_HOST_NAME) @@ -314,8 +315,12 @@ async def on_message(message): logger.info("Poczatek procedury obrazkowej") message.content = message.content.replace("imaginuje sobie: ", "") logger.debug("Wywolanie obrazka: %s", message.content) - response = openai.Image.create( - prompt=message.content, n=1, size="1024x1024" + response = openAIClient.images.generate( + model="dall-e-3", + prompt=message.content, + size="1024x1024", + quality="standard", + n=1, ) image_url = response["data"][0]["url"] logger.debug("Wynikowy obrazek pod url: %s", image_url) @@ -435,7 +440,7 @@ async def play(ctx, zamawial=None, arg=None): await connect(ctx) if MUZYKA["queue"]: file_to_play = MUZYKA["queue"].pop() - zamawial = MUZYKA["requestor"].pop() + zamawial = MUZYKA["requester"].pop() else: index = random.randint(0, len(music_file_list) - 1) file_to_play = music_file_list[index] @@ -511,7 +516,7 @@ async def check(): and not voice_client.is_playing() and not voice_client.is_paused() ): - await play(MUZYKA["ctx"], MUZYKA["requestor"]) + await play(MUZYKA["ctx"], MUZYKA["requester"]) await asyncio.sleep(0.1) if os.path.getsize(LOGFILE) > 600000: shutil.copyfile(LOGFILE, f"{LOGSTORE}discord{datetime.now()}") @@ -655,10 +660,10 @@ async def handle_response(prompt, vykidailo, bartender, history, username, music temp = {"role": "user", "content": final_prompt} history.append(temp) logger.info("Rozmiar zapytania po wyslaniu %s", chat_gpt_config_request_size) + response = await openAIClient.chat.completions.create( + model=algorithm, + messages=history) - response = await openai.ChatCompletion.acreate( - model=algorithm, messages=history - ) logger.info("Historia wysłana:") logger.info(history) await asyncio.sleep(15) @@ -1055,7 +1060,7 @@ async def dej_co_ze_spotifaja(ctx): + file["name"] + ".mp3", ) - MUZYKA["requestor"].insert(0, ctx.author) + MUZYKA["requester"].insert(0, ctx.author) music_file_list.append(file) await ctx.send( f"Dodałem do listy {dir_path}{file['artist']} - + {file['name']}.mp3" @@ -1091,7 +1096,7 @@ async def dej_co_z_jutuba(ctx): global MUZYKA # pylint: disable=global-variable-not-assigned MUZYKA["queue"].insert(0, file_iter) music_file_list.append(file_iter) - MUZYKA["requestor"].insert(0, ctx.author) + MUZYKA["requester"].insert(0, ctx.author) await ctx.send(f"Dodałem do listy {file_iter}") logger.info("Jutub udany") @@ -1282,7 +1287,7 @@ async def cisza(ctx): await disconnect(ctx=ctx) while MUZYKA["queue"]: MUZYKA["queue"].pop() - MUZYKA["requestor"].pop() + MUZYKA["requester"].pop() logger.info("Stop completed") @@ -1390,7 +1395,7 @@ async def zagraj_mi_kawalek(ctx): for plik in file: global MUZYKA # pylint: disable=global-variable-not-assigned MUZYKA["queue"].insert(0, plik[1]) - MUZYKA["requestor"].insert(0, ctx.author) + MUZYKA["requester"].insert(0, ctx.author) metadata = eyed3.load(plik[1]) reply = "Dodałem do playlisty {plik[1]} z prywatnej kolekcji Hammera." if metadata.tag.title: @@ -1440,7 +1445,7 @@ async def zrob_mi_plejliste(ctx): for plik in file: global MUZYKA # pylint: disable=global-variable-not-assigned MUZYKA["queue"].insert(0, plik[1]) - MUZYKA["requestor"].insert(0, ctx.author) + MUZYKA["requester"].insert(0, ctx.author) metadata = eyed3.load(plik[1]) if metadata and metadata.tag: if metadata.tag.title: @@ -1679,7 +1684,7 @@ async def zagraj_muzyke_mego_ludu(ctx): for plik in file: global MUZYKA # pylint: disable=global-variable-not-assigned MUZYKA["queue"].insert(0, plik[1]) - MUZYKA["requestor"].insert(0, ctx.author) + MUZYKA["requester"].insert(0, ctx.author) metadata = eyed3.load(plik[1]) if metadata and metadata.tag: if metadata.tag.title: diff --git a/settings.json b/settings.json index 13d0aa1..f423305 100644 --- a/settings.json +++ b/settings.json @@ -211,7 +211,7 @@ false, false ], - "burdel": [ + "burdel\\b": [ "Oooo... Hammer nadal nie zmienił firanek - ale całkiem nieźle sprząta, jebaniutki.", 15.0, 0,