Update app.py

app.py

@@ -2,11 +2,19 @@ import streamlit as st
 import ffmpeg
 import os
 from PIL import Image
+import re
+import tempfile
 
 # Supported formats
 supported_formats = ['avi', 'mp4', 'mov', 'mkv', 'flv', 'wmv', 'webm', 'mpeg', 'mpg', '3gp']
 audio_formats = ['mp3', 'wav', 'aac', 'flac']
-image_formats = sorted(Image.SAVE.keys())  # Supported image formats
+gif_formats = ['gif']
+image_formats = list(Image.SAVE.keys())  # Supported image formats
+
+def sanitize_filename(filename):
+    """Sanitize filename by removing special characters and spaces."""
+    filename = re.sub(r'[^a-zA-Z0-9_.-]', '_', filename)  # Replace invalid characters with '_'
+    return filename
 
 def get_video_duration(video_path):
     """Get video duration in seconds using ffmpeg."""
@@ -15,24 +23,29 @@ def get_video_duration(video_path):
 
 def convert_video(video, target_format, conversion_type, time_in_seconds=None):
     try:
+        # 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(video.name))[0]
-        video_path = f"./temp/{base_name}"
+        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())  # Save the uploaded video to a local file
 
         if conversion_type == 'Video to Video':
-            output_file = f"flowly_ai_video_converter_{base_name}.{target_format.lower()}"
+            output_file = f"flowly_ai_video_converter_{sanitized_base_name}.{target_format.lower()}"
             ffmpeg.input(video_path).output(output_file).run()  # Use ffmpeg.input properly
             return output_file
 
         elif conversion_type == 'Video to Audio':
-            audio_output_file = f"flowly_ai_video_to_audio_{base_name}.{target_format.lower()}"
+            audio_output_file = f"flowly_ai_video_to_audio_{sanitized_base_name}.{target_format.lower()}"
             ffmpeg.input(video_path).output(audio_output_file).run()  # Use ffmpeg.input properly
             return audio_output_file
 
         elif conversion_type == 'Video to GIF':
-            gif_output_file = f"flowly_ai_video_to_gif_{base_name}.gif"
+            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").run()  # Use ffmpeg.input properly
             return gif_output_file
 
@@ -40,13 +53,13 @@ def convert_video(video, target_format, conversion_type, time_in_seconds=None):
             if time_in_seconds is None:
                 return "Please specify a valid time in seconds for image extraction."
 
-            image_output_file = f"flowly_ai_video_to_image_{base_name}_{time_in_seconds}.png"
+            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).run()  # Use ffmpeg.input properly
 
             # 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 = f"flowly_ai_video_to_image_{base_name}_{time_in_seconds}.{target_format.lower()}"
+                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