Update stream_server.py

stream_server.py

@@ -71,7 +71,7 @@ def convert_to_hls(input_path, output_dir, video_name, start_number):
 
 
 
-async def add_video(video_name: str):
+async def add_video(video_name: str, m3u8_file_path: Path):
     """
     Endpoint to add a video to the streaming queue.
 
@@ -98,16 +98,22 @@ 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
+
+    existing_segments = set()
+    if m3u8_file_path.exists():
+        with open(m3u8_file_path, 'r') as m3u8_file:
+            for line in m3u8_file:
+                if line.startswith('/live_stream/video_stream'):
+                    existing_segments.add(line.strip())
+
+    new_ts_files = sorted([f for f in os.listdir(HLS_DIR) if f.startswith('video_stream') and f.endswith('.ts')])
+
     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")
+        for ts_file in new_ts_files[-num_new_segments:]:  # Only process the new segments
+            segment_path = f"/live_stream/{ts_file}"
+            if segment_path not in existing_segments:  # Check if already exists
+                m3u8_file.write(f"#EXTINF:3.000,\n")  # Adjust duration as necessary
+                m3u8_file.write(f"{segment_path}\n")
     
     # Update concatenated playlist
     #await concatenate_playlists(video_names, HLS_DIR)