Update app.py

app.py

@@ -6,28 +6,49 @@ import ffmpeg
 # Atur tema Gradio
 gr.themes.Soft()
 
+def delete_temp_dir(directory, delay=900):  # 15 minutes = 900 seconds
+    timer = threading.Timer(delay, shutil.rmtree, [directory])
+    timer.start()
+    
 # Format gambar yang didukung
 supported_formats = sorted(Image.SAVE.keys()) or ['BLP', 'BMP', 'BUFR', 'DDS', 'DIB', 'EPS',
                                                   'GIF', 'GRIB', 'HDF5', 'ICNS', 'ICO', 'IM',
                                                   'JPEG', 'JPEG2000', 'MPO', 'MSP', 'PALM', 'PCX',
                                                   'PDF', 'PNG', 'PPM', 'SGI', 'SPIDER', 'TGA', 'TIFF']
 
+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."""
+    return re.sub(r'[^a-zA-Z0-9_.-]', '_', filename)
+
+def clean_cache():
+    """Remove files in the cache directory older than 15 minutes."""
+    now = time.time()
+    for file in os.listdir(CACHE_DIR):
+        file_path = os.path.join(CACHE_DIR, file)
+        if os.path.isfile(file_path) and now - os.path.getmtime(file_path) > 900:  # 15 minutes
+            os.remove(file_path)
+            
 def convert_image_ffmpeg(image, target_format):
     try:
-        # Ambil nama file asli tanpa ekstensi
-        base_name = os.path.splitext(os.path.basename(image))[0]
+        # Create a temporary directory for the uploaded file
+        temp_dir = tempfile.mkdtemp()
+
+        # Sanitize the filename and save the uploaded video to the temp directory
+        base_name = os.path.splitext(os.path.basename(image.name))[0]
+        sanitized_base_name = sanitize_filename(base_name)
+        image_path = os.path.join(temp_dir, f"{sanitized_base_name}.png")  # Saving as mp4 by default for now
         
         # Nama file keluaran dengan awalan yang diinginkan
         output_file = f"flowly_ai_image_converter_{base_name}.{target_format.lower()}"
-        
+
+        with open(image_path, "wb") as f:
+            f.write(image.getbuffer())  # Save the uploaded video to a local file
+            
         # Gunakan ffmpeg untuk konversi
-        (
-            ffmpeg
-            .input(image)
-            .output(output_file, format=target_format)
-            .overwrite_output()
-            .run(capture_stdout=True, capture_stderr=True)
-        )
+        ffmpeg.input(image_path).output(output_file, format=target_format).overwrite_output().run(capture_stdout=True, capture_stderr=True)
 
         return output_file
     except Exception as e: