mirror of
https://github.com/nexy7574/LCC-bot.git
synced 2024-09-19 18:16:34 +01:00
Slap AI into this bitch
This commit is contained in:
parent
76b17aca5c
commit
be71b5c684
2 changed files with 45 additions and 1 deletions
|
@ -3,6 +3,10 @@ import fnmatch
|
|||
import functools
|
||||
import glob
|
||||
import io
|
||||
import pathlib
|
||||
|
||||
import openai
|
||||
import pydub
|
||||
import json
|
||||
import math
|
||||
import os
|
||||
|
@ -43,6 +47,7 @@ from selenium.webdriver.chrome.service import Service as ChromeService
|
|||
from selenium.webdriver.firefox.options import Options as FirefoxOptions
|
||||
from selenium.webdriver.firefox.service import Service as FirefoxService
|
||||
|
||||
import config
|
||||
from utils import Timer, console
|
||||
|
||||
try:
|
||||
|
@ -2438,6 +2443,43 @@ class OtherCog(commands.Cog):
|
|||
embed2.add_field(name="Source", value=used)
|
||||
await ctx.edit(embeds=[embed, embed2])
|
||||
|
||||
@commands.message_command(name="Transcribe")
|
||||
async def transcribe_message(self, ctx: discord.ApplicationContext, message: discord.Message):
|
||||
"""Transcribes a message."""
|
||||
await ctx.defer()
|
||||
if not message.attachments:
|
||||
return await ctx.respond("No attachments found.")
|
||||
|
||||
for attachment in message.attachments:
|
||||
if attachment.content_type == "audio/ogg":
|
||||
break
|
||||
else:
|
||||
return await ctx.respond("No voice messages.")
|
||||
if getattr(config, "OPENAI_KEY", None) is None:
|
||||
return await ctx.respond("Service unavailable.")
|
||||
client = openai.OpenAI(api_key=config.OPENAI_KEY)
|
||||
with tempfile.NamedTemporaryFile("wb+", suffix=".mp4") as f:
|
||||
with tempfile.NamedTemporaryFile("wb+", suffix="-" + attachment.filename) as f2:
|
||||
await attachment.save(f2.name)
|
||||
f2.seek(0)
|
||||
seg: pydub.AudioSegment = await asyncio.to_thread(pydub.AudioSegment.from_ogg, file=f2)
|
||||
seg = seg.set_channels(1)
|
||||
await asyncio.to_thread(
|
||||
seg.export, f.name, format="mp4"
|
||||
)
|
||||
f.seek(0)
|
||||
|
||||
transcript = await asyncio.to_thread(
|
||||
client.audio.transcriptions.create,
|
||||
file=pathlib.Path(f.name),
|
||||
model="whisper-1"
|
||||
)
|
||||
paginator = commands.Paginator("", "", 4096)
|
||||
for line in transcript.text.splitlines():
|
||||
paginator.add_line(textwrap.shorten(line, 4096))
|
||||
embeds = list(map(lambda p: discord.Embed(description=p), paginator.pages))
|
||||
return await ctx.respond(embeds=embeds or [discord.Embed(description="No text found.")])
|
||||
|
||||
|
||||
def setup(bot):
|
||||
bot.add_cog(OtherCog(bot))
|
||||
|
|
4
main.py
4
main.py
|
@ -1,6 +1,7 @@
|
|||
import asyncio
|
||||
import sys
|
||||
import logging
|
||||
import textwrap
|
||||
from datetime import datetime, timedelta, timezone
|
||||
|
||||
import config
|
||||
|
@ -48,7 +49,8 @@ async def on_application_command_error(ctx: discord.ApplicationContext, error: E
|
|||
if ctx.user.id == 1019233057519177778:
|
||||
await ctx.respond("Uh oh! I did a fucky wucky >.< I'll make sure to let important peoplez know straight away!!")
|
||||
else:
|
||||
await ctx.respond("Application Command Error: `%r`" % error)
|
||||
text = "Application Command Error: `%r`" % error
|
||||
await ctx.respond(textwrap.shorten(text, 2000))
|
||||
raise error
|
||||
|
||||
|
||||
|
|
Loading…
Reference in a new issue