Update app.py

app.py

@@ -1,7 +1,10 @@
+#
 import gradio as gr
 import subprocess as sp
 import os
 import uuid
+import time
+import shutil
 
 os.makedirs("./output", exist_ok=True)
 
@@ -11,7 +14,7 @@ def run(*args):
         return "Source file does not exist"
     if not os.path.exists(target):
         return "Target file does not exist"
-
+    remove_old_directories("./output", num_minutes=2)
     filename = os.path.basename(target)
     os.makedirs(f"./output/{unique_id}",exist_ok=True)
     output = f"./output/{unique_id}/{filename}"
@@ -68,6 +71,24 @@ def clear_output(unique_id):
     except Exception as e:
         return f"An error occurred: {str(e)}"
 
+def remove_old_directories(directory, num_minutes=10):
+    now = time.time()
+
+    for r, d, f in os.walk(directory):
+        for dir_name in d:
+            dir_path = os.path.join(r, dir_name)
+            timestamp = os.path.getmtime(dir_path)
+            age_minutes = (now - timestamp) / 60 # Convert to minutes
+            
+            if age_minutes >= num_minutes:
+                try:
+                    print("Removing", dir_path)
+                    shutil.rmtree(dir_path)
+                    print("Directory removed:", dir_path)
+                except Exception as e:
+                    print(e)
+                    pass
+
 def get_theme() -> gr.Theme:
     return gr.themes.Soft(
         primary_hue = gr.themes.colors.red,
@@ -91,6 +112,7 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
             value = ['face_swapper']  # Default value
         )
 
+    
     with gr.Box():
         with gr.Column(scale=3):
             face_analyser_direction_dropdown = gr.Dropdown(