edited delete function

app.py

@@ -1,10 +1,22 @@
 import os
+import threading
+import time
+from pathlib import Path
 
 import gradio as gr
 
-from pathlib import Path
+from utlis import (HHMMSS_to_sec, convert_video_format, molly_xlsx_to_table,
+                   table_to_ELAN_tsv, trim_media)
+
 
-from utlis import HHMMSS_to_sec, convert_video_format, trim_media, molly_xlsx_to_table, table_to_ELAN_tsv
+def delete_files(files):
+    time.sleep(300)
+    for file in files:
+        try:
+            os.remove(file)
+        except FileNotFoundError:
+            pass
+    print("files deleted")
 
 def trim_video_wt(input_file, input_transcript, output_format, start_time, end_time):
     # trim video with transcript
@@ -32,6 +44,12 @@ def trim_video_wt(input_file, input_transcript, output_format, start_time, end_t
         transcript_name = f"{Path(input_transcript.name).stem.partition('.')[0]}_trimmed.tsv"
         transcript_file = os.path.join(output_folder, transcript_name)
         table_to_ELAN_tsv(table, transcript_file)
+
+        # remove file after 10 minutes for security
+        print("start deleting files")
+        path_to_delete = [input_file.name, input_transcript.name, output_file, transcript_file]
+        threading.Thread(target=delete_files, args=([path_to_delete])).start()    
+
         return output_file, transcript_file
     except Exception as e:
         return f"Error: {str(e)}"
@@ -54,7 +72,12 @@ def trim_video(input_file, output_format, start_time, end_time):
         # Trim the video
         print("start trimming")
         output_file = trim_media(input_file.name, output_file, start_time, end_time)
-        
+
+        # remove file after 10 minutes for security
+        print("start deleting files")
+        path_to_delete = [input_file.name, output_file]
+        threading.Thread(target=delete_files, args=([path_to_delete])).start()
+
         return output_file
     except Exception as e:
         return f"Error: {str(e)}"
@@ -74,6 +97,11 @@ def convert_video(input_file, output_format):
         print("start converting")
         output_file = convert_video_format(input_file.name, output_file)
         
+        # remove file after 10 minutes for security 
+        print("start deleting files")
+        path_to_delete = [input_file.name, output_file]
+        threading.Thread(target=delete_files, args=([path_to_delete])).start()
+
         return output_file
     except Exception as e:
         return f"Error: {str(e)}"

utlis.py

@@ -6,6 +6,7 @@ from pathlib import Path
 
 import pandas as pd
 
+
 def convert_video_format(media_in, media_out):
     WAV_CHANNELS = 1
     WAV_SAMPLE_RATE = 16000