Update app.py

app.py

@@ -10,6 +10,32 @@ st.set_page_config(layout="wide", page_title="Video Conversion Tool")
 # Supported formats
 supported_formats = sorted(['3GP', 'ASF', 'AVI', 'DIVX', 'FLV', 'M2TS', 'M4V', 'MKV', 'MOV', 'MP4', 'MPEG', 'MPG', 'MTS', 'TS', 'VOB', 'WEBM', 'WMV', 'XVID'])
 audio_formats = sorted(['MP3', 'WAV', 'AAC', 'FLAC', 'OGG', 'M4A', 'ALAC', 'WMA', 'AIFF', 'OPUS', 'APE', 'CAF', 'PCM', 'DTS', 'TTA', 'AMR', 'MID', 'SPX', 'WV', 'RA', 'TAK'])
+gif_formats = ['GIF']
+image_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',
+                                              'WEBP', 'WMX', 'XBM']
+
+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."""
+    probe = ffmpeg.probe(video_path, v='error', select_streams='v:0', show_entries='stream=duration')
+    duration_in_seconds = float(probe['streams'][0]['duration'])
+    return duration_in_seconds
+
+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]
+        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:
@@ -17,6 +43,46 @@ audio_formats = sorted(['MP3', 'WAV', 'AAC', 'FLAC', 'OGG', 'M4A', 'ALAC', 'WMA'
 
         if conversion_type == 'Video to Video':
             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_{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_{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
+
+        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 = 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_{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':
+        return supported_formats
+    elif conversion_type == 'Video to Audio':
+        return audio_formats
+    elif conversion_type == 'Video to GIF':
+        return gif_formats
+    elif conversion_type == 'Video to Image':
         return image_formats
     return []
 
@@ -80,4 +146,4 @@ def main():
                         )
 
 if __name__ == "__main__":
-    main()
+    main()