Add option to stitch moviepy audio w/ ffmpeg

Author: Snazzah

README.md

@@ -33,6 +33,7 @@ Make sure to copy and paste `config-example.json` into `config.json` and fill in
 | owners | array[int] | The Discord IDs of the people able to use dev commands. |
 | case_insensitive | bool | Whether or not commands aren't case sensitive |
 | custom_help | bool | Whether or not to use custom help |
+| stitch_mpy_audio | bool | If your server has problems with MoviePy having no audio in its output, enable this to have FFmpeg add audio instead. This will make rendering slower than usual. |
 | botlist | object | Bot list tokens supported by [dbots.py](https://github.com/dbots-pkg/dbots.py) |
 
 ### Sources

config-example.json

@@ -1,6 +1,6 @@
 {
   "name": "VideoBox",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "a bot that creates funny videos",
   "prefixes": [
     "📹 ", "🎥 ", "vbox ", "videobox ",
@@ -11,6 +11,7 @@
 
   "case_insensitive": true,
   "custom_help": true,
+  "stitch_mpy_audio": false,
 
   "blocked": [],
 

extensions/models/videocog.py

@@ -34,7 +34,22 @@ async def _send_video(self, ctx, video, clips=[], spoiler=False, preset='ultrafa
         # Write and send file
         async with ctx.typing():
             async_write = self.bot.utils.force_async(video.write_videofile)
-            await async_write(videoname, threads=threads, preset=preset, verbose=False, logger=None)
+            await async_write(videoname, threads=threads, preset=preset, verbose=False, logger=None, audio=True,
+                codec='libx264', audio_codec='libmp3lame', temp_audiofile=f'{videoname}.temp-audio.mp3', remove_temp=False)
+
+            # In the rare case that moviepy doesn't correctly add audio, let FFmpeg do it.
+            if self.bot.config['stitch_mpy_audio']:
+                stream = ffmpeg.output(ffmpeg.input(videoname).video,
+                    ffmpeg.input(f'{videoname}.temp-audio.mp3').audio,
+                    f'{videoname}.final.mp4')
+                async_run = self.bot.utils.force_async(stream.run)
+                await async_run(quiet=True)
+                os.remove(f'{videoname}.temp-audio.mp3')
+                os.remove(videoname)
+                videoname += '.final.mp4'
+            else:
+                os.remove(f'{videoname}.temp-audio.mp3')
+
             if os.path.getsize(videoname) > 1000000 * 8 and self.bot.config['owo_key']: # 8 MB
                 await status_message.edit(
                     content=f"`📹` {ctx.author.mention}'s **`{ctx.command.name}`**: Uploading..."

main.py

@@ -119,8 +119,8 @@ async def on_ready(self):
         print('Initialized.\n')
 
         if len(self.config['botlist']) != 0:
-            await self.poster.post()
             self.poster.start_loop()
+            await self.poster.post()
 
     async def on_message(self, message):
         """Handles what the bot does whenever a message comes across."""