|
import os |
|
|
|
import asyncio |
|
import discord |
|
|
|
from gradio_client import Client |
|
from gradio_client.utils import QueueError |
|
|
|
|
|
BOT_USER_ID = 1102236653545861151 |
|
MUSIC_CHANNEL_ID = 1140990231730987058 |
|
|
|
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}") |
|
|
