Add application files and dependencies

video_processor.py

@@ -4,12 +4,17 @@ import glob
 import re
 import select
 import shutil
+import threading
+import psutil
+import signal
 
 # Dictionary to store the status of each video processing task
 status_dict = {}
+process_dict = {}
 
 def run_command_with_progress(command, session_id, tool_name, is_trimming=False):
     process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, universal_newlines=True)
+    process_dict[session_id] = process
     
     status_dict[session_id]["status"] = f"Using {tool_name}..."
     
@@ -19,6 +24,11 @@ def run_command_with_progress(command, session_id, tool_name, is_trimming=False)
     duration = None
     
     while True:
+        if status_dict[session_id].get("status") == "Cancelled":
+            process.terminate()
+            process.wait()
+            return -1
+
         reads = [process.stdout.fileno(), process.stderr.fileno()]
         ret = select.select(reads, [], [], 0.1)
 
@@ -65,6 +75,7 @@ def run_command_with_progress(command, session_id, tool_name, is_trimming=False)
     
     return return_code
 
+
 def process_video(url, session_id):
     output_dir = f"./output_folder_{session_id}"
     os.makedirs(output_dir, exist_ok=True)
@@ -135,6 +146,7 @@ def process_video(url, session_id):
         status_dict[session_id]["status"] = f"Processing completed! Directory: {trimmed_dir}"
         return trimmed_dir
 
+
 def get_status(session_id):
     return status_dict.get(session_id, {}).get("status", "Invalid session ID.")
 
@@ -144,3 +156,21 @@ def get_error_details(session_id):
 def cancel_job(session_id):
     if session_id in status_dict:
         status_dict[session_id]["status"] = "Cancelled"
+        if session_id in process_dict:
+            process = process_dict[session_id]
+            if process.poll() is None:  # Check if the process is still running
+                parent = psutil.Process(process.pid)
+                for child in parent.children(recursive=True):
+                    child.terminate()
+                parent.terminate()
+                process_dict.pop(session_id)
+        return True
+    return False
+
+def cleanup_cancelled_jobs():
+    for session_id in list(status_dict.keys()):
+        if status_dict[session_id]["status"] == "Cancelled":
+            output_dir = f"./output_folder_{session_id}"
+            if os.path.exists(output_dir):
+                shutil.rmtree(output_dir)
+            status_dict.pop(session_id)