Spaces:
Sleeping
Sleeping
Aditya DN commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,8 +6,7 @@ from PIL import Image
|
|
| 6 |
# Supported formats
|
| 7 |
supported_formats = ['avi', 'mp4', 'mov', 'mkv', 'flv', 'wmv', 'webm', 'mpeg', 'mpg', '3gp']
|
| 8 |
audio_formats = ['mp3', 'wav', 'aac', 'flac']
|
| 9 |
-
|
| 10 |
-
image_formats = list(Image.SAVE.keys()) # Supported image formats
|
| 11 |
|
| 12 |
def get_video_duration(video_path):
|
| 13 |
"""Get video duration in seconds using ffmpeg."""
|
|
@@ -17,20 +16,24 @@ def get_video_duration(video_path):
|
|
| 17 |
def convert_video(video, target_format, conversion_type, time_in_seconds=None):
|
| 18 |
try:
|
| 19 |
base_name = os.path.splitext(os.path.basename(video.name))[0]
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
if conversion_type == 'Video to Video':
|
| 22 |
output_file = f"flowly_ai_video_converter_{base_name}.{target_format.lower()}"
|
| 23 |
-
ffmpeg.input(
|
| 24 |
return output_file
|
| 25 |
|
| 26 |
elif conversion_type == 'Video to Audio':
|
| 27 |
audio_output_file = f"flowly_ai_video_to_audio_{base_name}.{target_format.lower()}"
|
| 28 |
-
ffmpeg.input(
|
| 29 |
return audio_output_file
|
| 30 |
|
| 31 |
elif conversion_type == 'Video to GIF':
|
| 32 |
gif_output_file = f"flowly_ai_video_to_gif_{base_name}.gif"
|
| 33 |
-
ffmpeg.input(
|
| 34 |
return gif_output_file
|
| 35 |
|
| 36 |
elif conversion_type == 'Video to Image':
|
|
@@ -38,8 +41,9 @@ def convert_video(video, target_format, conversion_type, time_in_seconds=None):
|
|
| 38 |
return "Please specify a valid time in seconds for image extraction."
|
| 39 |
|
| 40 |
image_output_file = f"flowly_ai_video_to_image_{base_name}_{time_in_seconds}.png"
|
| 41 |
-
ffmpeg.input(
|
| 42 |
|
|
|
|
| 43 |
img = Image.open(image_output_file)
|
| 44 |
if target_format.lower() in image_formats:
|
| 45 |
image_output_file = f"flowly_ai_video_to_image_{base_name}_{time_in_seconds}.{target_format.lower()}"
|
|
|
|
| 6 |
# Supported formats
|
| 7 |
supported_formats = ['avi', 'mp4', 'mov', 'mkv', 'flv', 'wmv', 'webm', 'mpeg', 'mpg', '3gp']
|
| 8 |
audio_formats = ['mp3', 'wav', 'aac', 'flac']
|
| 9 |
+
image_formats = sorted(Image.SAVE.keys()) # Supported image formats
|
|
|
|
| 10 |
|
| 11 |
def get_video_duration(video_path):
|
| 12 |
"""Get video duration in seconds using ffmpeg."""
|
|
|
|
| 16 |
def convert_video(video, target_format, conversion_type, time_in_seconds=None):
|
| 17 |
try:
|
| 18 |
base_name = os.path.splitext(os.path.basename(video.name))[0]
|
| 19 |
+
video_path = f"./temp/{base_name}"
|
| 20 |
+
|
| 21 |
+
with open(video_path, "wb") as f:
|
| 22 |
+
f.write(video.getbuffer()) # Save the uploaded video to a local file
|
| 23 |
|
| 24 |
if conversion_type == 'Video to Video':
|
| 25 |
output_file = f"flowly_ai_video_converter_{base_name}.{target_format.lower()}"
|
| 26 |
+
ffmpeg.input(video_path).output(output_file).run() # Use ffmpeg.input properly
|
| 27 |
return output_file
|
| 28 |
|
| 29 |
elif conversion_type == 'Video to Audio':
|
| 30 |
audio_output_file = f"flowly_ai_video_to_audio_{base_name}.{target_format.lower()}"
|
| 31 |
+
ffmpeg.input(video_path).output(audio_output_file).run() # Use ffmpeg.input properly
|
| 32 |
return audio_output_file
|
| 33 |
|
| 34 |
elif conversion_type == 'Video to GIF':
|
| 35 |
gif_output_file = f"flowly_ai_video_to_gif_{base_name}.gif"
|
| 36 |
+
ffmpeg.input(video_path).output(gif_output_file, vf="fps=10,scale=320:-1:flags=lanczos").run() # Use ffmpeg.input properly
|
| 37 |
return gif_output_file
|
| 38 |
|
| 39 |
elif conversion_type == 'Video to Image':
|
|
|
|
| 41 |
return "Please specify a valid time in seconds for image extraction."
|
| 42 |
|
| 43 |
image_output_file = f"flowly_ai_video_to_image_{base_name}_{time_in_seconds}.png"
|
| 44 |
+
ffmpeg.input(video_path, ss=time_in_seconds).output(image_output_file, vframes=1).run() # Use ffmpeg.input properly
|
| 45 |
|
| 46 |
+
# Convert the image to the desired format using Pillow
|
| 47 |
img = Image.open(image_output_file)
|
| 48 |
if target_format.lower() in image_formats:
|
| 49 |
image_output_file = f"flowly_ai_video_to_image_{base_name}_{time_in_seconds}.{target_format.lower()}"
|