Synced repo using 'sync_with_huggingface' Github Action

app.py

@@ -3,7 +3,7 @@ import threading
 
 import discord
 import gradio as gr
-from audioldm2 import audioldm2_create
+from audiodlm2 import audiodlm2_create
 from deepfloydif import deepfloydif_stage_1, deepfloydif_stage_2_react_check
 from discord import app_commands
 from discord.ext import commands
@@ -85,15 +85,15 @@ async def musicgen(ctx, prompt: str):
 
 
 @client.hybrid_command(
-    name="audioldm2",
+    name="audiodlm2",
     with_app_command=True,
     description="Enter a prompt to generate music!",
 )
 @app_commands.guilds(MY_GUILD)
-async def audioldm2(ctx, prompt: str):
-    """Audioldm2 generation"""
+async def audiodlm2(ctx, prompt: str):
+    """Audiodlm2 generation"""
     try:
-        await audioldm2_create(ctx, prompt)
+        await audiodlm2_create(ctx, prompt)
     except Exception as e:
         print(f"Error: (app.py){e}")
 

audioldm2.py

@@ -63,9 +63,12 @@ async def audioldm2_create(ctx, prompt):
                 except QueueError:
                     await thread.send("The gradio space powering this bot is really busy! Please try again later!")
 
+                short_filename = prompt[:20]
+                video_filename = f"{short_filename}.mp4"
+
                 with open(video, "rb") as file:
-                    discord_file = discord.File(file)
-                await thread.send(file=discord_file)
+                    discord_video_file = discord.File(file, filename=video_filename)
+                await thread.send(file=discord_video_file)
 
     except Exception as e:
         print(f"audioldm2_create Error: {e}")

musicgen.py

@@ -59,14 +59,17 @@ async def music_create(ctx, prompt):
 
                 audio = media_files[0]
                 video = media_files[1]
-
-                with open(audio, "rb") as file:
-                    discord_file = discord.File(file)
-                await thread.send(file=discord_file)
+                short_filename = prompt[:20]
+                audio_filename = f"{short_filename}.mp3"
+                video_filename = f"{short_filename}.mp4"
 
                 with open(video, "rb") as file:
-                    discord_file = discord.File(file)
-                await thread.send(file=discord_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)
 
                 embed = discord.Embed()
                 embed.set_thumbnail(url="https://abs.twimg.com/icons/apple-touch-icon-192x192.png")