feat: add a function to clean tmp after 30 min

app/helpers.py

@@ -5,6 +5,7 @@ from base64 import b64encode
 from io import BytesIO
 from pathlib import Path
 from loguru import logger as log
+import time
 
 import matplotlib.pyplot as plt
 import numpy as np
@@ -173,3 +174,37 @@ def file_size_is_valid(response):
         st.error(
             f"The file is too large to download. Maximum size allowed: {max_size_mb}MB. Duplicate this space to remove any limit."
         )
+
+
+def _get_files_to_not_delete():
+    not_delete = []
+    if os.environ.get("PREPARE_SAMPLES"):
+        for filename in ["sample_songs.json", "separate_songs.json"]:
+            try:
+                with open(filename) as f:
+                    not_delete += list(json.load(f).keys())
+            except Exception as e:
+                log.warning(e)
+    return not_delete
+
+
+def _remove_file_older_than(file_path: str, max_age_limit: float):
+    # If the file is older than the age limit, delete it
+    if os.path.getmtime(file_path) < max_age_limit:
+        try:
+            os.remove(file_path)
+        except OSError as e:
+            log.warning(f"Error: Could not delete {file_path}. Reason: {e.strerror}")
+
+
+def delete_old_files(directory: str, age_limit_seconds: int):
+    files_to_not_delete = _get_files_to_not_delete()
+    age_limit = time.time() - age_limit_seconds
+
+    # Walk through the directory
+    for dirpath, dirnames, filenames in os.walk(directory):
+        if dirpath.split("/")[-1] not in files_to_not_delete:
+            for filename in filenames:
+                if filename.split(".")[0] not in files_to_not_delete:
+                    file_path = os.path.join(dirpath, filename)
+                    _remove_file_older_than(file_path, age_limit)

app/pages/About.py

@@ -2,6 +2,7 @@ import streamlit as st
 
 from header import header
 from footer import footer
+from helpers import delete_old_files
 
 
 def body():
@@ -152,3 +153,4 @@ if __name__ == "__main__":
     header(logo_and_title=False)
     body()
     footer()
+    delete_old_files("/tmp", 60 * 30)

app/pages/Karaoke.py

@@ -14,6 +14,7 @@ from helpers import (
     load_audio_segment,
     streamlit_player,
     local_audio,
+    delete_old_files,
 )
 
 from service.vocal_remover.runner import separate, load_model
@@ -182,3 +183,4 @@ if __name__ == "__main__":
     header()
     body()
     footer()
+    delete_old_files("/tmp", 60 * 30)

app/pages/Separate.py

@@ -15,6 +15,7 @@ from helpers import (
     st_local_audio,
     url_is_valid,
     file_size_is_valid,
+    delete_old_files,
 )
 from service.demucs_runner import separator
 from service.vocal_remover.runner import load_model, separate
@@ -281,3 +282,4 @@ if __name__ == "__main__":
     header()
     body()
     footer()
+    delete_old_files("/tmp", 60 * 30)