Update helper_func/mux.py with custom content

helper_func/mux.py

@@ -286,7 +286,7 @@ CODECS = {
     }
 }
 
-async def hardmux_vid(vid_filename, sub_filename, msg, crf='23', preset='medium', resolution='original', codec='h264', fps='original', custom_filename=None, audio_bitrate='128k'):
+async def hardmux_vid(vid_filename, sub_filename, msg, crf='23', preset='medium', resolution='original', codec='h264', fps='original', custom_filename=None, audio_bitrate='128k', audio_track='all'):
     chat_id = msg.chat.id
     
     # Set active process
@@ -364,8 +364,25 @@ async def hardmux_vid(vid_filename, sub_filename, msg, crf='23', preset='medium'
             '-preset', preset,
             '-c:a', 'aac',
             '-b:a', audio_bitrate,
-            '-map', '0:v:0',
-            '-map', '0:a:0?',
+        ])
+        
+        # Handle audio track mapping
+        if audio_track == 'all':
+            # Map all audio tracks
+            command.extend([
+                '-map', '0:v:0',  # First video stream
+                '-map', '0:a',    # All audio streams
+            ])
+        else:
+            # Map specific audio track (convert from 1-based to 0-based indexing)
+            audio_stream_index = int(audio_track) - 1
+            command.extend([
+                '-map', '0:v:0',                      # First video stream
+                '-map', f'0:a:{audio_stream_index}',  # Specific audio stream
+            ])
+        
+        # Add remaining parameters
+        command.extend([
             '-map_metadata', '-1',  # Remove all metadata
             '-sn',  # Remove all subtitle streams
         ])