Spaces:
Running
Running
File size: 3,331 Bytes
f3da3f2 9de7e25 d81f996 9de7e25 f3da3f2 9de7e25 18a0f73 d81f996 d2cdcd3 f3da3f2 9de7e25 d81f996 9de7e25 f3da3f2 9de7e25 f3da3f2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
import streamlit as st
import ffmpeg
import os
from PIL import Image
import re
import tempfile
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'])
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_{sanitized_base_name}.{target_format.lower()}"
return image_formats
return []
def main():
st.title("Video Conversion Tool")
st.write("Convert videos to audio, GIFs, images, or other formats easily with this powerful tool.")
# Create two columns
col1, col2 = st.columns([1, 1])
with col1:
# Upload video file
video_file = st.file_uploader("Upload a Video", type=supported_formats)
if video_file:
st.video(video_file)
with col2:
if video_file:
# Create a temporary file and save uploaded video
temp_video_path = os.path.join(tempfile.mkdtemp(), video_file.name)
with open(temp_video_path, "wb") as f:
f.write(video_file.getbuffer())
# Get video duration
video_duration = get_video_duration(temp_video_path)
# Select conversion type
conversion_type = st.selectbox(
"Select Conversion Type",
['Video to Video', 'Video to Audio', 'Video to GIF', 'Video to Image']
)
# Update format choices based on conversion type
target_format_choices = update_format_choices(conversion_type)
target_format = st.selectbox("Select Target Format", target_format_choices)
# If 'Video to Image' conversion, ask for time in seconds
if conversion_type == 'Video to Image':
time_in_seconds = st.slider(
label="Time (in seconds) for image extraction",
min_value=0,
max_value=int(video_duration),
value=0,
step=1
)
else:
time_in_seconds = None
if st.button("Convert"):
with st.spinner("Converting..."):
output_file = convert_video(video_file, target_format, conversion_type, time_in_seconds)
if "Error" in output_file:
st.error(output_file)
else:
st.success(f"Conversion Successful! Download the file:")
st.download_button(
label="Download Converted File",
data=open(output_file, 'rb').read(),
file_name=output_file
)
if __name__ == "__main__":
main() |