Update app.py

app.py

@@ -84,7 +84,7 @@ def create_master_playlist(output_paths, input_filename):
             # read the .m3u8 file and replace relative links with absolute links
             with open(path, 'r') as playlist_file:
                 content = playlist_file.readlines()
-            content = [line.replace('.ts', f'.ts\n') if line.endswith('.ts\n') else line for line in content]
+            # replace .ts extension with complete URL pointing to the TS segments on S3
             content = [f"https://{AWS_S3_BUCKET}.s3.amazonaws.com/{input_filename}/{res}/{line}" if line.endswith('.ts\n') else line for line in content]
             f.writelines(content)
     return master_playlist_path
@@ -134,7 +134,7 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
 
     output_links = []
     for path in output_paths:
-        res_folder = Path(path).stem.split("_")[1]
+        res_folder = Path(path).stem.split("_")[1] if "_" in Path(path).stem else ""
         s3_path = f"s3://{AWS_S3_BUCKET}/{Path(input_path).stem}/{res_folder}/{Path(path).name}"
         upload_file_to_s3(path, s3_path)
         output_links.append(f'<a href="https://s3-us-east-1.amazonaws.com/{AWS_S3_BUCKET}/{Path(input_path).stem}/{res_folder}/{Path(path).name}" target="_blank">Download {Path(path).stem}</a>')
@@ -144,7 +144,6 @@ def convert_video(video_file, quality, aspect_ratio, video_url):
         res_folder = Path(path).stem.split("_")[1]
         s3_path = f"s3://{AWS_S3_BUCKET}/{Path(input_path).stem}/{res_folder}/{Path(path).name}"
         upload_file_to_s3(path, s3_path)
-        output_links.append(f'<a href="https://s3-us-east-1.amazonaws.com/{AWS_S3_BUCKET}/{Path(input_path).stem}/{res_folder}/{Path(path).name}" target="_blank">Download {Path(path).stem}</a>')
 
     output_html = "<br>".join(output_links)
     return output_html