Spaces:
Runtime error
Runtime error
File size: 2,696 Bytes
ec2b971 |
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 |
import streamlit as st
import imageio
import cv2
import os
def create_gif(video_path, start_time, end_time, gif_path, speed_multiplier):
# Open the video file
video = cv2.VideoCapture(video_path)
# Get the FPS of the original video
fps = video.get(cv2.CAP_PROP_FPS)
# Calculate the frame numbers for start and end times
start_frame = int(start_time * fps)
end_frame = int(end_time * fps)
# Set the video to start at the start_frame
video.set(cv2.CAP_PROP_POS_FRAMES, start_frame)
frames = []
success, frame = video.read()
current_frame = start_frame
while success and current_frame <= end_frame:
frames.append(frame)
current_frame += 1
success, frame = video.read()
video.release()
# Adjust the speed of the GIF
if speed_multiplier != 1.0:
frames = frames[::int(speed_multiplier)]
# Convert frames to RGB and save as GIF
rgb_frames = [cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) for frame in frames]
imageio.mimsave(gif_path, rgb_frames, fps=fps/speed_multiplier)
def main():
st.title("Automatic GIF Creator from Videos")
uploaded_file = st.file_uploader("Upload a Video", type=["mp4", "avi", "mov", "mkv"])
if uploaded_file is not None:
with open("temp_video.mp4", "wb") as f:
f.write(uploaded_file.getbuffer())
st.video("temp_video.mp4")
start_time = st.number_input("Start Time (seconds)", min_value=0.0, value=0.0)
end_time = st.number_input("End Time (seconds)", min_value=0.0, value=5.0)
speed_multiplier = st.number_input("Speed Multiplier", min_value=0.1, max_value=10.0, value=1.0)
if st.button("Create GIF"):
if end_time > start_time:
with st.spinner("Creating GIF..."):
gif_path = "output.gif"
create_gif("temp_video.mp4", start_time, end_time, gif_path, speed_multiplier)
st.success("GIF created successfully!")
st.image(gif_path)
# Create a download button for the GIF
with open(gif_path, "rb") as file:
st.download_button(
label="Download GIF",
data=file,
file_name="output.gif",
mime="image/gif"
)
else:
st.error("End time must be greater than start time")
if __name__ == "__main__":
main()
|