Merge pull request 'massive changes' (#3) from dev into main

Reviewed-on: https://codeberg.org/salmonik/senator-sperma/pulls/3
stable enough

1 files changed, 18 insertions, 17 deletions

diff --git a/cogs/ai.py b/cogs/ai.py
index 4a04693..6f79ac5 100644
--- a/cogs/ai.py
+++ b/cogs/ai.py
@@ -7,7 +7,7 @@ import os
 import discord
 from discord.ext import commands
 import logging
-import requests
+import httpx
 from gtts import gTTS
 from cogs.music import audio_playback
 import io
@@ -52,23 +52,24 @@ class AI(commands.Cog):
              ],
 
          }
-        response = requests.post(url=url, headers=headers, json=payload)
-        response_data = response.json()
-        logger.info(response_data)
-        if not response_data["choices"]:
-            await ctx.send_followup("bot siem zesroł")
-        else:
-            response_text = response_data["choices"][0]["message"]["content"]
-            if tts:
-                # Generate TTS
-                tts_obj = gTTS(text=response_text, lang='pl')
-                tts_buffer = io.BytesIO()
-                tts_obj.write_to_fp(tts_buffer)
-                tts_buffer.seek(0)
-                await audio_playback.AudioPlaybackManager.play(ctx, discord.FFmpegPCMAudio(tts_buffer), delete_after_play=True)
-
+        async with httpx.AsyncClient() as client:
+            response = await client.post(url=url, headers=headers, json=payload)
+            response_data = response.json()
+            logger.info(response_data)
+            if not response_data["choices"]:
+                await ctx.send_followup("bot siem zesroł")
             else:
-                await ctx.send_followup(response_text)
+                response_text = response_data["choices"][0]["message"]["content"]
+                if tts:
+                    # Generate TTS
+                    tts_obj = gTTS(text=response_text, lang='pl')
+                    tts_buffer = io.BytesIO()
+                    tts_obj.write_to_fp(tts_buffer)
+                    tts_buffer.seek(0)
+                    await audio_playback.AudioPlaybackManager.play(ctx, discord.FFmpegPCMAudio(tts_buffer), delete_after_play=True)
+
+                else:
+                    await ctx.send_followup(response_text)