Spaces:
Runtime error
Runtime error
| 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() | |