Update app.py

app.py

@@ -6,6 +6,11 @@ from PIL import Image
 import re
 import tempfile
 import shutil
+import threading
+
+def delete_temp_dir(directory, delay=900):  # 15 minutes = 900 seconds
+    timer = threading.Timer(delay, shutil.rmtree, [directory])
+    timer.start()
 
 st.set_page_config(layout="wide", page_title="Video Conversion Tool")
 
@@ -18,6 +23,7 @@ image_formats = sorted(Image.SAVE.keys()) or ['BLP', 'BMP', 'BUFR', 'DDS', 'DIB'
                                               'TGA', 'TIFF', 'WEBP', 'WMX', 'XBM']
 
 CACHE_DIR = tempfile.mkdtemp()
+delete_temp_dir(CACHE_DIR, delay=900)  # Clean up cache after 15 minutes
 
 def sanitize_filename(filename):
     """Sanitize filename by removing special characters and spaces."""
@@ -38,48 +44,50 @@ def get_video_duration(video_path):
 
 def convert_video(video, target_format, conversion_type, time_in_seconds=None):
     try:
-        clean_cache()  # Clean up cache before processing
+        # Create a temporary directory for the uploaded file
+        temp_dir = tempfile.mkdtemp()
 
-        # Save the uploaded video to the temp directory
-        sanitized_base_name = sanitize_filename(os.path.splitext(video.name)[0])
-        video_path = os.path.join(CACHE_DIR, f"{sanitized_base_name}.mp4")
+        # Sanitize the filename and save the uploaded video to the temp directory
+        base_name = os.path.splitext(os.path.basename(video.name))[0]
+        sanitized_base_name = sanitize_filename(base_name)
+        video_path = os.path.join(temp_dir, f"{sanitized_base_name}.mp4")  # Saving as mp4 by default for now
 
         with open(video_path, "wb") as f:
-            f.write(video.getbuffer())
+            f.write(video.getbuffer())  # Save the uploaded video to a local file
 
         if conversion_type == 'Video to Video':
-            output_file = os.path.join(CACHE_DIR, f"flowly_ai_video_converter_{sanitized_base_name}.{target_format.lower()}")
-            ffmpeg.input(video_path).output(output_file, overwrite_output=True).run(overwrite_output=True)
+            output_file = f"flowly_ai_video_converter_{sanitized_base_name}.{target_format.lower()}"
+            ffmpeg.input(video_path).output(output_file).overwrite_output().run()  # Add .overwrite_output()
             return output_file
 
         elif conversion_type == 'Video to Audio':
-            output_file = os.path.join(CACHE_DIR, f"flowly_ai_video_to_audio_{sanitized_base_name}.{target_format.lower()}")
-            ffmpeg.input(video_path).output(output_file, overwrite_output=True).run(overwrite_output=True)
-            return output_file
+            audio_output_file = f"flowly_ai_video_to_audio_{sanitized_base_name}.{target_format.lower()}"
+            ffmpeg.input(video_path).output(audio_output_file).overwrite_output().run()  # Add .overwrite_output()
+            return audio_output_file
 
         elif conversion_type == 'Video to GIF':
-            output_file = os.path.join(CACHE_DIR, f"flowly_ai_video_to_gif_{sanitized_base_name}.gif")
-            ffmpeg.input(video_path).output(output_file, vf="fps=10,scale=320:-1:flags=lanczos", overwrite_output=True).run(overwrite_output=True)
-            return output_file
+            gif_output_file = f"flowly_ai_video_to_gif_{sanitized_base_name}.gif"
+            ffmpeg.input(video_path).output(gif_output_file, vf="fps=10,scale=320:-1:flags=lanczos").overwrite_output().run()  # Add .overwrite_output()
+            return gif_output_file
 
         elif conversion_type == 'Video to Image':
             if time_in_seconds is None:
                 return "Please specify a valid time in seconds for image extraction."
 
-            image_output_file = os.path.join(CACHE_DIR, f"flowly_ai_video_to_image_{sanitized_base_name}_{time_in_seconds}.png")
-            ffmpeg.input(video_path, ss=time_in_seconds).output(image_output_file, vframes=1, overwrite_output=True).run(overwrite_output=True)
+            image_output_file = f"flowly_ai_video_to_image_{sanitized_base_name}_{time_in_seconds}.png"
+            ffmpeg.input(video_path, ss=time_in_seconds).output(image_output_file, vframes=1).overwrite_output().run()  # Add .overwrite_output()
 
             # Convert the image to the desired format using Pillow
             img = Image.open(image_output_file)
             if target_format.lower() in image_formats:
-                image_output_file = os.path.join(CACHE_DIR, f"flowly_ai_video_to_image_{sanitized_base_name}_{time_in_seconds}.{target_format.lower()}")
-                img.save(image_output_file, target_format.upper())
+                image_output_file = f"flowly_ai_video_to_image_{sanitized_base_name}_{time_in_seconds}.{target_format.lower()}"
+                img.save(image_output_file, target_format.upper())  # Save with the desired format
 
             return image_output_file
 
     except Exception as e:
         return f"Error: {e}"
-
+        
 def update_format_choices(conversion_type):
     """Update available format choices based on the conversion type."""
     if conversion_type == 'Video to Video':