File size: 2,504 Bytes
58c201b 08fb6c7 58c201b 08fb6c7 58c201b 08fb6c7 58c201b 08fb6c7 58c201b e50b540 58c201b e50b540 58c201b 08fb6c7 58c201b e50b540 58c201b f82ba90 58c201b e50b540 08fb6c7 58c201b 08fb6c7 2a3915b 58c201b 2a3915b 58c201b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
import os
import asyncio
import discord
from gradio_client import Client
from gradio_client.utils import QueueError
BOT_USER_ID = 1102236653545861151 # real
MUSIC_CHANNEL_ID = 1140990231730987058 # real musicgen channel
musicgen = Client("huggingface-projects/transformers-musicgen", hf_token=os.getenv("HF_TOKEN"))
def music_create_job(prompt, seed):
"""Generates music based on a given prompt"""
try:
job = musicgen.submit(prompt, seed, api_name="/predict")
while not job.done():
pass
return job
except Exception as e:
print(f"music_create_job Error: {e}")
async def music_create(ctx, prompt, seed):
"""Runs music_create_job in executor"""
try:
if ctx.author.id != BOT_USER_ID:
if ctx.channel.id == MUSIC_CHANNEL_ID:
if os.environ.get("TEST_ENV") == "True":
print("Safetychecks passed for music_create")
message = await ctx.send(f"**{prompt}** - {ctx.author.mention}")
if len(prompt) > 99:
small_prompt = prompt[:99]
else:
small_prompt = prompt
thread = await message.create_thread(name=small_prompt, auto_archive_duration=60)
if os.environ.get("TEST_ENV") == "True":
print("Running music_create_job...")
loop = asyncio.get_running_loop()
job = await loop.run_in_executor(None, music_create_job, prompt, seed)
try:
job.result()
files = job.outputs()
media_files = files[0]
except QueueError:
await thread.send("The gradio space powering this bot is really busy! Please try again later!")
audio = media_files[0]
video = media_files[1]
short_filename = prompt[:20]
audio_filename = f"{short_filename}.mp3"
video_filename = f"{short_filename}.mp4"
with open(video, "rb") as file:
discord_video_file = discord.File(file, filename=video_filename)
await thread.send(file=discord_video_file)
with open(audio, "rb") as file:
discord_audio_file = discord.File(file, filename=audio_filename)
await thread.send(file=discord_audio_file)
except Exception as e:
print(f"music_create Error: {e}")
|