Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,4 +1,7 @@
|
|
| 1 |
import gradio as gr
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
def save_and_display_video(video_file):
|
| 4 |
if video_file is None:
|
|
@@ -10,17 +13,37 @@ def save_and_display_video(video_file):
|
|
| 10 |
file_path = 'uploaded_video.mp4' # Assuming .mp4 for simplicity
|
| 11 |
with open(file_path, 'wb') as f:
|
| 12 |
f.write(video_file)
|
| 13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
else:
|
| 15 |
return None, "Uploaded file is empty."
|
| 16 |
except Exception as e:
|
| 17 |
return None, f"An error occurred: {str(e)}"
|
| 18 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 19 |
with gr.Blocks() as demo:
|
| 20 |
with gr.Column():
|
| 21 |
video_file = gr.File(label="Upload Video File", type="binary", file_types=["video"], interactive=True)
|
| 22 |
output_video = gr.Video()
|
| 23 |
-
output_message = gr.Textbox(label="Output Message")
|
| 24 |
submit_button = gr.Button("Display Video")
|
| 25 |
submit_button.click(
|
| 26 |
fn=save_and_display_video,
|
|
|
|
| 1 |
import gradio as gr
|
| 2 |
+
import os
|
| 3 |
+
from scenedetect import VideoManager, SceneManager
|
| 4 |
+
from scenedetect.detectors import ContentDetector
|
| 5 |
|
| 6 |
def save_and_display_video(video_file):
|
| 7 |
if video_file is None:
|
|
|
|
| 13 |
file_path = 'uploaded_video.mp4' # Assuming .mp4 for simplicity
|
| 14 |
with open(file_path, 'wb') as f:
|
| 15 |
f.write(video_file)
|
| 16 |
+
|
| 17 |
+
scenes = find_scenes(file_path)
|
| 18 |
+
scene_info = ', '.join([f"Start: {start.get_seconds()}, End: {end.get_seconds()}" for start, end in scenes])
|
| 19 |
+
if scenes:
|
| 20 |
+
return file_path, f"Video uploaded and displayed successfully. Scenes detected: {scene_info}"
|
| 21 |
+
else:
|
| 22 |
+
return file_path, "Video uploaded but no scenes were detected."
|
| 23 |
else:
|
| 24 |
return None, "Uploaded file is empty."
|
| 25 |
except Exception as e:
|
| 26 |
return None, f"An error occurred: {str(e)}"
|
| 27 |
|
| 28 |
+
def find_scenes(video_path):
|
| 29 |
+
video_manager = VideoManager([video_path])
|
| 30 |
+
scene_manager = SceneManager()
|
| 31 |
+
scene_manager.add_detector(ContentDetector(threshold=30))
|
| 32 |
+
video_manager.set_downscale_factor()
|
| 33 |
+
|
| 34 |
+
video_manager.start()
|
| 35 |
+
scene_manager.detect_scenes(frame_source=video_manager)
|
| 36 |
+
|
| 37 |
+
scene_list = scene_manager.get_scene_list(base_timecode=video_manager.get_base_timecode())
|
| 38 |
+
video_manager.release()
|
| 39 |
+
|
| 40 |
+
return scene_list
|
| 41 |
+
|
| 42 |
with gr.Blocks() as demo:
|
| 43 |
with gr.Column():
|
| 44 |
video_file = gr.File(label="Upload Video File", type="binary", file_types=["video"], interactive=True)
|
| 45 |
output_video = gr.Video()
|
| 46 |
+
output_message = gr.Textbox(label="Output Message", lines=4)
|
| 47 |
submit_button = gr.Button("Display Video")
|
| 48 |
submit_button.click(
|
| 49 |
fn=save_and_display_video,
|