fix video

video.py

@@ -1,8 +1,7 @@
 from ffmpy import FFmpeg
 import os
-import subprocess
 import logging
-import uuid
+import subprocess
 
 # Set up logging
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -14,61 +13,45 @@ def ffmpeg_stream(file_url, token, output_dir="tmp/cache/stream"):
         os.makedirs(stream_dir)
 
     # Set up the FFmpeg command for HLS
-    output_path = stream_dir+'/output.m3u8'
-    segment_filename = stream_dir+'/segment_%03d.ts'
-    
-    ffmpeg_command = [
-        'ffmpeg',
-        '-headers', f'Authorization: Bearer {token}',
-        '-i', file_url,
-        '-c:v', 'libx264',
-        '-crf', '23',
-        '-preset', 'medium',
-        '-c:a', 'aac',
-        '-b:a', '192k',
-        '-f', 'hls',
-        '-hls_time', '10',
-        '-hls_list_size', '0',
-        '-hls_segment_filename', segment_filename,
-        output_path
-    ]
+    output_path = os.path.join(stream_dir, 'output.m3u8')
+    segment_filename = os.path.join(stream_dir, 'segment_%03d.ts')
+
+    ffmpeg = FFmpeg(
+        inputs={file_url: None},
+        outputs={
+            output_path: (
+                '-c:v libx264 '
+                '-crf 23 '
+                '-preset medium '
+                '-c:a aac '
+                '-b:a 192k '
+                '-f hls '
+                '-hls_time 10 '
+                '-hls_list_size 0 '
+                '-hls_segment_filename ' + segment_filename
+            )
+        },
+        global_options='-headers "Authorization: Bearer ' + token + '"'
+    )
 
     try:
         # Log the command being executed
-        logging.debug(f"Running FFmpeg command: {' '.join(ffmpeg_command)}")
-        
-        # Run the FFmpeg command
-        process = subprocess.Popen(
-            ffmpeg_command,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE
-        )
-
-        # Capture stdout and stderr
-        stdout, stderr = process.communicate()
-
-        # Log FFmpeg output
-        logging.debug("FFmpeg stdout: %s", stdout.decode())
-        logging.error("FFmpeg stderr: %s", stderr.decode())
+        command_str = ' '.join(ffmpeg.cmd)
+        logging.debug(f"Running FFmpeg command: {command_str}")
 
+        # Run the FFmpeg command
+        ffmpeg.run(stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        
         # 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 the unique stream ID and process
-        return process
-
     except Exception as e:
         logging.error(f"Error using FFmpeg to stream file: {e}")
-        return None, None
 
 # 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, process = 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.")
+#     ffmpeg_stream(url, token)