Update stream_server.py

stream_server.py

@@ -98,6 +98,17 @@ async def add_video(video_name: str):
             segment_counter += num_new_segments
 
     video_names.append(video_name)
+    video_file_name = os.path.basename(video_name)
+    # Append .ts segments to the corresponding .m3u8 file
+    m3u8_file_path = HLS_DIR / f'{video_name}.m3u8'
+    # ts_files = sorted([f for f in os.listdir(HLS_DIR) if f.startswith('video_stream') and f.endswith('.ts')])
+    
+    # Open the M3U8 file in append mode and write the new segments
+    with open(m3u8_file_path, 'a') as m3u8_file:
+        # for ts_file in ts_files[-num_new_segments:]:
+        m3u8_file.write(f"#EXTINF:3.000,\n")  # Adjust duration as necessary
+        m3u8_file.write(f"/live_stream/{video_file_name}\n")
+    
     # Update concatenated playlist
     #await concatenate_playlists(video_names, HLS_DIR)