Update stream_server.py

stream_server.py

@@ -13,6 +13,7 @@ HLS_DIR.mkdir(exist_ok=True)
 # Keep track of video names added to the queue
 video_names = []
 segment_counter = 1
+total_processed_duration = 0
 segment_lock = asyncio.Lock()
 
 def is_valid_path(video_name):
@@ -82,7 +83,7 @@ async def add_video(video_name, output_path, audio_duration):
     Returns:
         dict: Success message.
     """
-    
+    global total_processed_duration
     print((HLS_DIR / f'{video_name}.m3u8').exists())
     if not (HLS_DIR / f'{video_name}.m3u8').exists():
         async with segment_lock:
@@ -99,7 +100,6 @@ 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
     
@@ -135,6 +135,7 @@ async def add_video(video_name, output_path, audio_duration):
     if total_processed_duration == video_duration:
         with open(output_path, 'a') as m3u8_file:
             m3u8_file.write("#EXT-X-ENDLIST\n")
+            total_processed_duration = 0
 
     return {"message": f'"{video_name}" added to the streaming queue.'}