working
Browse files
app.py
CHANGED
|
@@ -8,12 +8,30 @@ def process_video(video_path, nframes, height, width, direction, trim, average,
|
|
| 8 |
try:
|
| 9 |
fv = FrameVis()
|
| 10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
# Process the video
|
| 12 |
output_image = fv.visualize(
|
| 13 |
video_path,
|
| 14 |
nframes=nframes,
|
| 15 |
-
height=height
|
| 16 |
-
width=width
|
| 17 |
direction=direction,
|
| 18 |
trim=trim,
|
| 19 |
quiet=False
|
|
@@ -37,7 +55,7 @@ with gr.Blocks(title="FrameVis - Video Frame Visualizer") as demo:
|
|
| 37 |
gr.Markdown("""
|
| 38 |
# 🎬 FrameVis - Video Frame Visualizer
|
| 39 |
Upload a video to create a beautiful visualization of its frames. The tool will extract frames at regular intervals
|
| 40 |
-
and combine them into a single image.
|
| 41 |
""")
|
| 42 |
|
| 43 |
with gr.Row():
|
|
@@ -45,7 +63,7 @@ with gr.Blocks(title="FrameVis - Video Frame Visualizer") as demo:
|
|
| 45 |
# Input components
|
| 46 |
video_input = gr.Video(label="Upload Video")
|
| 47 |
with gr.Row():
|
| 48 |
-
nframes = gr.Slider(minimum=1, maximum=
|
| 49 |
label="Number of Frames")
|
| 50 |
direction = gr.Radio(["horizontal", "vertical"], value="horizontal",
|
| 51 |
label="Direction")
|
|
@@ -64,7 +82,7 @@ with gr.Blocks(title="FrameVis - Video Frame Visualizer") as demo:
|
|
| 64 |
|
| 65 |
with gr.Column(scale=2):
|
| 66 |
# Output component
|
| 67 |
-
output_image = gr.Image(label="Visualization Result"
|
| 68 |
|
| 69 |
# Handle processing
|
| 70 |
process_btn.click(
|
|
|
|
| 8 |
try:
|
| 9 |
fv = FrameVis()
|
| 10 |
|
| 11 |
+
# Get video properties to calculate appropriate dimensions
|
| 12 |
+
cap = cv2.VideoCapture(video_path)
|
| 13 |
+
video_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|
| 14 |
+
video_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
|
| 15 |
+
cap.release()
|
| 16 |
+
|
| 17 |
+
# Calculate dimensions to maintain aspect ratio
|
| 18 |
+
if width <= 0:
|
| 19 |
+
# If width is auto, use the video's native width * 2 as a reasonable default
|
| 20 |
+
width = video_width * 2
|
| 21 |
+
|
| 22 |
+
if height <= 0:
|
| 23 |
+
# Calculate height to maintain aspect ratio
|
| 24 |
+
if direction == "horizontal":
|
| 25 |
+
height = int((width / video_width) * video_height)
|
| 26 |
+
else:
|
| 27 |
+
height = video_height
|
| 28 |
+
|
| 29 |
# Process the video
|
| 30 |
output_image = fv.visualize(
|
| 31 |
video_path,
|
| 32 |
nframes=nframes,
|
| 33 |
+
height=height,
|
| 34 |
+
width=width,
|
| 35 |
direction=direction,
|
| 36 |
trim=trim,
|
| 37 |
quiet=False
|
|
|
|
| 55 |
gr.Markdown("""
|
| 56 |
# 🎬 FrameVis - Video Frame Visualizer
|
| 57 |
Upload a video to create a beautiful visualization of its frames. The tool will extract frames at regular intervals
|
| 58 |
+
and combine them into a single image. For best results with horizontal layout, try setting width to match your screen width.
|
| 59 |
""")
|
| 60 |
|
| 61 |
with gr.Row():
|
|
|
|
| 63 |
# Input components
|
| 64 |
video_input = gr.Video(label="Upload Video")
|
| 65 |
with gr.Row():
|
| 66 |
+
nframes = gr.Slider(minimum=1, maximum=2000, value=100, step=1,
|
| 67 |
label="Number of Frames")
|
| 68 |
direction = gr.Radio(["horizontal", "vertical"], value="horizontal",
|
| 69 |
label="Direction")
|
|
|
|
| 82 |
|
| 83 |
with gr.Column(scale=2):
|
| 84 |
# Output component
|
| 85 |
+
output_image = gr.Image(label="Visualization Result")
|
| 86 |
|
| 87 |
# Handle processing
|
| 88 |
process_btn.click(
|