Spaces:
Paused
Paused
File size: 2,079 Bytes
f505705 24381da fc60743 f505705 fc60743 f505705 fc60743 d2e0e4c fc60743 5c6ae28 f505705 fc60743 0201f08 24381da fc60743 24381da f505705 fc60743 24381da 5c6ae28 f505705 24381da f505705 fc60743 f505705 fc60743 24381da |
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 |
import os
import logging
import uuid
from ffmpy import FFmpeg
# Set up logging
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
def ffmpeg_stream(file_url, token, output_dir="tmp/cache/stream"):
# Generate a unique directory for the stream
unique_id = str(uuid.uuid4())
stream_dir = os.path.join(output_dir, unique_id)
if not os.path.exists(stream_dir):
os.makedirs(stream_dir)
# Set the output paths
output_path = os.path.join(stream_dir, 'output.m3u8')
segment_filename = os.path.join(stream_dir, 'segment_%03d.ts')
# Construct the FFmpeg command using ffmpy
ff = FFmpeg(
inputs={file_url: None},
outputs={
output_path: f'-c:v libx264 -crf 23 -preset medium -c:a aac -b:a 192k -f hls '
f'-hls_time 10 -hls_list_size 0 -hls_segment_filename {segment_filename}'
},
global_options=f'-headers \"Authorization: Bearer {token}\"'
)
try:
# Log the command being executed
logging.debug(f"Running FFmpeg command: {ff.cmd}")
# Run the FFmpeg command
ff.run()
# Check if the output file was created
if os.path.exists(output_path):
logging.info(f"HLS playlist created at {output_path}")
else:
logging.error(f"HLS playlist not created at {output_path}")
return unique_id, output_path
except Exception as e:
logging.error(f"Error using FFmpeg to stream file: {e}")
return None, None
# Uncomment and modify the following lines if you want to test the script
# if __name__ == "__main__":
# url = "https://huggingface.co/Unicone-Studio/jellyfin_media/resolve/main/films/Funky%20Monkey%202004/Funky%20Monkey%20(2004)%20Web-dl%201080p.mp4"
# token = os.getenv("TOKEN")
# stream_id, output_path = ffmpeg_stream(url, token)
# if stream_id:
# logging.info(f"HLS playlist created with stream ID: {stream_id}")
# else:
# logging.error("Failed to create HLS playlist.")
|