Update stream_server.py

stream_server.py

@@ -99,6 +99,7 @@ async def add_video(video_name, output_path, audio_duration):
 
     video_names.append(video_name)
     segment_duration = 3
+    total_processed_duration = 0
     full_segments = int(audio_duration // segment_duration)
     remaining_duration = audio_duration % segment_duration
     
@@ -115,6 +116,7 @@ async def add_video(video_name, output_path, audio_duration):
         new_segment_path = f"/live_stream/video_stream{i}.ts"
         if new_segment_path not in existing_segments:
             new_segments.append(new_segment_path)
+            total_processed_duration += segment_duration
 
     
     with open(output_path, 'a') as m3u8_file:
@@ -127,10 +129,10 @@ async def add_video(video_name, output_path, audio_duration):
             if remaining_segment_path not in existing_segments:
                 m3u8_file.write(f"#EXTINF:{remaining_duration:.3f},\n")
                 m3u8_file.write(f"{remaining_segment_path}\n")
-                remaining_duration = 0           
+                total_processed_duration += remaining_duration          
 
     # Add #EXT-X-ENDLIST only once after all segments have been added
-    if remaining_duration == 0:
+    if total_processed_duration == video_duration:
         with open(output_path, 'a') as m3u8_file:
             m3u8_file.write("#EXT-X-ENDLIST\n")
 
@@ -198,8 +200,6 @@ def generate_m3u8(video_duration, output_path,segment_duration=3):
     m3u8_content += f"#EXT-X-TARGETDURATION:{segment_duration}\n"
     m3u8_content += "#EXT-X-VERSION:4\n"
     m3u8_content += "#EXT-X-MEDIA-SEQUENCE:1\n"
-    # m3u8_content += f"#EXTINF:{segment_duration:.6f},\n"
-    # m3u8_content += f"/live_stream/video_stream{i + 1}.ts\n"
     
     # # Calculate the number of full segments and the remaining duration