Update app.py

app.py

@@ -74,13 +74,18 @@ def get_output_path(input_path, res):
     output_path = output_dir / (Path(input_path).stem + f"_{res}p.m3u8")
     return output_path
 
-def create_master_playlist(output_paths):
-    master_playlist_path = output_dir / "master_playlist.m3u8"
+def create_master_playlist(output_paths, input_filename):
+    master_playlist_path = output_dir / f"{input_filename}_master_playlist.m3u8"
     with open(master_playlist_path, "w") as f:
         f.write("#EXTM3U\n")
         for path in output_paths:
-            f.write(f'#EXT-X-STREAM-INF:BANDWIDTH=8000000,RESOLUTION={Path(path).stem.split("_")[1]}\n')
-            f.write(f'{Path(path).name}\n')
+            res = Path(path).stem.split("_")[1]
+            f.write(f'#EXT-X-STREAM-INF:BANDWIDTH=8000000,RESOLUTION={res}\n')
+            # read the .m3u8 file and replace relative links with absolute links
+            with open(path, 'r') as playlist_file:
+                content = playlist_file.read()
+            content = content.replace('.ts', f'.ts?s3://{AWS_S3_BUCKET}')
+            f.write(content + '\n')
     return master_playlist_path
 
 def upload_file_to_s3(file_path, s3_path):
@@ -125,7 +130,8 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
 
         output_paths.append(output_path)
 
-    master_playlist_path = create_master_playlist(output_paths)
+    master_playlist_path = create_master_playlist(output_paths, Path(input_path).stem)
+
     output_paths.append(master_playlist_path)
 
     server_ip = get_ip_address()