fix: fix the queue communication (#136)

Author: timerring

src/burn/only_render.py

@@ -11,8 +11,6 @@
 import threading
 import time
 
-render_queue = queue.Queue()
-
 def normalize_video_path(filepath):
     """Normalize the video path to upload
     Args:
@@ -57,17 +55,21 @@ def render_video_only(video_path):
     with open(f"{SRC_DIR}/upload/uploadVideoQueue.txt", "a") as file:
         file.write(f"{format_video_path}\n")
 
-def monitor_queue():
-    while True:
-        if not render_queue.empty():
-            video_path = render_queue.get()
-            render_video_only(video_path)
-        else:
-            time.sleep(1)
+class VideoRenderQueue:
+    def __init__(self):
+        self.render_queue = queue.Queue()
+
+    def pipeline_render(self, video_path):
+        generate_subtitles(video_path)
+        self.render_queue.put(video_path)
 
-def pipeline_render(video_path):
-    generate_subtitles(video_path)
-    render_queue.put(video_path)
+    def monitor_queue(self):
+        while True:
+            if not self.render_queue.empty():
+                video_path = self.render_queue.get()
+                render_video_only(video_path)
+            else:
+                time.sleep(1)
 
 if __name__ == '__main__':
     # Read and define variables

src/burn/scan.py

@@ -2,7 +2,7 @@
 
 import os
 from pathlib import Path
-from src.burn.only_render import render_video_only, pipeline_render, monitor_queue
+from src.burn.only_render import render_video_only, VideoRenderQueue
 from src.burn.render_and_merge import render_and_merge
 import time
 from src.config import VIDEOS_DIR, MODEL_TYPE
@@ -44,13 +44,14 @@ def process_folder_append(folder_path):
     for file in mp4_files:
         print(f"Processing {file}...", flush=True)
         if MODEL_TYPE == "pipeline":
-            pipeline_render(file)
+            video_render_queue.pipeline_render(file)
         else:
             render_video_only(file)
 
 if __name__ == "__main__":
     room_folder_path = VIDEOS_DIR
-    monitor_thread = threading.Thread(target=monitor_queue, daemon=True)
+    video_render_queue = VideoRenderQueue()
+    monitor_thread = threading.Thread(target=video_render_queue.monitor_queue)
     monitor_thread.start()
     while True:
         for room_folder in Path(room_folder_path).iterdir():