Spaces:
Runtime error
Runtime error
File size: 3,720 Bytes
5957fa3 |
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 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 |
import cv2
import numpy as np
import pandas as pd
import tempfile
import os
import ffmpeg
from typing import Dict, Any, Optional, Tuple
import time
def extract_video_info(video_path: str) -> Dict[str, Any]:
"""Extract video information using OpenCV."""
cap = cv2.VideoCapture(video_path)
if not cap.isOpened():
raise ValueError("Could not open video file")
# Get video properties
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = cap.get(cv2.CAP_PROP_FPS)
frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
duration = frame_count / fps
cap.release()
return {
'width': width,
'height': height,
'fps': fps,
'frame_count': frame_count,
'duration': duration
}
def create_timeline_figure(video_info: Dict[str, Any]) -> go.Figure:
"""Create a timeline visualization for the video."""
# Create timeline data
time_points = np.linspace(0, video_info['duration'], 100)
intensity = np.sin(time_points * 2) * 0.5 + 0.5 # Simulated intensity
fig = go.Figure()
fig.add_trace(go.Scatter(
x=time_points,
y=intensity,
mode='lines',
fill='tozeroy',
name='Video Intensity'
))
fig.update_layout(
title="Video Timeline",
xaxis_title="Time (seconds)",
yaxis_title="Intensity",
hovermode='x unified'
)
return fig
def apply_editing_effects(
video_path: str,
start_time: float,
end_time: float,
effects: List[str],
filter_type: str,
speed: float
) -> str:
"""Apply editing effects to the video using ffmpeg."""
# Create temporary output file
output_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name
# Build ffmpeg command
cmd = [
'ffmpeg',
'-i', video_path,
'-ss', str(start_time),
'-to', str(end_time),
'-vf', f"setpts={1/speed}*PTS"
]
# Add effects
if effects:
effect_filters = []
if 'Fade In' in effects:
effect_filters.append('fade=in:st=0:d=1')
if 'Fade Out' in effects:
effect_filters.append('fade=out:st={}:d=1'.format(end_time - 1))
if 'Blur' in effects:
effect_filters.append('boxblur=5:1')
if 'Sharpen' in effects:
effect_filters.append('unsharp=5:5:1.0:5:5:0.0')
if effect_filters:
cmd.extend(['-vf', ','.join(effect_filters)])
# Add filter
if filter_type != 'None':
filter_map = {
'Grayscale': 'hue=s=0',
'Sepia': 'colorchannelmixer=.393:.769:.189:0:.349:.686:.168:0:.272:.534:.131',
'Vintage': 'lut3d=vintage.lut',
'Cool': 'lut3d=cool.lut',
'Warm': 'lut3d=warm.lut'
}
if filter_type in filter_map:
cmd.extend(['-vf', filter_map[filter_type]])
# Add output
cmd.extend(['-c:v', 'libx264', '-c:a', 'aac', output_file])
# Execute ffmpeg
try:
ffmpeg.run(cmd, overwrite_output=True, quiet=True)
return output_file
except Exception as e:
raise RuntimeError(f"Error applying effects: {str(e)}")
def process_video_upload(uploaded_file) -> Tuple[str, Dict[str, Any]]:
"""Process uploaded video file and return path and info."""
# Save uploaded file to temporary location
temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
temp_file.write(uploaded_file.read())
temp_file.close()
# Extract video info
video_info = extract_video_info(temp_file.name)
return temp_file.name, video_info |